soaska/sd_telegram
soaska/sd_telegram
1
0
Fork 1
Code Issues 2 Pull Requests Actions Packages Activity

fix gen

Browse Source
This commit is contained in:
Mikhail Guseletov 2023-06-27 13:27:51 +07:00
parent f6fb06d8f1
commit fe965a4084
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
bot.py
View File

@ -576,7 +576,19 @@ async def inl_gen(message: Union[types.Message, types.CallbackQuery]) -> None:
asyncio.create_task(getProgress(msgTime)) asyncio.create_task(getProgress(msgTime))
# TODO try catch if wrong data # TODO try catch if wrong data
res = await api.txt2img(**data) res = await api.txt2img(**data)
await show_thumbs(chatId, res) # show_thumbs dont work because use_async
if dataParams["img_thumb"] == "true" or dataParams["img_thumb"] == "True":
await bot.send_media_group(
chat_id=chatId, media=pilToImages(res, "thumbs")
)
if dataParams["img_tg"] == "true" or dataParams["img_tg"] == "True":
await bot.send_media_group(
chat_id=chatId, media=pilToImages(res, "tg")
)
if dataParams["img_real"] == "true" or dataParams["img_real"] == "True":
await bot.send_media_group(
chat_id=chatId, media=pilToImages(res, "real")
)
await bot.send_message( await bot.send_message(
chat_id=chatId, chat_id=chatId,
text=data["prompt"] + "\n" + str(res.info["all_seeds"]), text=data["prompt"] + "\n" + str(res.info["all_seeds"]),
@ -755,7 +767,6 @@ async def inl_rnd_inf(message: Union[types.Message, types.CallbackQuery]) -> Non
data['sampler_name'] = samplers[num_smp]['name'] # Ý data['sampler_name'] = samplers[num_smp]['name'] # Ý
data["use_async"] = False data["use_async"] = False
#data["enable_hr"] = True
# GEN # GEN
res = api.txt2img(**data) res = api.txt2img(**data)
await show_thumbs(chatId, res) await show_thumbs(chatId, res)