tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Testing text2img command (will be replaced, only for tests); added exception handling, queue and database table with generated images

Browse Source
This commit is contained in:
BarsTiger
2023-02-20 16:41:55 +02:00
parent 3bd34009ae
commit d56f1d386f
21 changed files with 219 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
bot/modules/api/txt2img.py
View File

@@ -1,26 +1,25 @@
import aiohttp
from bot.db import db, DBTables
from .objects.prompt_request import Prompt
import json
import base64
async def txt2img(prompt: str, negative_prompt: str = None, steps: int = 20,
cfg_scale: int = 7, width: int = 768, height: int = 768,
restore_faces: bool = True, sampler: str = "Euler a") -> list[bytes, dict] | None:
async def txt2img(prompt: Prompt, ignore_exceptions: bool = False) -> list[bytes, dict] | None:
endpoint = db[DBTables.config].get('endpoint')
try:
async with aiohttp.ClientSession() as session:
r = await session.post(
endpoint + "/sdapi/v1/txt2img",
json={
"prompt": prompt,
"steps": steps,
"cfg_scale": cfg_scale,
"width": width,
"height": height,
"restore_faces": restore_faces,
"negative_prompt": negative_prompt,
"sampler_index": sampler
"prompt": prompt.prompt,
"steps": prompt.steps,
"cfg_scale": prompt.cfg_scale,
"width": prompt.width,
"height": prompt.height,
"restore_faces": prompt.restore_faces,
"negative_prompt": prompt.negative_prompt,
"sampler_index": prompt.sampler
}
)
if r.status != 200:
@@ -28,5 +27,6 @@ async def txt2img(prompt: str, negative_prompt: str = None, steps: int = 20,
return [base64.b64decode((await r.json())["images"][0]),
json.loads((await r.json())["info"])]
except Exception as e:
assert e
return None
if not ignore_exceptions:
raise e
return