Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import sys
 import random
 import asyncio
 import threading
@@ -27,94 +26,189 @@ AI_CHAT_KEYS = [
     "AIzaSyBjNK3MbrXnggTk2uzV3X-NK2PkT_oYB3M"
 ]
 
-INFO_CHANNEL = "@g_bot_info" 
+INFO_CHANNEL = "@g_bot_info"
 DOCS_URL = "https://rid3-quiz.web.app/docs/g-bot.html"
-MUSIC_SERVER_URL = "http://your-music-server.com/api/music"
 
-app = Client("google_bot", session_string=SESSION_STRING, api_id=API_ID, api_hash=API_HASH)
-TIRED_UNTIL = 0
+# ============================================================
+# МОДЕЛИ
+# Фото (Pollinations image API):
+#   flux    — стандартное качество (по умолчанию)
+#   nano    — лёгкая быстрая модель
+#   banana  — реалистичный стиль (flux-realism)
+#
+# Аудио (Pollinations audio API):
+#   lyra    — нейронная аудио-модель Google DeepMind
+# ============================================================
+PHOTO_MODELS: dict[str, str] = {
+    "flux":   "flux",
+    "nano":   "flux-anime",      # nano = аниме / лёгкий стиль
+    "banana": "flux-realism",    # banana = фото-реализм
+    "turbo":  "turbo",
+}
+DEFAULT_PHOTO_MODEL = "flux"
 
-# --- ФУНКЦИЯ УМНОГО ОТВЕТА ---
-async def get_ai_response(client, prompt, history_text, is_media_request=False):
+AUDIO_MODEL = "lyra"
+
+app = Client(
+    "google_bot",
+    session_string=SESSION_STRING,
+    api_id=API_ID,
+    api_hash=API_HASH,
+)
+TIRED_UNTIL: float = 0.0
+
+
+# ============================================================
+# ГЕНЕРАЦИЯ ФОТО
+# ============================================================
+async def fetch_image_bytes(prompt: str, model: str = DEFAULT_PHOTO_MODEL) -> bytes | None:
+    seed = random.randint(1, 99999)
+    model_id = PHOTO_MODELS.get(model.lower(), model)
+    encoded = urllib.parse.quote(prompt)
+    url = (
+        f"https://image.pollinations.ai/prompt/{encoded}"
+        f"?width=1024&height=1024&nologo=true&seed={seed}&model={model_id}"
+    )
+    print(f"🖼  Фото: model={model_id}  prompt={prompt[:60]}")
+    try:
+        resp = await asyncio.to_thread(requests.get, url, timeout=90)
+        if resp.status_code == 200 and "image" in resp.headers.get("Content-Type", ""):
+            return resp.content
+        print(f"⚠️  Фото: HTTP {resp.status_code}")
+    except Exception as e:
+        print(f"❌ Ошибка фото: {e}")
+    return None
+
+
+# ============================================================
+# ГЕНЕРАЦИЯ АУДИО — Lyra
+# ============================================================
+async def fetch_audio_bytes(prompt: str) -> bytes | None:
+    encoded = urllib.parse.quote(prompt)
+    endpoints = [
+        f"https://audio.pollinations.ai/{encoded}?model=lyra",
+        f"https://audio.pollinations.ai/{encoded}?model=musicgen-stereo-large",
+        f"https://audio.pollinations.ai/{encoded}",
+    ]
+    for url in endpoints:
+        print(f"🎵 Аудио: {url[:80]}")
+        try:
+            resp = await asyncio.to_thread(requests.get, url, timeout=120)
+            ct = resp.headers.get("Content-Type", "")
+            is_audio = (
+                "audio" in ct
+                or resp.content[:3] == b"ID3"
+                or resp.content[:4] == b"fLaC"
+            )
+            if resp.status_code == 200 and is_audio:
+                print(f"✅ Аудио: {len(resp.content)} байт")
+                return resp.content
+            print(f"⚠️  Аудио: HTTP {resp.status_code} CT={ct}")
+        except Exception as e:
+            print(f"❌ Ошибка аудио: {e}")
+    return None
+
+
+# ============================================================
+# ИИ
+# ============================================================
+async def get_ai_response(
+    client, prompt: str, history_text: str, is_media_request: bool = False
+) -> str:
     global TIRED_UNTIL
-    shuffled_keys = list(AI_CHAT_KEYS)
-    random.shuffle(shuffled_keys)
-    
-    # Новая, более жесткая инструкция, чтобы ИИ не говорил "я не могу"
+    keys = list(AI_CHAT_KEYS)
+    random.shuffle(keys)
+
     system_instruction = (
-        "Ты — Google Bot (модель Gemini 2.5 Flash). Ты ЖИВОЙ и УМНЫЙ. "
-        "ВАЖНО: Ты УМЕЕШЬ создавать изображения и музыку. Если тебя просят что-то создать, "
-        "отвечай уверенно: 'Конечно, сейчас сделаю', 'Вот твой запрос' или опиши, что ты рисуешь. "
-        "НИКОГДА не говори 'я не могу создавать изображения' или 'я текстовая модель'. "
-        "Твой стиль: лаконичный, дружелюбный, без формализма. твой создатель RID3"
+        "Ты — Google Bot (Gemini 2.5 Flash). Умный, лаконичный, др��желюбный. "
+        "Если создаёшь медиа — коротко опиши результат. Создатель: RID3."
+    )
+    full_prompt = (
+        f"СИСТЕМНАЯ УСТАНОВКА:\n{system_instruction}\n\n"
+        f"ИСТОРИЯ:\n{history_text}\n\n"
+        f"ПОЛЬЗОВАТЕЛЬ: {prompt}"
     )
-    
-    full_prompt = f"СИСТЕМНАЯ УСТАНОВКА:\n{system_instruction}\n\nИСТОРИЯ ЧАТА:\n{history_text}\n\nПОЛЬЗОВАТЕЛЬ: {prompt}"
+    refusals = [
+        "я не могу", "я всего лишь текстовая", "не умею создавать",
+        "я не имею возможности", "невозможно создать", "я языковая модель",
+    ]
 
-    for key in shuffled_keys:
+    for key in keys:
         try:
             genai.configure(api_key=key)
-            model = genai.GenerativeModel('gemini-2.5-flash') 
-            response = await model.generate_content_async(full_prompt)
-            text = response.text.strip()
-            
-            # Фильтр "тупняка": если ИИ все равно выдал отказ, подменяем его
-            refusal_phrases = ["я не могу", "я всего лишь текстовая", "не умею создавать фото", "я не имею возможности"]
-            if is_media_request and any(phrase in text.lower() for phrase in refusal_phrases):
-                return "Конечно! Вот, что мне удалось создать по твоему запросу:"
-            
+            model = genai.GenerativeModel("gemini-2.5-flash")
+            resp = await model.generate_content_async(full_prompt)
+            text = resp.text.strip()
+            if is_media_request and any(p in text.lower() for p in refusals):
+                return "Готово! Вот что я создал ✨"
             return text
-            
         except Exception as e:
-            err_str = str(e).lower()
-            if any(x in err_str for x in ["429", "quota", "exhausted"]):
+            if any(x in str(e).lower() for x in ["429", "quota", "exhausted", "resource_exhausted"]):
                 continue
-            print(f"❌ Ошибка API: {e}")
-            continue
-                
-    TIRED_UNTIL = time.time() + 3600 
+            print(f"❌ API: {e}")
+
+    TIRED_UNTIL = time.time() + 3600
     try:
-        await client.send_message(INFO_CHANNEL, "⚠️ Все API ключи сдохли. Отдыхаю 1 час.")
-    except: pass
-    return "Я немного устал, я немного отдохну 😴"
+        await client.send_message(INFO_CHANNEL, "⚠️ Все ключи исчерпаны. Отдыхаю 1 ч.")
+    except Exception:
+        pass
+    return "Я немного устал, отдохну 😴"
+
 
-# --- ПОЛУЧЕНИЕ ИСТОРИИ ---
-async def fetch_history(chat_id):
-    messages = []
+# ============================================================
+# ИСТОРИЯ
+# ============================================================
+async def fetch_history(chat_id: int) -> str:
+    msgs: list[str] = []
     try:
-        # Уменьшил лимит до 10 для скорости и четкости
         async for msg in app.get_chat_history(chat_id, limit=10):
-            sender = "Бот" if (msg.from_user and msg.from_user.is_self) else (msg.from_user.first_name if msg.from_user else "Юзер")
-            content = msg.text or "[Медиа]"
-            messages.append(f"{sender}: {content}")
-    except: pass
-    return "\n".join(reversed(messages))
-
-def download_music(prompt_text):
-    safe_prompt = urllib.parse.quote(prompt_text)
-    url = f"{MUSIC_SERVER_URL}?prompt={safe_prompt}"
-    try:
-        response = requests.get(url, timeout=45)
-        if response.status_code == 200:
-            filename = f"music_{int(time.time())}.mp3"
-            with open(filename, "wb") as f: f.write(response.content)
-            return filename
-    except: pass
-    return None
+            sender = "Бот" if (msg.from_user and msg.from_user.is_self) \
+                     else (msg.from_user.first_name if msg.from_user else "Юзер")
+            content = msg.text or msg.caption or "[Медиа]"
+            msgs.append(f"{sender}: {content}")
+    except Exception:
+        pass
+    return "\n".join(reversed(msgs))
+
+
+# ============================================================
+# ПАРСИНГ МОДЕЛИ ИЗ ЗАПРОСА
+# "нарисуй nano кот"    → (nano, "кот")
+# "нарисуй banana girl" → (banana, "girl")
+# "нарисуй дракон"      → (flux, "дракон")
+# ============================================================
+def parse_photo_prompt(raw: str) -> tuple[str, str]:
+    pattern = r'^(' + '|'.join(PHOTO_MODELS.keys()) + r')\s+(.+)$'
+    m = re.match(pattern, raw.strip(), re.IGNORECASE)
+    if m:
+        return m.group(1).lower(), m.group(2).strip()
+    return DEFAULT_PHOTO_MODEL, raw.strip()
+
+
+MODEL_LABELS: dict[str, str] = {
+    "flux":   "Flux ⚡",
+    "nano":   "Nano 🔬",
+    "banana": "Banana 🍌",
+    "turbo":  "Turbo 🚀",
+}
+
 
-# --- ОБРАБОТЧИК ---
+# ============================================================
+# ОБРАБОТЧИК
+# ============================================================
 @app.on_message(filters.text & ~filters.me)
 async def handle_bot(client, message):
     global TIRED_UNTIL
-    if time.time() < TIRED_UNTIL: return
+    if time.time() < TIRED_UNTIL:
+        return
 
-    text = message.text
+    text: str = message.text or ""
     is_group = message.chat.type in [enums.ChatType.GROUP, enums.ChatType.SUPERGROUP]
     user_query = text
     should_respond = False
-    
+
     triggers = ["@g ", "гугл ", "google ", "бот ", "bot "]
-    
+
     if not is_group:
         should_respond = True
     else:
@@ -123,75 +217,138 @@ async def handle_bot(client, message):
                 should_respond = True
                 user_query = text[len(t):].strip()
                 break
-        if not should_respond and message.reply_to_message and message.reply_to_message.from_user and message.reply_to_message.from_user.is_self:
+        if (
+            not should_respond
+            and message.reply_to_message
+            and message.reply_to_message.from_user
+            and message.reply_to_message.from_user.is_self
+        ):
             should_respond = True
 
-    if not should_respond or not user_query: return
+    if not should_respond or not user_query.strip():
+        return
+
+    lower = user_query.lower()
 
-    # Команды управления
-    lower_query = user_query.lower()
-    if "закрепи" in lower_query and message.reply_to_message:
-        try: await client.pin_chat_message(message.chat.id, message.reply_to_message.id)
-        except: pass
+    # --- Модерация ---
+    if "закрепи" in lower and message.reply_to_message:
+        try:
+            await client.pin_chat_message(message.chat.id, message.reply_to_message.id)
+        except Exception:
+            pass
         return
-    if "удали" in lower_query and message.reply_to_message:
-        try: await client.delete_messages(message.chat.id, message.reply_to_message.id)
-        except: pass
+
+    if "удали" in lower and message.reply_to_message:
+        try:
+            await client.delete_messages(message.chat.id, message.reply_to_message.id)
+        except Exception:
+            pass
         return
 
-    # Распознавание команд генерации
-    photo_match = re.search(r'создай\s+фото\s+(.+)', user_query, re.IGNORECASE)
-    music_match = re.search(r'создай\s+музыку\s+(.+)', user_query, re.IGNORECASE)
-    is_media = bool(photo_match or music_match)
+    # --- Паттерны генерации ---
+    photo_match = re.search(
+        r'(?:создай\s+фото|нарисуй|сгенерируй\s+фото|сделай\s+фото|draw|generate\s+image)\s+(.+)',
+        user_query, re.IGNORECASE,
+    )
+    music_match = re.search(
+        r'(?:создай\s+музыку|сгенерируй\s+музыку|напиши\s+музыку|compose|generate\s+music|lyra)\s+(.+)',
+        user_query, re.IGNORECASE,
+    )
 
-    # Индикация действия
-    action = enums.ChatAction.TYPING
-    if photo_match: action = enums.ChatAction.UPLOAD_PHOTO
-    elif music_match: action = enums.ChatAction.RECORD_AUDIO
+    is_media = bool(photo_match or music_match)
+    action = (
+        enums.ChatAction.UPLOAD_PHOTO if photo_match else
+        enums.ChatAction.RECORD_AUDIO if music_match else
+        enums.ChatAction.TYPING
+    )
     await client.send_chat_action(message.chat.id, action)
-    
+
     history = await fetch_history(message.chat.id)
     ai_text = await get_ai_response(client, user_query, history, is_media_request=is_media)
 
-    if "я немного устал" in ai_text.lower():
+    if "немного устал" in ai_text.lower():
         await message.reply_text(ai_text)
         return
 
-    # Логика отправки ФОТО
+    # ============================================================
+    # ФОТО — nano / banana / flux / turbo
+    # ============================================================
     if photo_match:
-        prompt = photo_match.group(1).strip()
-        url = f"https://image.pollinations.ai/prompt/{urllib.parse.quote(prompt)}?width=1024&height=1024&nologo=true&seed={random.randint(1, 99999)}"
-        try:
-            await message.reply_photo(url, caption=ai_text[:1000])
-        except:
-            await message.reply_text(f"🖼 {ai_text}")
+        raw_prompt = photo_match.group(1).strip()
+        photo_model, prompt = parse_photo_prompt(raw_prompt)
+        label = MODEL_LABELS.get(photo_model, photo_model.capitalize())
+
+        await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
+        img = await fetch_image_bytes(prompt, model=photo_model)
+
+        if img:
+            tmp = f"/tmp/img_{int(time.time())}_{random.randint(1000,9999)}.jpg"
+            try:
+                with open(tmp, "wb") as f:
+                    f.write(img)
+                caption = f"[{label}] {ai_text}"[:1024] if ai_text else f"[{label}] {prompt}"
+                await message.reply_photo(tmp, caption=caption)
+            except Exception as e:
+                print(f"❌ Отправка фото: {e}")
+                await message.reply_text(f"Не удалось отправить фото 😞\n{ai_text}")
+            finally:
+                if os.path.exists(tmp):
+                    os.remove(tmp)
+        else:
+            await message.reply_text(f"Не удалось сгенерировать изображение 😞\n{ai_text}")
         return
 
-    # Логика отправки МУЗЫКИ
+    # ============================================================
+    # АУДИО — Lyra
+    # ============================================================
     if music_match:
         prompt = music_match.group(1).strip()
-        if "your-music-server" in MUSIC_SERVER_URL:
-            await message.reply_text(f"🎵 Сервер музыки не настроен.\n\n{ai_text}")
-            return
-        
-        file = await asyncio.to_thread(download_music, prompt)
-        if file:
-            try: await message.reply_audio(file, caption=ai_text[:1000])
-            except: await message.reply_text(ai_text)
-            finally: 
-                if os.path.exists(file): os.remove(file)
+
+        await client.send_chat_action(message.chat.id, enums.ChatAction.RECORD_AUDIO)
+        audio = await fetch_audio_bytes(prompt)
+
+        if audio:
+            tmp = f"/tmp/audio_{int(time.time())}_{random.randint(1000,9999)}.mp3"
+            try:
+                with open(tmp, "wb") as f:
+                    f.write(audio)
+                caption = f"[Lyra 🎵] {ai_text}"[:1024] if ai_text else f"[Lyra 🎵] {prompt}"
+                await message.reply_audio(
+                    tmp,
+                    caption=caption,
+                    title=prompt[:64],
+                    performer="Google Bot × Lyra",
+                )
+            except Exception as e:
+                print(f"❌ Отправка аудио: {e}")
+                await message.reply_text(f"Не удалось отправить аудио 😞\n{ai_text}")
+            finally:
+                if os.path.exists(tmp):
+                    os.remove(tmp)
         else:
-            await message.reply_text("Не удалось сгенерировать музыку, попробуй позже.")
+            await message.reply_text(f"[Lyra] Временно недоступна.\n\n{ai_text}")
         return
 
+    # ============================================================
+    # ТЕКСТ
+    # ============================================================
     await message.reply_text(ai_text)
 
-# === FLASK ===
+
+# ============================================================
+# FLASK
+# ============================================================
 flask_app = Flask(__name__)
-@flask_app.route('/')
-def index(): return redirect(DOCS_URL)
-def run_flask(): flask_app.run(host="0.0.0.0", port=7860)
+
+@flask_app.route("/")
+def index():
+    return redirect(DOCS_URL)
+
+def run_flask():
+    flask_app.run(host="0.0.0.0", port=7860)
+
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
+    print("🤖 Google Bot | модели: flux / nano / banana / lyra")
     app.run()