Added config and current commands, Close button for inline menus, pass to all __init__s
This commit is contained in:
@@ -2,6 +2,7 @@ from bot.common import dp
|
||||
from .txt2img import generate_command
|
||||
from .set_model import set_model_command
|
||||
from .status import get_status
|
||||
from .current import get_current
|
||||
from .set_settings import (
|
||||
set_height_command, set_negative_prompt_command, set_size_command, set_steps_command, set_width_command,
|
||||
set_prompt_command, set_sampler_command, set_cfg_scale_command, set_restore_faces_command
|
||||
@@ -21,3 +22,4 @@ def register():
|
||||
dp.register_message_handler(set_settings.set_restore_faces_command, commands='setfaces')
|
||||
dp.register_message_handler(set_model.set_model_command, commands='setmodel')
|
||||
dp.register_message_handler(status.get_status, commands='status')
|
||||
dp.register_message_handler(current.get_current, commands='current')
|
||||
|
||||
24
bot/handlers/txt2img/current.py
Normal file
24
bot/handlers/txt2img/current.py
Normal file
@@ -0,0 +1,24 @@
|
||||
from aiogram import types
|
||||
from bot.db import db, DBTables
|
||||
from bot.utils.cooldown import throttle
|
||||
from bot.modules.api.objects.prompt_request import Prompt
|
||||
from bot.utils.errorable_command import wrap_exception
|
||||
|
||||
|
||||
@wrap_exception()
|
||||
@throttle(cooldown=5, admin_ids=db[DBTables.config].get('admins'))
|
||||
async def get_current(message: types.Message):
|
||||
prompt: Prompt = db[DBTables.prompts].get(message.from_id)
|
||||
if prompt is None:
|
||||
await message.reply('❌ No prompts found for you')
|
||||
|
||||
await message.reply(
|
||||
f"🖤 Prompt: {prompt.prompt} \n"
|
||||
f"🐊 Negative: {prompt.negative_prompt} \n"
|
||||
f"🪜 Steps: {prompt.steps} \n"
|
||||
f"🧑🎨 CFG Scale: {prompt.cfg_scale} \n"
|
||||
f"🖥️ Size: {prompt.width}x{prompt.height} \n"
|
||||
f"😀 Restore faces: {'on' if prompt.restore_faces else 'off'} \n"
|
||||
f"⚒️ Sampler: {prompt.sampler} \n",
|
||||
parse_mode='html'
|
||||
)
|
||||
Reference in New Issue
Block a user