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

fix fast_params

Browse Source 
begin upload
This commit is contained in:
Mikhail Guseletov 2023-07-04 17:40:03 +07:00
parent 0d89767ce8
commit fcd63a75c8
1 changed files with 32 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
bot.py
View File

@ -328,6 +328,7 @@ def getOpt(returnAll = 1) -> InlineKeyboardMarkup:
def getScripts(returnAll = 1) -> InlineKeyboardMarkup:
keysArr = [
InlineKeyboardButton("get_lora", callback_data="get_lora"),
#InlineKeyboardButton("upload", callback_data="upload"),
InlineKeyboardButton("rnd_mdl", callback_data="rnd_mdl"),
InlineKeyboardButton("rnd_smp", callback_data="rnd_smp"),
InlineKeyboardButton("inf", callback_data="inf"),
@ -340,7 +341,15 @@ def getSet(returnAll = 1) -> InlineKeyboardMarkup:
keysArr = [
InlineKeyboardButton("change_param", callback_data="change_param"),
InlineKeyboardButton("reset_param", callback_data="reset_param"),
InlineKeyboardButton("fast_param", callback_data="fast_param"),
InlineKeyboardButton("fast_param", callback_data="fast_param"),
]
return (getKeyboard(keysArr, returnAll))
# Меню загрузки
def getUpl(returnAll = 1) -> InlineKeyboardMarkup:
keysArr = [
InlineKeyboardButton("uplora", callback_data="uplora"),
InlineKeyboardButton("uplmodel", callback_data="uplmodel"),
]
return (getKeyboard(keysArr, returnAll))
@ -531,6 +540,26 @@ async def inl_sd(message: Union[types.Message, types.CallbackQuery]) -> None:
"SD запущена\n" + getTxt(), reply_markup=getStart()
)
# Вызов reset_param, сброс JSON
@dp.message_handler(commands=["upload"])
@dp.callback_query_handler(text="upload")
async def inl_upload(message: Union[types.Message, types.CallbackQuery]) -> None:
print("inl_upload")
keyboard = InlineKeyboardMarkup(inline_keyboard=[getUpl(0), getSet(0), getOpt(0), getStart(0)])
await message.answer(
"Грузим LORA или Model?", reply_markup=keyboard
)
# upload Lora
@dp.message_handler(commands=["uplora"])
@dp.callback_query_handler(text="uplora")
async def inl_uplora(message: Union[types.Message, types.CallbackQuery]) -> None:
print("inl_uplora")
keyboard = InlineKeyboardMarkup(inline_keyboard=[getUpl(0), getSet(0), getOpt(0), getStart(0)])
await message.answer(
"Вставьте LORA файлом", reply_markup=keyboard
)
# Вызов reset_param, сброс JSON
@dp.message_handler(commands=["reset_param"])
@dp.callback_query_handler(text="reset_param")
@ -569,7 +598,8 @@ async def inl_fast_param(message: Union[types.Message, types.CallbackQuery]) ->
dataParams = {"img_thumb": "true",
"img_tg": "false",
"img_real": "true",
"stop_sd": "true"}
"stop_sd": "true",
"use_prompt": "true"}
keyboard = InlineKeyboardMarkup(inline_keyboard=[getSet(0), getOpt(0), getStart(0)])
txt = f"JSON сброшен\n{getJson()}\n{getJson(1)}"
await getKeyboardUnion(txt, message, keyboard, '')