requirements fix, gpt2 prompt helper works now!

+ actions stage renamed
This commit is contained in:
Alex Borisov 2024-02-11 14:39:27 +03:00
parent 5921e18977
commit ac16c265f2
3 changed files with 66 additions and 49 deletions

View File

@ -3,7 +3,7 @@ run-name: ${{ gitea.actor }} is building new image 🚀
on: [push] on: [push]
jobs: jobs:
Explore-Gitea-Actions: Build-Docker:
runs-on: soaska runs-on: soaska
steps: steps:
- run: echo "🎉 The job was automatically triggered by a ${{ gitea.event_name }} event." - run: echo "🎉 The job was automatically triggered by a ${{ gitea.event_name }} event."

111
bot.py
View File

@ -282,12 +282,14 @@ async def getProgress(msgTime):
# TODO aiogram.utils.exceptions.MessageToEditNotFound: Message to edit not found # TODO aiogram.utils.exceptions.MessageToEditNotFound: Message to edit not found
proc = round(api.get_progress()['progress']*100) proc = round(api.get_progress()['progress']*100)
points = '.' * (proc % 9) points = '.' * (proc % 9)
await bot.edit_message_text( try:
chat_id=msgTime.chat.id, await bot.edit_message_text(
message_id=msgTime.message_id, chat_id=msgTime.chat.id,
text=str(proc)+'% ' + points message_id=msgTime.message_id,
) text=str(proc)+'% ' + points
await asyncio.sleep(1) )
finally:
await asyncio.sleep(1)
#TODO #TODO
async def getProgress2(msgTime): async def getProgress2(msgTime):
points = '.' points = '.'
@ -365,21 +367,24 @@ def getKeyboard(keysArr, returnAll):
# Стандартное меню # Стандартное меню
async def getKeyboardUnion(txt, message, keyboard, parse_mode = 'Markdown'): async def getKeyboardUnion(txt, message, keyboard, parse_mode = 'Markdown'):
# Если команда с слешем # Если команда с слешем
if hasattr(message, "content_type"): try:
await bot.send_message( if hasattr(message, "content_type"):
chat_id=message.from_user.id, await bot.send_message(
text=txt, chat_id=message.from_user.id,
reply_markup=keyboard, text=txt,
parse_mode=parse_mode reply_markup=keyboard,
) parse_mode=parse_mode
else: )
await bot.edit_message_text( else:
chat_id=message.message.chat.id, await bot.edit_message_text(
message_id=message.message.message_id, chat_id=message.message.chat.id,
text=txt, message_id=message.message.message_id,
reply_markup=keyboard, text=txt,
parse_mode=parse_mode reply_markup=keyboard,
) parse_mode=parse_mode
)
except:
pass
def getStart(returnAll = 1) -> InlineKeyboardMarkup: def getStart(returnAll = 1) -> InlineKeyboardMarkup:
keysArr = [ keysArr = [
@ -664,12 +669,15 @@ async def inl_save_prompt(callback: types.CallbackQuery) -> None:
global data, chatHistoryPrompt global data, chatHistoryPrompt
data['prompt'] = chatHistoryPrompt data['prompt'] = chatHistoryPrompt
keyboard = InlineKeyboardMarkup(inline_keyboard=[getPromptFromJson(0), getStart(0)]) keyboard = InlineKeyboardMarkup(inline_keyboard=[getPromptFromJson(0), getStart(0)])
await bot.edit_message_text( try:
chat_id=callback.message.chat.id, await bot.edit_message_text(
message_id=callback.message.message_id, chat_id=callback.message.chat.id,
text='Промпт сохранён: ' + chatHistoryPrompt, message_id=callback.message.message_id,
reply_markup=keyboard text='Промпт сохранён: ' + chatHistoryPrompt,
) reply_markup=keyboard
)
except:
pass
# upload result.json from chat history # upload result.json from chat history
@dp.callback_query_handler(text="uplchat") @dp.callback_query_handler(text="uplchat")
@ -691,13 +699,16 @@ async def inl_uplchat(callback: types.CallbackQuery) -> None:
chatHistoryPrompt = t#translateRuToEng(t) chatHistoryPrompt = t#translateRuToEng(t)
keyboard = InlineKeyboardMarkup(inline_keyboard=[getPromptFromJson(0), getStart(0)]) keyboard = InlineKeyboardMarkup(inline_keyboard=[getPromptFromJson(0), getStart(0)])
await bot.edit_message_text( try:
chat_id=callback.message.chat.id, await bot.edit_message_text(
message_id=callback.message.message_id, chat_id=callback.message.chat.id,
text=t.replace('<', '&lt;').replace('>', '&gt;'),#translateRuToEng(t).replace('<', '&lt;').replace('>', '&gt;'), message_id=callback.message.message_id,
reply_markup=keyboard, text=t.replace('<', '&lt;').replace('>', '&gt;'),#translateRuToEng(t).replace('<', '&lt;').replace('>', '&gt;'),
parse_mode = types.ParseMode.HTML reply_markup=keyboard,
) parse_mode = types.ParseMode.HTML
)
except:
pass
# upload Lora/Model # upload Lora/Model
@ -778,7 +789,7 @@ async def inl_fp(message: Union[types.Message, types.CallbackQuery]) -> None:
data['hr_second_pass_steps'] = '10' data['hr_second_pass_steps'] = '10'
data['cfg_scale'] = '6' data['cfg_scale'] = '6'
data['width'] = '512' data['width'] = '512'
data['height'] = '768' data['height'] = '512'
data['restore_faces'] = 'false' data['restore_faces'] = 'false'
data['do_not_save_grid'] = 'true' data['do_not_save_grid'] = 'true'
data['negative_prompt'] = 'easynegative, bad-hands-5, bad-picture-chill-75v, bad-artist, bad_prompt_version2, rmadanegative4_sd15-neg, bad-image-v2-39000, illustration, painting, cartoons, sketch, (worst quality:2), (low quality:2), (normal quality:2), lowres, bad anatomy, bad hands, ((monochrome)), ((grayscale)), collapsed eyeshadow, multiple eyeblows, vaginas in breasts, (cropped), oversaturated, extra limb, missing limbs, deformed hands, long neck, long body, imperfect, (bad hands), signature, watermark, username, artist name, conjoined fingers, deformed fingers, ugly eyes, imperfect eyes, skewed eyes, unnatural face, unnatural body, error, asian, obese, tatoo, stacked torsos, totem pole, watermark, black and white, close up, cartoon, 3d, denim, (disfigured), (deformed), (poorly drawn), (extra limbs), blurry, boring, sketch, lackluster, signature, letters' data['negative_prompt'] = 'easynegative, bad-hands-5, bad-picture-chill-75v, bad-artist, bad_prompt_version2, rmadanegative4_sd15-neg, bad-image-v2-39000, illustration, painting, cartoons, sketch, (worst quality:2), (low quality:2), (normal quality:2), lowres, bad anatomy, bad hands, ((monochrome)), ((grayscale)), collapsed eyeshadow, multiple eyeblows, vaginas in breasts, (cropped), oversaturated, extra limb, missing limbs, deformed hands, long neck, long body, imperfect, (bad hands), signature, watermark, username, artist name, conjoined fingers, deformed fingers, ugly eyes, imperfect eyes, skewed eyes, unnatural face, unnatural body, error, asian, obese, tatoo, stacked torsos, totem pole, watermark, black and white, close up, cartoon, 3d, denim, (disfigured), (deformed), (poorly drawn), (extra limbs), blurry, boring, sketch, lackluster, signature, letters'
@ -892,12 +903,15 @@ async def inl_skip(message: Union[types.Message, types.CallbackQuery]) -> None:
if hasattr(message, "content_type"): if hasattr(message, "content_type"):
await message.answer("skip") await message.answer("skip")
else: else:
await bot.edit_message_text( try:
chat_id=message.message.chat.id, await bot.edit_message_text(
message_id=message.message.message_id, chat_id=message.message.chat.id,
text="Пропущено", message_id=message.message.message_id,
reply_markup=getStart(), text="Пропущено",
) reply_markup=getStart(),
)
except:
pass
@dp.message_handler(lambda message: message.from_user.id in ALLOWED_USERS or len(ALLOWED_USERS) == 0, commands=["gen"]) @dp.message_handler(lambda message: message.from_user.id in ALLOWED_USERS or len(ALLOWED_USERS) == 0, commands=["gen"])
@dp.callback_query_handler(text="gen") @dp.callback_query_handler(text="gen")
@ -1332,12 +1346,15 @@ async def inl_yes_no(callback: types.CallbackQuery) -> None:
if callback.data[1:] in dataParams.keys(): if callback.data[1:] in dataParams.keys():
dataParams[callback.data[1:]] = 'False' dataParams[callback.data[1:]] = 'False'
#await bot.delete_message(chat_id=callback.message.chat.id, message_id=callback.message.message_id) #await bot.delete_message(chat_id=callback.message.chat.id, message_id=callback.message.message_id)
await bot.edit_message_text( try:
chat_id=callback.message.chat.id, await bot.edit_message_text(
message_id=callback.message.message_id, chat_id=callback.message.chat.id,
text=f"JSON параметры:\n{getJson()}\n{getJson(1)}", message_id=callback.message.message_id,
reply_markup=keyboard, text=f"JSON параметры:\n{getJson()}\n{getJson(1)}",
) reply_markup=keyboard,
)
except:
pass
# отлов поста с канала и мгновенная генерация если включен just_gen # отлов поста с канала и мгновенная генерация если включен just_gen
@dp.channel_post_handler() @dp.channel_post_handler()

View File

@ -1,7 +1,7 @@
aiogram==2.22.2 aiogram==2.22.2
webuiapi webuiapi
translate translate
tensorflow torch
transformers transformers
vk_api vk_api
ok_api ok_api