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

config command now works fully

Browse Source
This commit is contained in:
BarsTiger
2023-03-11 23:13:55 +02:00
parent b83e649349
commit 66b6da88ad
18 changed files with 297 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
bot/keyboards/config.py
View File

@@ -1,22 +1,66 @@
from aiogram import types
from bot.db import db, DBTables
from bot.callbacks.factories.config import (prompt_settings_data, global_settings_data, admin_settings_data)
from bot.callbacks.factories.common import close_keyboard_data
from bot.callbacks.factories.config import prompt_settings_data, admin_settings_data
def get_config_keyboard(user_id: int) -> types.InlineKeyboardMarkup:
buttons = [
types.InlineKeyboardButton("Prompt settings", callback_data=prompt_settings_data.new()),
types.InlineKeyboardButton("Global settings", callback_data=global_settings_data.new())
types.InlineKeyboardButton("Prompt settings", callback_data="prompt_settings_kb"),
types.InlineKeyboardButton("Global settings", callback_data="global_settings_kb")
]
if user_id in db[DBTables.config].get('admins'):
buttons.append(
types.InlineKeyboardButton("Admin settings", callback_data=admin_settings_data.new())
types.InlineKeyboardButton("Admin settings", callback_data="admin_settings_kb")
)
buttons.append(
types.InlineKeyboardButton("🔻 Close", callback_data=close_keyboard_data.new())
types.InlineKeyboardButton("🔻 Close", callback_data="close_keyboard")
)
keyboard = types.InlineKeyboardMarkup(row_width=1)
keyboard.add(*buttons)
return keyboard
def get_prompt_settings_keyboard() -> types.InlineKeyboardMarkup:
buttons = [
types.InlineKeyboardButton("Prompt", callback_data=prompt_settings_data.new("prompt")),
types.InlineKeyboardButton("Negative prompt", callback_data=prompt_settings_data.new("negative_prompt")),
types.InlineKeyboardButton("Steps count", callback_data=prompt_settings_data.new("steps")),
types.InlineKeyboardButton("CFG Scale (model creativity)", callback_data=prompt_settings_data.new("cfg_scale")),
types.InlineKeyboardButton("Size", callback_data=prompt_settings_data.new("size")),
types.InlineKeyboardButton("Restore faces", callback_data=prompt_settings_data.new("restore_faces")),
types.InlineKeyboardButton("Sampler", callback_data=prompt_settings_data.new("sampler")),
types.InlineKeyboardButton("👈 Back", callback_data="config_back"),
types.InlineKeyboardButton("🔻 Close", callback_data="close_keyboard")
]
keyboard = types.InlineKeyboardMarkup(row_width=1)
keyboard.add(*buttons)
return keyboard
def get_global_settings_keyboard() -> types.InlineKeyboardMarkup:
buttons = [
types.InlineKeyboardButton("Set model", callback_data="global_settings_set_model"),
types.InlineKeyboardButton("👈 Back", callback_data="config_back"),
types.InlineKeyboardButton("🔻 Close", callback_data="close_keyboard")
]
keyboard = types.InlineKeyboardMarkup(row_width=1)
keyboard.add(*buttons)
return keyboard
def get_admin_settings_keyboard() -> types.InlineKeyboardMarkup:
buttons = [
types.InlineKeyboardButton("Add admin", callback_data=admin_settings_data.new("aliases.add_admin")),
types.InlineKeyboardButton("Remove admin", callback_data=admin_settings_data.new("aliases.remove_admin")),
types.InlineKeyboardButton("Set API endpoint", callback_data=admin_settings_data.new("aliases.set_endpoint")),
types.InlineKeyboardButton("Reset generation queue", callback_data=admin_settings_data.new("reset.resetqueue")),
types.InlineKeyboardButton("👈 Back", callback_data="config_back"),
types.InlineKeyboardButton("🔻 Close", callback_data="close_keyboard")
]
keyboard = types.InlineKeyboardMarkup(row_width=1)
keyboard.add(*buttons)
return keyboard

3
bot/keyboards/set_model.py
View File

@@ -1,7 +1,6 @@
from aiogram import types
from bot.db import db, DBTables
from bot.callbacks.factories.set_model import set_model_page, set_model
from bot.callbacks.factories.common import close_keyboard_data
def get_set_model_keyboard(page: int) -> types.InlineKeyboardMarkup:
@@ -30,7 +29,7 @@ def get_set_model_keyboard(page: int) -> types.InlineKeyboardMarkup:
keyboard.add(*models_buttons)
keyboard.add(
types.InlineKeyboardButton("🔻 Cancel", callback_data=close_keyboard_data.new())
types.InlineKeyboardButton("🔻 Cancel", callback_data="close_keyboard")
)
return keyboard