Update app.py

app.py

@@ -3,6 +3,10 @@ import sys
 import random
 import asyncio
 import threading
+import time
+import re
+import urllib.parse
+import requests
 from pyrogram import Client, filters, enums
 import google.generativeai as genai
 from flask import Flask, redirect
@@ -26,14 +30,21 @@ AI_CHAT_KEYS = [
 INFO_CHANNEL = "@g_bot_info" # Канал для уведомлений об исчерпании лимитов
 DOCS_URL = "https://rid3-quiz.web.app/docs/g-bot.html"
 
+# Укажи здесь URL твоего сервера, который возвращает музыку (MP3/WAV) в ответ на GET-запрос.
+# Пример запроса бота: http://твой-сервер.com/api/music?prompt=грустная%20мелодия
+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
+
 # --- ФУНКЦИЯ УМНОГО ОТВЕТА (АСИНХРОННАЯ С РОТАЦИЕЙ КЛЮЧЕЙ) ---
 async def get_ai_response(client, prompt, history_text):
+    global TIRED_UNTIL
     shuffled_keys = list(AI_CHAT_KEYS)
     random.shuffle(shuffled_keys)
     
-    # Делаем поведение ИИ более "человечным" и естественным
     system_instruction = (
         "Ты — умный и живой ИИ-собеседник в Telegram (модель Gemini 2.5 Flash). "
         "Общайся естественно, лаконично, как человек. Без лишней воды, без странного форматирования "
@@ -46,13 +57,11 @@ async def get_ai_response(client, prompt, history_text):
             genai.configure(api_key=key)
             model = genai.GenerativeModel('gemini-2.5-flash') 
             
-            # Используем асинхронный вызов, чтобы бот не зависал для других пользователей
             response = await model.generate_content_async(full_prompt)
             return response.text.strip()
             
         except Exception as e:
             err_str = str(e).lower()
-            # Проверяем ошибки связанные с лимитами
             if "429" in err_str or "quota" in err_str or "exhausted" in err_str:
                 print(f"⚠️ Ключ {key[:10]}... исчерпал лимит.")
                 continue
@@ -60,13 +69,14 @@ async def get_ai_response(client, prompt, history_text):
                 print(f"❌ Ошибка API: {e}")
                 continue
                 
-    # Если цикл закончился и мы здесь — значит все ключи не сработали (лимит)
+    # Если мы дошли до сюда, значит ВСЕ ключи выдали ошибку квоты. Бот уходит в спячку на час (3600 секунд).
+    TIRED_UNTIL = time.time() + 3600 
     try:
-        await client.send_message(INFO_CHANNEL, "⚠️ **Внимание!** Лимит всех API-ключей Google Bot на сегодня полностью исчерпан.")
-    except Exception as e:
-        print(f"Не удалось написать в канал: {e}")
+        await client.send_message(INFO_CHANNEL, "⚠️ **Внимание!** Лимит всех API-ключей Google Bot на сегодня полностью исчерпан. Я ушел в спячку на 1 час.")
+    except Exception:
+        pass
         
-    return "Мои вычислительные ресурсы на сегодня исчерпаны (лимиты API). Я уже уведомил об этом в своем канале 😴"
+    return "Я немного устал, я немного отдохну 😴"
 
 # --- ПОЛУЧЕНИЕ ИСТОРИИ ---
 async def fetch_history(chat_id):
@@ -80,7 +90,22 @@ async def fetch_history(chat_id):
         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=60)
+        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 Exception as e:
+        print(f"Ошибка скачивания музыки: {e}")
+    return None
+
+# --- КОМАНДЫ И ОБРАБОТКА ---
 @app.on_message(filters.command("рестарт", prefixes=".") & filters.me)
 async def restart_bot(client, message):
     await message.edit("🔄 **Перезагрузка ИИ...**")
@@ -88,6 +113,12 @@ async def restart_bot(client, message):
 
 @app.on_message(filters.text & ~filters.me)
 async def handle_bot(client, message):
+    global TIRED_UNTIL
+    
+    # Если бот "устал", он полностью игнорирует сообщения до истечения таймера
+    if time.time() < TIRED_UNTIL:
+        return
+
     text = message.text
     chat_type = message.chat.type
     is_group = chat_type in [enums.ChatType.GROUP, enums.ChatType.SUPERGROUP]
@@ -95,21 +126,17 @@ async def handle_bot(client, message):
     user_query = text
     should_respond = False
     
-    # ТРИГГЕРЫ ДЛЯ ГРУПП
     triggers = ["@g ", "гугл ", "google ", "бот ", "bot "]
 
     if not is_group:
-        # В личных сообщениях отвечаем всегда
         should_respond = True
     else:
-        # 1. Отвечаем, если обратились по триггеру
         for t in triggers:
             if text.lower().startswith(t):
                 should_respond = True
                 user_query = text[len(t):].strip()
                 break
         
-        # 2. Отвечаем, если пользователь сделал "Reply" на наше сообщение
         if not should_respond and message.reply_to_message:
             if message.reply_to_message.from_user and message.reply_to_message.from_user.is_self:
                 should_respond = True
@@ -134,40 +161,72 @@ async def handle_bot(client, message):
             action_taken = True
         except: pass
 
-    # Если это была просто короткая команда ("закрепи" или "удали"), не дергаем нейросеть
     if action_taken and len(user_query.split()) <= 2:
         return
 
+    # Проверка на конкретные команды для создания фото/музыки
+    photo_match = re.search(r'создай\s+фото\s+(.+)', user_query, re.IGNORECASE)
+    music_match = re.search(r'создай\s+музыку\s+(.+)', user_query, re.IGNORECASE)
+
     # Эффект печати
-    await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
+    if photo_match:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
+    elif music_match:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.RECORD_AUDIO)
+    else:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
     
-    # Получаем контекст и генерируем ответ
+    # Получаем контекст и генерируем ответ от ИИ
     history = await fetch_history(message.chat.id)
     ai_text = await get_ai_response(client, user_query, history)
 
-    # --- ГЕНЕРАЦИЯ ФОТО ---
-    photo_triggers = ["нарисуй", "фото", "изображение", "сгенерируй"]
-    if any(word in lower_query for word in photo_triggers):
-        await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
-        
-        # Формируем URL. Убираем лишние слова из запроса для Pollinations
-        safe_query = user_query.replace(' ', '%20')
-        photo_url = f"https://image.pollinations.ai/prompt/{safe_query}?width=1024&height=1024&nologo=true"
+    # Если ИИ вернул сообщение об усталости (произошло только что), отправляем его и завершаем
+    if ai_text == "Я немного устал, я немного отдохну 😴":
+        await message.reply_text(ai_text)
+        return
+
+    # --- ГЕНЕРАЦИЯ ФОТО ПО КОМАНДЕ ---
+    if photo_match:
+        photo_prompt = photo_match.group(1).strip()
+        safe_photo_prompt = urllib.parse.quote(photo_prompt)
+        photo_url = f"https://image.pollinations.ai/prompt/{safe_photo_prompt}?width=1024&height=1024&nologo=true"
         
-        # Telegram ограничивает подписи к фото (до 1024 символов)
-        caption = ai_text[:1000]
         try:
-            await message.reply_photo(photo_url, caption=caption)
-            return
+            await message.reply_photo(photo_url, caption=ai_text[:1000])
         except Exception:
-            # Если генерация фото не удалась, просто отправляем текст
-            pass 
+            await message.reply_text(ai_text)
+        return
+
+    # --- ГЕНЕРАЦИЯ МУЗЫКИ ПО КОМАНДЕ ---
+    if music_match:
+        music_prompt = music_match.group(1).strip()
+        
+        if MUSIC_SERVER_URL == "http://your-music-server.com/api/music":
+            # Заглушка, если свой сервер еще не подключен
+            await message.reply_text(f"🎵 **[Сервер не настроен] Ищу музыку:** `{music_prompt}`\n\n_{ai_text}_")
+            return
+            
+        # Загружаем музыку ��о своего сервера
+        downloaded_file = await asyncio.to_thread(download_music, music_prompt)
+        
+        if downloaded_file:
+            try:
+                await message.reply_audio(downloaded_file, caption=ai_text[:1000])
+            except Exception:
+                await message.reply_text(f"Музыка сгенерирована, но произошла ошибка при отправке в Telegram.\n\n_{ai_text}_")
+            finally:
+                # Удаляем временный файл
+                if os.path.exists(downloaded_file):
+                    os.remove(downloaded_file)
+        else:
+            await message.reply_text(f"❌ Мой сервер не смог сгенерировать музыку по запросу: `{music_prompt}`")
+        return
 
-    # --- ОТПРАВКА ОТВЕТА ---
+    # --- СТАНДАРТНЫЙ ОТВЕТ ТЕКСТОМ ---
     await message.reply_text(ai_text)
 
 
-# === FLASK ДЛЯ ПОДДЕРЖАНИЯ АКТИВНОСТИ (Replit, Render, HuggingFace) ===
+# === FLASK ДЛЯ ПОДДЕРЖАНИЯ АКТИВНОСТИ ===
 flask_app = Flask(__name__)
 
 @flask_app.route('/')