Базовая обнова, запрет на комментирование фоток /ebalnik ну и поменял ai чтобы пиздел меньше, а не километровыми пастами

Reviewed-on: zovos/bot_tg#20

AI/talk_handler.py

@@ -2,6 +2,7 @@ import asyncio
 import os
 import logging
 import sqlite3
+import re
 from pathlib import Path
 
 import aiohttp
@@ -69,6 +70,8 @@ _CONTEXT_SIZE = 20
 _MAX_PROMPT_CHARS = 8_000
 # Путь к SQLite базе (берётся из переменной окружения или из config)
 _DB_PATH = os.getenv("CHAT_HISTORY_DB_PATH", "/db/chat_history.sqlite3")
+_MAX_REPLY_SENTENCES = 3
+_MAX_REPLY_CHARS = 280
 
 
 def _init_db() -> None:
@@ -139,20 +142,49 @@ def _build_messages(chat_id: int, system_prompt: str, current_text: str, current
     return messages
 
 
+def _shorten_reply(text: str, max_sentences: int = _MAX_REPLY_SENTENCES, max_chars: int = _MAX_REPLY_CHARS) -> str:
+    """Ограничивает ответ до 1-3 предложений и разумной длины."""
+    cleaned = " ".join((text or "").replace("\n", " ").split()).strip()
+    if not cleaned:
+        return "Братуха, коротко: чёт пусто получилось."
+
+    # Делим по окончаниям предложений, сохраняя знаки.
+    parts = re.split(r"(?<=[.!?…])\s+", cleaned)
+    sentences = [p.strip() for p in parts if p.strip()]
+
+    if not sentences:
+        # На случай ответа без пунктуации.
+        words = cleaned.split()
+        return " ".join(words[:25]).strip()
+
+    short = " ".join(sentences[:max_sentences]).strip()
+
+    # Дополнительный предохранитель по символам.
+    if len(short) > max_chars:
+        clipped = short[:max_chars].rstrip()
+        last_break = max(clipped.rfind("."), clipped.rfind("!"), clipped.rfind("?"), clipped.rfind("…"))
+        if last_break >= 40:
+            clipped = clipped[: last_break + 1].rstrip()
+        short = clipped
+
+    return short
+
+
 async def _generate_response(chat_id: int, user_text: str, user_name: str) -> str:
     # Формируем запрос к /v1/chat/completions с историей чата и возвращаем текст первого ответа.
     url = f"{LLAMA_API_URL.rstrip('/')}/v1/chat/completions"
     msgs = await asyncio.to_thread(_build_messages, chat_id, SYSTEM_PROMPT, user_text, user_name)
     payload = {
         "messages": msgs,
-        "max_tokens": 200,
+        "max_tokens": 90,
         "temperature": 0.8,
         "top_p": 0.9,
     }
     async with aiohttp.ClientSession() as session:
         async with session.post(url, json=payload, timeout=aiohttp.ClientTimeout(total=120)) as resp:
             data = await resp.json()
-            return data["choices"][0]["message"]["content"].strip()
+            content = data["choices"][0]["message"]["content"].strip()
+            return _shorten_reply(content)
 
 
 async def handle_talk(message: Message):
@@ -230,14 +262,15 @@ async def generate_autoreply(chat_id: int, text: str, user_name: str) -> str:
     msgs = await asyncio.to_thread(_build_messages, chat_id, AUTOREPLY_SYSTEM_PROMPT, text, user_name)
     payload = {
         "messages": msgs,
-        "max_tokens": 150,
+        "max_tokens": 70,
         "temperature": 0.9,
         "top_p": 0.9,
     }
     async with aiohttp.ClientSession() as session:
         async with session.post(url, json=payload, timeout=aiohttp.ClientTimeout(total=120)) as resp:
             data = await resp.json()
-            return data["choices"][0]["message"]["content"].strip()
+            content = data["choices"][0]["message"]["content"].strip()
+            return _shorten_reply(content)
 
 
 # Инициализация БД при импорте модуля

games/nude.py

@@ -3,13 +3,14 @@ from aiogram import types
 import config
 import logging
 import random
-import json
+import re
 
 logger = logging.getLogger(__name__)
 
 CAPTIONS = [
     "Hot stuff 🔥",
-    "Sexy content 😏"
+    "Sexy content 😏",
+    "ШПАЧИХА"
 ]
 
 async def handle_nude_cmd(message: types.Message):
@@ -19,66 +20,90 @@ async def handle_nude_cmd(message: types.Message):
         help_msg = (
             "🔥 <b>Справка по команде /nude:</b>\n\n"
             "Получает случайные изображения взрослых женщин из открытых источников.\n"
-            "• Использует Pornhub API для поиска контента\n"
-            "• По умолчанию ищет популярные видео с высоким рейтингом\n"
-            "• Отправляет превью видео как изображение\n\n"
+            "• Использует прямое сканирование сайтов для поиска контента\n"
+            "• По умолчанию ищет популярные изображения\n"
+            "• Отправляет изображения напрямую\n\n"
             "<i>Пример:</i> <code>/nude</code>\n"
-            "<i>По умолчанию:</i> Популярный контент для взрослых"
+            "<i>По умолчанию:</i> Случайные изображения с популярных сайтов"
         )
         await message.reply(help_msg, parse_mode="HTML")
         return
 
-    # Используем Pornhub API для поиска видео
-    url = "https://www.pornhub.com/webmasters/search"
+    # Список сайтов для сканирования
+    sites = [
+        "https://www.imagefap.com/gallery.php?pg=random",
+        "https://xhamster.com/photos?sort=random",
+        "https://spankbang.com/random"
+    ]
     
-    params = {
-        "category": "straight",
-        "ordering": "mostviewed",
-        "period": "week",
-        "thumbsize": "medium"
-    }
+    random_site = random.choice(sites)
     
     headers = {
-        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+        "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
+        "Accept-Language": "en-US,en;q=0.5",
+        "Accept-Encoding": "gzip, deflate",
+        "Connection": "keep-alive",
+        "Upgrade-Insecure-Requests": "1"
     }
     
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.get(url, params=params, headers=headers) as resp:
+            async with session.get(random_site, headers=headers) as resp:
                 if resp.status != 200:
-                    logger.error(f"Pornhub API Error: {resp.status}")
-                    await message.reply("Не удалось получить контент от сервера :(")
+                    logger.error(f"Site access error: {resp.status}")
+                    await message.reply("Не удалось получить доступ к сайту :(")
                     return
                 
-                # Pornhub API возвращает HTML, нужно парсить
-                text = await resp.text()
+                html = await resp.text()
                 
-                # Ищем URL превью в HTML
-                import re
-                thumb_pattern = r'<img[^>]+src="([^"]+)"[^>]+class="thumb"'
-                matches = re.findall(thumb_pattern, text)
+                # Ищем URL изображений в HTML
+                img_patterns = [
+                    r'<img[^>]+src="([^"]+)"',
+                    r'<a[^>]+href="([^"]+\.(jpg|jpeg|png|gif|webp))"',
+                    r'data-src="([^"]+)"'
+                ]
                 
-                if not matches:
-                    await message.reply("Не удалось найти контент :(")
+                img_urls = []
+                for pattern in img_patterns:
+                    matches = re.findall(pattern, html, re.IGNORECASE)
+                    img_urls.extend(matches)
+                
+                if not img_urls:
+                    await message.reply("Изображения не найдены на странице :(")
                     return
                 
-                # Берем случайное превью
-                thumb_url = random.choice(matches)
+                # Фильтруем и выбираем случайное изображение
+                valid_imgs = []
+                for img_url in img_urls:
+                    if img_url.startswith('//'):
+                        img_url = 'https:' + img_url
+                    elif img_url.startswith('/'):
+                        if 'imagefap.com' in random_site:
+                            img_url = 'https://www.imagefap.com' + img_url
+                        elif 'xhamster.com' in random_site:
+                            img_url = 'https://xhamster.com' + img_url
+                        elif 'spankbang.com' in random_site:
+                            img_url = 'https://spankbang.com' + img_url
                     
-                # Убедимся что URL полный
-                if not thumb_url.startswith('http'):
-                    thumb_url = f"https://thumb.pornhub.com{thumb_url}"
+                    if any(ext in img_url.lower() for ext in ['.jpg', '.jpeg', '.png', '.gif', '.webp']):
+                        valid_imgs.append(img_url)
                 
+                if not valid_imgs:
+                    await message.reply("Подходящие изображения не найдены :(")
+                    return
+                
+                final_img_url = random.choice(valid_imgs)
                 caption = random.choice(CAPTIONS)
                 
-        # Отправляем как фото
+                # Отправляем изображение
                 try:
-            input_file = types.URLInputFile(thumb_url)
+                    input_file = types.URLInputFile(final_img_url)
                     await message.answer_photo(photo=input_file, caption=caption)
                 except Exception as e:
                     logger.error(f"Error sending photo: {e}")
-            await message.answer(f"{caption}\n{thumb_url}")
+                    await message.answer(f"{caption}\n{final_img_url}")
         
     except Exception as e:
         logger.exception("Error in nude command")
-        await message.reply("Произошла ошибка при обращении к API :(")
+        await message.reply("Произошла ошибка при обработке запроса :(")

main.py

@@ -1145,6 +1145,10 @@ async def handle_keywords(message: Message):
     chat_id = message.chat.id
     user_name = (message.from_user.first_name or "кент") if message.from_user else "кент"
 
+    # /ebalnik должен глушить все авто-реакции в чате, включая медиа/фото-ветки.
+    if chat_id in _autoreply_disabled_chats:
+        return
+
     if any(re.search(rf'\b{word}\b', text_lower) for word in config.KEYWORDS):
         audio_path = config.BASE_DIR / "maket" / "Radiohead - Creep.mp3"
         if audio_path.exists():
@@ -1171,9 +1175,6 @@ async def handle_keywords(message: Message):
     # Записываем каждое обычное сообщение в историю чата
     await asyncio.to_thread(push_message, chat_id, "user", user_name, message.text)
 
-    if chat_id in _autoreply_disabled_chats:
-        return
-
     import time as _time
     now = _time.time()
     triggered = any(trigger in text_lower for trigger in config.AUTOREPLY_TRIGGERS)