AI/talk_handler.py

@@ -7,10 +7,8 @@ from aiogram.types import Message
 
 logger = logging.getLogger(__name__)
 
-# Адрес OpenAI-совместимого API у llama.cpp сервера.
 LLAMA_API_URL = os.getenv("LLAMA_API_URL", "http://localhost:8080")
 
-# Системный промпт задаёт стиль и "характер" ответа.
 SYSTEM_PROMPT = (
     "Ты — чёткий пацан с района. Базаришь строго на фене — блатной жаргон, "
     "тюремный сленг, дворовый язык. Используй слова: кент, братуха, мусора, "
@@ -25,7 +23,6 @@ SYSTEM_PROMPT = (
 
 
 async def _generate_response(user_text: str) -> str:
-    # Формируем запрос к /v1/chat/completions и возвращаем текст первого ответа.
     url = f"{LLAMA_API_URL.rstrip('/')}/v1/chat/completions"
     payload = {
         "messages": [
@@ -43,7 +40,6 @@ async def _generate_response(user_text: str) -> str:
 
 
 async def handle_talk(message: Message):
-    # Обработчик /talk: валидация ввода, запрос к LLM, отправка ответа пользователю.
     parts = message.text.split(maxsplit=1)
     if len(parts) < 2 or not parts[1].strip():
         await message.reply("Ты чё, кент? Напиши /talk [текст], побазарим.")
@@ -51,7 +47,6 @@ async def handle_talk(message: Message):
 
     user_text = parts[1].strip()
 
-    # Показываем "печатает..." пока ждём ответ модели.
     await message.bot.send_chat_action(chat_id=message.chat.id, action="typing")
 
     try:
@@ -60,6 +55,5 @@ async def handle_talk(message: Message):
             response = "Братуха, чёт базар не клеится, попробуй ещё раз."
         await message.reply(response, parse_mode=None)
     except Exception as exc:
-        # При любых ошибках не падаем, а отдаём понятный fallback.
         logger.exception("Talk generation failed")
         await message.reply("Бля, кент, чёт движок заглох. Попробуй позже.")

main.py

@@ -18,21 +18,9 @@ from PIL import Image, ImageDraw, ImageFont
 from datetime import datetime, time, timedelta
 from zoneinfo import ZoneInfo
 
-# Базовые пути/настройки проекта.
 BASE_DIR = Path(__file__).parent
 FONT_PATH = BASE_DIR / "impact.ttf"
 DEFAULT_TZ = os.getenv("TZ", "Europe/Moscow")
-
-# In-memory состояние миниигры /penis.
-# Важно: после перезапуска процесса всё обнуляется.
-PENIS_STATE: dict[int, dict[str, object]] = {}
-PENIS_START_LENGTH = 10.0
-PENIS_MIN_DELTA = 0.1
-PENIS_MAX_DELTA = 7.0
-PENIS_COOLDOWN = timedelta(hours=24)
-PENIS_TOP_LIMIT = 10
-
-# Конфигурация текстовых макетов /maket.
 TEMPLATES = {
     "1": {
         "path": BASE_DIR / "maket" / "1.png",
@@ -43,7 +31,6 @@ TEMPLATES = {
 }
 
 def templates_list_text() -> str:
-    # Собираем человекочитаемый список доступных макетов.
     parts = []
     for tid in sorted(TEMPLATES.keys()):
         label = TEMPLATES[tid].get("label", "")
@@ -56,7 +43,6 @@ def templates_list_text() -> str:
 # ---- Schedule & breaks ----
 
 def make_time(h: int, m: int) -> time:
-    # Короткий хелпер для заполнения расписания.
     return time(hour=h, minute=m)
 
 
@@ -107,7 +93,6 @@ SCHEDULE: dict[int, list[tuple[time, time, int | None]]] = {
 
 
 def minutes_until_break(now: datetime | None = None) -> str:
-    # Возвращает статус: до перекура / идёт перекур / пары окончены.
     if now is None:
         now = datetime.now(ZoneInfo(DEFAULT_TZ))
     elif now.tzinfo is None:
@@ -144,13 +129,10 @@ def minutes_until_break(now: datetime | None = None) -> str:
     return "Пары закончились, перекур только завтра."
 
 HINT_TEXT = (
-    # Общая подсказка по командам.
     "Команды:\n"
     f"• /maket «id» «текст» — доступные: {templates_list_text()} (1: 737x414, точка 104,240).\n"
     "• /gen_mem Up:верх Down:низ — отправь с фото (как подпись) или ответом на фото.\n"
     "• /break — узнать, сколько минут до перекура.\n"
-    "• /penis — миниигра: +/− от 0.1 до 7 см, раз в 24 часа.\n"
-    "• /top_penis — динамический топ по текущей длине.\n"
     "• /talk [сообщение] — побазарить с ИИ на фене.\n"
     "• /prices — старый способ цен.\n"
     "• /fetch — свежие цены с x-server.\n"
@@ -162,14 +144,12 @@ logger = logging.getLogger(__name__)
 
 
 def load_font(size: int) -> ImageFont.FreeTypeFont:
-    # Грузим Impact нужного размера.
     if not FONT_PATH.exists():
         raise FileNotFoundError("impact.ttf not found. Place the font in the project root.")
     return ImageFont.truetype(str(FONT_PATH), size=size)
 
 
 def wrap_text_to_width(text: str, font: ImageFont.FreeTypeFont, max_width: int, draw: ImageDraw.ImageDraw, stroke_width: int) -> list[str]:
-    # Переносим текст по словам так, чтобы каждая строка влезла в max_width.
     words = text.split()
     if not words:
         return []
@@ -190,7 +170,6 @@ def wrap_text_to_width(text: str, font: ImageFont.FreeTypeFont, max_width: int,
 
 
 def fit_text_in_box(text: str, max_width: int, max_height: int, start_size: int = 72, min_size: int = 14):
-    # Подбираем максимальный размер шрифта, который влезет в заданный прямоугольник.
     draw_dummy = ImageDraw.Draw(Image.new("RGB", (max_width, max_height)))
     for size in range(start_size, min_size - 1, -2):
         font = load_font(size)
@@ -205,7 +184,7 @@ def fit_text_in_box(text: str, max_width: int, max_height: int, start_size: int
         total_height = sum(line_heights) + max(0, len(lines) - 1) * max(6, size // 6)
         if total_height <= max_height:
             return font, lines, stroke_width, max(6, size // 6)
-    # Если не влезло ни на одном размере, используем минимальный.
+    # Fallback to minimal size
     font = load_font(min_size)
     stroke_width = max(1, min_size // 12)
     lines = wrap_text_to_width(text, font, max_width, draw_dummy, stroke_width)
@@ -214,7 +193,6 @@ def fit_text_in_box(text: str, max_width: int, max_height: int, start_size: int
 
 def draw_multiline(draw: ImageDraw.ImageDraw, lines: list[str], position: tuple[int, int], font: ImageFont.FreeTypeFont,
                    stroke_width: int, line_spacing: int, align_center: bool = False, max_width: int | None = None):
-    # Рисуем набор строк с опциональным центрированием по горизонтали.
     x, y = position
     for line in lines:
         bbox = draw.textbbox((0, 0), line, font=font, stroke_width=stroke_width)
@@ -227,7 +205,6 @@ def draw_multiline(draw: ImageDraw.ImageDraw, lines: list[str], position: tuple[
 
 
 def generate_maket(text: str, template_id: str = "1") -> BytesIO:
-    # Генерируем картинку макета с текстом в заранее заданной области.
     tpl = TEMPLATES.get(template_id)
     if not tpl:
         raise FileNotFoundError(f"Template {template_id} is missing")
@@ -251,7 +228,6 @@ def generate_maket(text: str, template_id: str = "1") -> BytesIO:
 
 
 def calculate_font_for_width(text: str, image_width: int, max_ratio: float = 0.09, min_size: int = 18):
-    # Подбор размера шрифта для верхнего/нижнего блока мема.
     start_size = max(int(image_width * max_ratio), min_size)
     draw_dummy = ImageDraw.Draw(Image.new("RGB", (image_width, image_width)))
     for size in range(start_size, min_size - 1, -2):
@@ -272,7 +248,6 @@ def calculate_font_for_width(text: str, image_width: int, max_ratio: float = 0.0
 
 
 def generate_meme(img: Image.Image, top_text: str, bottom_text: str) -> BytesIO:
-    # Рисуем классический мем-текст сверху/снизу и возвращаем JPEG в памяти.
     img = img.convert("RGB")
     draw = ImageDraw.Draw(img)
     width, height = img.size
@@ -302,8 +277,6 @@ def generate_meme(img: Image.Image, top_text: str, bottom_text: str) -> BytesIO:
 
 
 def parse_up_down(text: str) -> tuple[str, str]:
-    # Парсим формат /gen_mem Up:... Down:...
-    # Если меток нет, используем fallback: первое слово -> up, остальное -> down.
     up = ""
     down = ""
     # Normalize line breaks
@@ -323,7 +296,6 @@ def parse_up_down(text: str) -> tuple[str, str]:
     return up, down
 
 def minutes_until_next_lesson(now: datetime | None = None) -> str:
-    # Отдельная "угарная" логика для /zvetok.
     import random
 
     if now is None:
@@ -397,99 +369,18 @@ def minutes_until_next_lesson(now: datetime | None = None) -> str:
     return f"Твой мозг окончательно сгниет, а очко порвется через: {days} дн {hours} ч {minutes} мин"
 
 
-def play_penis(user_id: int, user_name: str | None = None, now: datetime | None = None) -> tuple[bool, str]:
-    # Миниигра /penis: изменение длины и кулдаун 24 часа от последнего броска.
-    if now is None:
-        now = datetime.now(ZoneInfo(DEFAULT_TZ))
-    elif now.tzinfo is None:
-        now = now.replace(tzinfo=ZoneInfo(DEFAULT_TZ))
-
-    state = PENIS_STATE.get(user_id)
-    if state is None:
-        state = {"length": PENIS_START_LENGTH, "last_used_at": None}
-        PENIS_STATE[user_id] = state
-
-    # Обновляем отображаемое имя для топа.
-    if user_name:
-        state["display_name"] = user_name
-
-    last_used_at = state.get("last_used_at")
-    if isinstance(last_used_at, datetime):
-        next_time = last_used_at + PENIS_COOLDOWN
-        if now < next_time:
-            # Если рано, считаем оставшееся время до следующей попытки.
-            left = next_time - now
-            total_seconds = int(left.total_seconds())
-            hours = total_seconds // 3600
-            minutes = (total_seconds % 3600) // 60
-            return False, f"Сегодня уже кидал. Попробуй через {hours} ч {minutes} мин."
-
-    current_length = float(state.get("length", PENIS_START_LENGTH))
-    delta = round(random.uniform(PENIS_MIN_DELTA, PENIS_MAX_DELTA), 1)
-    sign = random.choice([-1, 1])
-    # Не даём длине уйти в отрицательные значения.
-    new_length = round(max(0.1, current_length + sign * delta), 1)
-
-    state["length"] = new_length
-    state["last_used_at"] = now
-
-    sign_text = "+" if sign > 0 else "-"
-    return (
-        True,
-        f"📏 Изменение: {sign_text}{delta:.1f} см\n"
-        f"Текущая длина: {new_length:.1f} см\n"
-        "Следующая попытка через 24 часа.",
-    )
-
-
-def format_penis_user_name(user: types.User) -> str:
-    # Красиво подписываем пользователя для топа.
-    if user.username:
-        return f"@{user.username}"
-    if user.full_name:
-        return user.full_name
-    return f"user_{user.id}"
-
-
-def build_penis_top(limit: int = PENIS_TOP_LIMIT) -> str:
-    # Формируем динамический топ по текущим значениям /penis.
-    rows: list[tuple[float, str]] = []
-    for user_id, state in PENIS_STATE.items():
-        length = state.get("length")
-        if not isinstance(length, (int, float)):
-            continue
-        display_name = state.get("display_name")
-        if not isinstance(display_name, str) or not display_name.strip():
-            display_name = f"user_{user_id}"
-        rows.append((float(length), display_name.strip()))
-
-    if not rows:
-        return "🏆 Топ пуст. Сначала пусть кто-то крутнет /penis."
-
-    rows.sort(key=lambda item: item[0], reverse=True)
-    medals = {1: "🥇", 2: "🥈", 3: "🥉"}
-    lines = ["🏆 Топ по длине:"]
-    for index, (length, name) in enumerate(rows[:limit], start=1):
-        prefix = medals.get(index, f"{index}.")
-        lines.append(f"{prefix} {name} — {length:.1f} см")
-    return "\n".join(lines)
-    
 async def handle_start(message: Message):
-    # Базовый /start.
     await message.answer("Соси")
 
 
 async def handle_help(message: Message):
-    # Вывод подсказки по возможностям бота.
     await message.answer(HINT_TEXT)
 
 async def handle_break(message: Message):
-    # Команда расписания перекура.
     await message.answer(minutes_until_break())
 
 
 async def handle_maket(message: Message):
-    # Генерация текста в подготовленном шаблоне /maket.
     parts = message.text.split(maxsplit=2)
     if len(parts) == 1:
         await message.reply(f"Доступные макеты: {templates_list_text()}\nФормат: /maket «id» «текст»")
@@ -515,7 +406,6 @@ async def handle_maket(message: Message):
 
 
 async def handle_gen_mem(message: Message, bot: Bot):
-    # Мем можно сделать из фото в этом же сообщении или из reply на фото.
     photo_message = None
     if message.photo:
         photo_message = message
@@ -533,7 +423,6 @@ async def handle_gen_mem(message: Message, bot: Bot):
 
     photo = photo_message.photo[-1]
     try:
-        # Берём самое большое фото Telegram и скачиваем в память.
         file = await bot.get_file(photo.file_id)
         buf = BytesIO()
         await bot.download_file(file.file_path, destination=buf)
@@ -554,7 +443,6 @@ async def handle_gen_mem(message: Message, bot: Bot):
     await message.reply_photo(BufferedInputFile(output.getvalue(), filename="meme.jpg"))
 
 async def handle_size(message: Message):
-    # Простая рандом-команда /size.
     size = round(random.uniform(0.1, 35.0), 1)
 
     if size < 5:
@@ -575,28 +463,9 @@ async def handle_size(message: Message):
         f"Ты: {status}"
     )
 
-
-async def handle_penis(message: Message):
-    # Хендлер миниигры /penis.
-    if not message.from_user:
-        await message.answer("Команда доступна только пользователям.")
-        return
-
-    display_name = format_penis_user_name(message.from_user)
-    _, result_text = play_penis(message.from_user.id, display_name)
-    await message.answer(result_text)
-
-
-async def handle_top_penis(message: Message):
-    # Хендлер топа по значениям /penis.
-    await message.answer(build_penis_top())
-
-
-# Ключевые слова, на которые бот кидает трек Creep.
 KEYWORDS = {'крип', 'радиохэд', 'creep', 'radiohead'}
 # I'M CREEP
 async def handle_keywords(message: types.Message):
-    # Не реагируем на команды, только на обычный текст.
 
     if message.text and message.text.startswith('/'):
         return
@@ -615,7 +484,6 @@ async def handle_keywords(message: types.Message):
             logger.warning(f"Файл не найден по пути: {audio_path}")
 
 async def handle_prices(message: Message):
-    # Старый источник цен (/prices).
     def fetch():
         return requests.get(
             "https://bebrik.xyz/magnit/products",
@@ -649,7 +517,6 @@ async def handle_prices(message: Message):
 
 
 async def handle_fetch(message: Message):
-    # Новый источник цен (/fetch).
     def fetch():
         return requests.get(
             "https://mirror.porno4free.ru/magnit/magnit/prices/all",
@@ -687,13 +554,11 @@ async def handle_fetch(message: Message):
         await message.answer("unable to connect x-server гандон")
 
 async def handle_zvetok(message: Message):
-    # Команда с отдельной "мемной" логикой таймера.
     result = minutes_until_next_lesson()
     await message.answer(result)
 
 
 def main():
-    # Точка входа: создаём бота, регистрируем команды и запускаем polling.
     token = os.getenv("BOT_TOKEN")
     if not token:
         raise RuntimeError("BOT_TOKEN env var is required")
@@ -702,7 +567,6 @@ def main():
     dp = Dispatcher()
 
     async def on_startup(bot: Bot):
-        # Выставляем меню команд в интерфейсе Telegram.
         try:
             await bot.set_my_commands(
                 [
@@ -712,8 +576,6 @@ def main():
                     BotCommand(command="gen_mem", description="Мем из фото (ответом)"),
                     BotCommand(command="break", description="Сколько минут до перекура"),
                     BotCommand(command="size", description="хуй"),
-                    BotCommand(command="penis", description="миниигра 1 раз в 24ч"),
-                    BotCommand(command="top_penis", description="топ по длине"),
                     BotCommand(command="prices", description="цены x-server"),
                     BotCommand(command="fetch", description="цены (новый источник)"),
                     BotCommand(command="zvetok", description="Цветянский бля"),
@@ -725,14 +587,11 @@ def main():
 
     dp.startup.register(on_startup)
 
-    # Регистрация команд-обработчиков.
     dp.message.register(handle_start, CommandStart())
     dp.message.register(handle_help, Command("help"))
     dp.message.register(handle_break, Command("break"))
     dp.message.register(handle_maket, Command("maket"))
     dp.message.register(handle_size, Command("size"))
-    dp.message.register(handle_penis, Command("penis"))
-    dp.message.register(handle_top_penis, Command("top_penis"))
     dp.message.register(handle_prices, Command("prices"))
     dp.message.register(handle_fetch, Command("fetch"))
     dp.message.register(handle_zvetok, Command("zvetok"))
@@ -741,12 +600,10 @@ def main():
     
 
     async def handle_gen_mem_cmd(message: Message, bot: Bot):
-        # Обёртка под сигнатуру aiogram для /gen_mem.
         await handle_gen_mem(message, bot)
 
     dp.message.register(handle_gen_mem_cmd, Command("gen_mem"))
     
-    # Текстовые авто-реакции (без команд).
     dp.message.register(handle_keywords, F.text)
 
     logger.info("Bot is starting...")