kls/sd_telegram
1
0
Fork 0
forked from soaska/sd_telegram
Code Pull Requests Actions Packages Activity

Merge branch 'amputator84:master' into master

Browse Source
This commit is contained in:
Alex Borisov 2024-01-23 17:07:43 +03:00 committed by GitHub
commit c65836e5a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
bot.py
View File

@ -107,6 +107,7 @@ dataParams = {"img_thumb": "true",
"sd_model_checkpoint": "",
"use_prompt": "true",
"json_prompt": "false",
"just_gen": "false",
"send_vk": "false"}
dataOld = data.copy()
dataOldParams = dataParams.copy()
@ -515,7 +516,7 @@ async def rnd_script(message, typeScript):
text=dataPromptOld
)
for i, number in enumerate(numbers):
time.sleep(5)
time.sleep(1)
for itemTxt in data['prompt'].split(';'):
if typeScript == 'models':
api.util_wait_for_ready()
@ -848,13 +849,15 @@ async def inl_fp(message: Union[types.Message, types.CallbackQuery]) -> None:
"use_prompt": "true",
"json_prompt": "false"}
if m == 'fp_sdxl':
data['steps'] = 5
data['enable_hr'] = 'True'
data['denoising_strength'] = '0.3'
data['steps'] = 15
data['sampler_name'] = 'DPM++ SDE Karras'
data['enable_hr'] = 'false'
data['cfg_scale'] = '4'
data['width'] = '1024'
data['height'] = '1024'
data['restore_faces'] = 'false'
data['hr_upscaler'] = '4x_NMKD-Siax_200k'
data['do_not_save_grid'] = 'true'
data['negative_prompt'] = 'FastNegativeV2'
data['save_images'] = 'true'
@ -863,7 +866,8 @@ async def inl_fp(message: Union[types.Message, types.CallbackQuery]) -> None:
"img_real": "true",
"stop_sd": "true",
"use_prompt": "true",
"json_prompt": "false"}
"json_prompt": "false",
"just_gen": "false"}
txt = f"JSON отредактирован\n{getJson()}\n{getJson(1)}"
await getKeyboardUnion(txt, message, keyboard, '')
@ -1328,6 +1332,22 @@ async def inl_yes_no(callback: types.CallbackQuery) -> None:
reply_markup=keyboard,
)
# отлов поста с канала и мгновенная генерация если включен just_gen
@dp.channel_post_handler()
async def handle_channel_post(message: types.Message):
logging.info("handle_channel_post")
chatId = message.chat.id
if str(dataParams['just_gen']).lower() == 'true':
if len(message.text) < 500:
data["prompt"] = translateRuToEng(message.text)
else:
data["prompt"] = message.text
data["use_async"] = "True"
res = await api.txt2img(**data)
await bot.send_media_group(
chat_id=chatId, media=pilToImages(res, "tg")
)
# Ввели любой текст
@dp.message_handler(lambda message: True)
async def change_json(message: types.Message):