diff options
Diffstat (limited to 'commands.py')
| -rw-r--r-- | commands.py | 211 |
1 files changed, 211 insertions, 0 deletions
diff --git a/commands.py b/commands.py new file mode 100644 index 0000000..c6430b9 --- /dev/null +++ b/commands.py @@ -0,0 +1,211 @@ +from io import BytesIO +from typing import Optional, Union + +from aiogram import Bot, Dispatcher, F, Router +from aiogram.client.default import DefaultBotProperties +from aiogram.enums import ChatAction, ChatType +from aiogram.exceptions import TelegramForbiddenError +from aiogram.filters import Command +from aiogram.types import ( + BufferedInputFile, + CallbackQuery, + ChosenInlineResult, + InlineKeyboardButton, + InlineKeyboardMarkup, + InlineQuery, + InlineQueryResultArticle, + InputTextMessageContent, + Message, + User, +) +from aiogram.utils.deep_linking import create_start_link + +from colors import * +from config import * +from supergenerator import * + +router = Router() + + +@router.message(Command("broadcast")) +async def broadcast_handler(message: Message): + if message.from_user.id != ADMIN_ID: + return await message.reply("🚫 pashol naxxuy.") + + text = message.text.removeprefix("/broadcast").strip() + if not text: + return await message.reply("🚫 напиши текст после команды ало.") + + users = get_all_user_ids() + + success = 0 + failed = 0 + + for user_id in users: + try: + if user_id > 0: + await message.bot.send_message(user_id, text) + success += 1 + except TelegramForbiddenError: + failed += 1 + except Exception as e: + print(f"{RED} -- broadcast err - {e}{RESET}") + failed += 1 + + await message.reply( + f"✅ *готово врат.*\n- успешно: {success}\n- не удалось: {failed}" + ) + + +@router.message(Command("start")) +async def start_handler(message: Message): + if message.chat.type in [ChatType.GROUP, ChatType.SUPERGROUP]: + await message.reply( + "привет, запусти эту команду в ЛиС (личные сообщения) бота✅" + ) + return + + if message.sender_chat: + user_id = message.sender_chat.id + else: + user_id = message.from_user.id + + invite_url = f"https://t.me/{BOT_USERNAME}?startgroup=start" + + keyboard = InlineKeyboardMarkup( + inline_keyboard=[ + [InlineKeyboardButton(text="команды", callback_data="start")], + [ + InlineKeyboardButton( + text="настройки", callback_data=f"settings%{user_id}" + ) + ], + [InlineKeyboardButton(text="добавить в группу", url=invite_url)], + ] + ) + + try: + await message.reply( + "Привет, я УзбекГПТ✅ национальный бот Узбикестан и я готов не помочь. Напиши любае сообщение или отправь файл чтоб я быстра ответил!1!1 Или нажми кнопку ниже чтоба посмотрет мои команды..", + reply_markup=keyboard, + ) + except Exception as e: + print(f"{RED} -- {e}{RESET}") + + +@router.message(Command("prompt")) +async def prompt_handler(message: Message): + if PROMPT_COMMAND: + await message.reply( + f"🤝 *текущий системный промпт узбекГПТ*: ```prompt\n{SYSTEM_PROMPT}```*для фото*: ```prompt\n{IMAGE_PROMPT}```изменить системный промпт узбекгпт пока что нельзя." + ) + else: + if SHOW_ACC_MSG: + await message.reply( + "🚫 *данный команда отключен в этом боте.* попросите владельца включить🤝" + ) + + +@router.message(Command("config")) +async def config_handler(message: Message): + if CONFIG_COMMAND: + await message.reply( + f"🤝 *конфиг узбекгпт*:\n`MAX_CONTEXT` = `{MAX_CONTEXT}`\n`MAX_PROMPT` = `{MAX_PROMPT}`\n" + f"`DEFAULT_MODEL` = `{DEFAULT_MODEL}`\n`IMAGE_MODEL` = `{IMAGE_MODEL}`\n" + f"`DEFAULT_STREAM` = `{DEFAULT_STREAM}`\n" + "если нужно что-то изменить здесь, то обратитесь к владельцу бота🥺" + ) + else: + if SHOW_ACC_MSG: + await message.reply( + "🚫 *данный команда отключен в этом боте.* попросите владельца включить🤝" + ) + + +async def settings_handler(event: Union[Message, CallbackQuery]): + user_id = ( + event.sender_chat.id + if isinstance(event, Message) and event.sender_chat + else event.from_user.id + ) + + config = get_user_config(user_id) + models_count = sum(len(models) for models in MODELS.values()) + + if config["stream"] == "native": + stream_status = "натив" + stream_cb = "edit" + elif config["stream"] == "edit": + stream_status = "редакт" + stream_cb = "none" + else: + stream_status = "net" + stream_cb = "native" + + if config["call"] == "da": + call_status = "da" + call_cb = "net" + else: + call_status = "net" + call_cb = "da" + + keyboard = InlineKeyboardMarkup( + inline_keyboard=[ + [ + InlineKeyboardButton( + text=f"модели ({models_count})", callback_data=f"models%{user_id}" + ) + ], + [ + InlineKeyboardButton( + text=f"стрим: {stream_status}", + callback_data=f"set_stream%{user_id}%{stream_cb}", + ) + ], + [ + InlineKeyboardButton( + text=f"отклик: {call_status}", + callback_data=f"set_call%{user_id}%{call_cb}", + ) + ], + [InlineKeyboardButton(text="о боте", callback_data=f"about%{user_id}")], + ] + ) + + text = f"⚡ *алло узбекгпт настройки*\nтекущая модель: `{config.get('model')}`" + + if isinstance(event, Message): + await event.reply(text, reply_markup=keyboard) + else: + await event.message.edit_text(text, reply_markup=keyboard) + await event.answer() + + +@router.message(Command("settings")) +async def settings_command(message: Message): + if not SETTINGS_COMMAND: + if SHOW_ACC_MSG: + await message.reply( + "🚫 *данный команда отключен в этом боте.* попросите владельца включить🤝" + ) + return + await settings_handler(message) + + +@router.message(Command("clear")) +async def clear_handler(message: Message): + if not CLEAR_COMMAND: + if SHOW_ACC_MSG: + await message.reply( + "🚫 *данный команда отключен в этом боте.* попросите владельца включить🤝" + ) + return + user_id = message.sender_chat.id if message.sender_chat else message.from_user.id + + user_contexts[user_id] = [] + try: + await message.reply("контекст очищен. ✅") + except TelegramForbiddenError as e: + user_contexts[user_id] = [] + except Exception as e: + print(f"{RED} -- {e}{RESET}") |