43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import aiohttp, os
|
|
from util.general_utils import download_image_by_url, create_markdown_image
|
|
from type.config import VERSION
|
|
|
|
BASE_RENDER_URL = "https://t2i.soulter.top/text2img"
|
|
TEMPLATE_PATH = os.path.join(os.path.dirname(__file__), "template")
|
|
|
|
async def text_to_image_base(text: str, return_url: bool = False) -> str:
|
|
'''
|
|
返回图像的文件路径
|
|
'''
|
|
with open(os.path.join(TEMPLATE_PATH, "base.html"), "r", encoding='utf-8') as f:
|
|
tmpl_str = f.read()
|
|
|
|
assert(tmpl_str)
|
|
|
|
text = text.replace("`", "\`")
|
|
|
|
post_data = {
|
|
"tmpl": tmpl_str,
|
|
"json": return_url,
|
|
"tmpldata": {
|
|
"text": text,
|
|
"version": f"v{VERSION}",
|
|
},
|
|
"options": {
|
|
"full_page": True
|
|
}
|
|
}
|
|
|
|
if return_url:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.post(f"{BASE_RENDER_URL}/generate", json=post_data) as resp:
|
|
ret = await resp.json()
|
|
return f"{BASE_RENDER_URL}/{ret['data']['id']}"
|
|
else:
|
|
image_path = ""
|
|
try:
|
|
image_path = await download_image_by_url(f"{BASE_RENDER_URL}/generate", post=True, post_data=post_data)
|
|
except Exception as e:
|
|
print(f"调用 markdown 渲染 API 失败，错误信息：{e}，将使用本地渲染方式。")
|
|
image_path = create_markdown_image(text)
|
|
return image_path |