Update app.py

app.py

@@ -27,107 +27,94 @@ 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"
-
-# Укажи здесь 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):
+# --- ФУНКЦИЯ УМНОГО ОТВЕТА ---
+async def get_ai_response(client, prompt, history_text, is_media_request=False):
     global TIRED_UNTIL
     shuffled_keys = list(AI_CHAT_KEYS)
     random.shuffle(shuffled_keys)
     
+    # Новая, более жесткая инструкция, чтобы ИИ не говорил "я не могу"
     system_instruction = (
-        "Ты — умный и живой ИИ-собеседник в Telegram (модель Gemini 2.5 Flash). "
-        "Общайся естественно, лаконично, как человек. Без лишней воды, без странного форматирования "
-        "и без приветствий в каждом сообщении. Отвечай строго по сути вопроса или поддерживай беседу."
+        "Ты — Google Bot (модель Gemini 2.5 Flash). Ты ЖИВОЙ и УМНЫЙ. "
+        "ВАЖНО: Ты УМЕЕШЬ создавать изображения и музыку. Если тебя просят что-то создать, "
+        "отвечай уверенно: 'Конечно, сейчас сделаю', 'Вот твой запрос' или опиши, что ты рисуешь. "
+        "НИКОГДА не говори 'я не могу создавать изображения' или 'я текстовая модель'. "
+        "Твой стиль: лаконичный, дружелюбный, без формализма."
     )
-    full_prompt = f"ИНСТРУКЦИЯ:\n{system_instruction}\n\nКОНТЕКСТ ДИАЛОГА:\n{history_text}\n\nНОВОЕ СООБЩЕНИЕ:\n{prompt}"
+    
+    full_prompt = f"СИСТЕМНАЯ УСТАНОВКА:\n{system_instruction}\n\nИСТОРИЯ ЧАТА:\n{history_text}\n\nПОЛЬЗОВАТЕЛЬ: {prompt}"
 
     for key in shuffled_keys:
         try:
             genai.configure(api_key=key)
             model = genai.GenerativeModel('gemini-2.5-flash') 
-            
             response = await model.generate_content_async(full_prompt)
-            return response.text.strip()
+            text = response.text.strip()
+            
+            # Фильтр "тупняка": если ИИ все равно выдал отказ, подменяем его
+            refusal_phrases = ["я не могу", "я всего лишь текстовая", "не умею создавать фото", "я не имею возможности"]
+            if is_media_request and any(phrase in text.lower() for phrase in refusal_phrases):
+                return "Конечно! Вот, что мне удалось создать по твоему запросу:"
+            
+            return text
             
         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
-            else:
-                print(f"❌ Ошибка API: {e}")
+            if any(x in err_str for x in ["429", "quota", "exhausted"]):
                 continue
+            print(f"❌ Ошибка API: {e}")
+            continue
                 
-    # Если мы дошли до сюда, значит ВСЕ ключи выдали ошибку квоты. Бот уходит в спячку на час (3600 секунд).
     TIRED_UNTIL = time.time() + 3600 
     try:
-        await client.send_message(INFO_CHANNEL, "⚠️ **Внимание!** Лимит всех API-ключей Google Bot на сегодня полностью исчерпан. Я ушел в спячку на 1 час.")
-    except Exception:
-        pass
-        
+        await client.send_message(INFO_CHANNEL, "⚠️ Все API ключи сдохли. Отдыхаю 1 час.")
+    except: pass
     return "Я немного устал, я немн��го отдохну 😴"
 
 # --- ПОЛУЧЕНИЕ ИСТОРИИ ---
 async def fetch_history(chat_id):
     messages = []
     try:
-        async for msg in app.get_chat_history(chat_id, limit=12):
-            sender = msg.from_user.first_name if msg.from_user else "Собеседник"
-            content = msg.text or msg.caption or "[Медиа/Стикер]"
+        # Уменьшил лимит до 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
+    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=60)
+        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)
+            with open(filename, "wb") as f: f.write(response.content)
             return filename
-    except Exception as e:
-        print(f"Ошибка скачивания музыки: {e}")
+    except: pass
     return None
 
-# --- КОМАНДЫ И ОБРАБОТКА ---
-@app.on_message(filters.command("рестарт", prefixes=".") & filters.me)
-async def restart_bot(client, message):
-    await message.edit("🔄 **Перезагрузка ИИ...**")
-    os.execl(sys.executable, sys.executable, *sys.argv)
-
+# --- ОБРАБОТЧИК ---
 @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
-    chat_type = message.chat.type
-    is_group = chat_type in [enums.ChatType.GROUP, enums.ChatType.SUPERGROUP]
-    
+    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:
@@ -136,107 +123,75 @@ 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:
-            if message.reply_to_message.from_user and message.reply_to_message.from_user.is_self:
-                should_respond = True
-                user_query = text.strip()
+        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: return
 
-    # --- УПРАВЛЕНИЕ ЧАТОМ ---
-    action_taken = False
+    # Команды управления
     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)
-            action_taken = True
+        try: await client.pin_chat_message(message.chat.id, message.reply_to_message.id)
         except: pass
-        
+        return
     if "удали" in lower_query and message.reply_to_message:
-        try: 
-            await client.delete_messages(message.chat.id, message.reply_to_message.id)
-            action_taken = True
+        try: await client.delete_messages(message.chat.id, message.reply_to_message.id)
         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)
+    is_media = bool(photo_match or music_match)
 
-    # Эффект печати
-    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)
+    # Индикация действия
+    action = enums.ChatAction.TYPING
+    if photo_match: action = enums.ChatAction.UPLOAD_PHOTO
+    elif music_match: action = enums.ChatAction.RECORD_AUDIO
+    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)
+    ai_text = await get_ai_response(client, user_query, history, is_media_request=is_media)
 
-    # Если ИИ вернул сообщение об усталости (произошло только что), отправляем его и завершаем
-    if ai_text == "Я немного устал, я немного отдохну 😴":
+    if "я немного устал" in ai_text.lower():
         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"
-        
+        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(photo_url, caption=ai_text[:1000])
-        except Exception:
-            await message.reply_text(ai_text)
+            await message.reply_photo(url, caption=ai_text[:1000])
+        except:
+            await message.reply_text(f"🖼 {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}_")
+        prompt = music_match.group(1).strip()
+        if "your-music-server" in MUSIC_SERVER_URL:
+            await message.reply_text(f"🎵 Сервер музыки не настроен.\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)
+        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)
         else:
-            await message.reply_text(f"❌ Мой сервер не смог сгенерировать музыку по запросу: `{music_prompt}`")
+            await message.reply_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)
+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("🤖 Бот успешно запущен!")
     app.run()