Update app.py

app.py

@@ -3,10 +3,8 @@ import sys
 import random
 import asyncio
 import threading
-import requests
 from pyrogram import Client, filters, enums
 import google.generativeai as genai
-from google.api_core import exceptions
 from flask import Flask, redirect
 
 # === КОНФИГУРАЦИЯ ===
@@ -25,99 +23,161 @@ AI_CHAT_KEYS = [
     "AIzaSyBjNK3MbrXnggTk2uzV3X-NK2PkT_oYB3M"
 ]
 
-CREATOR = "rid3 (@rid3_yt)"
+INFO_CHANNEL = "@g_bot_info" # Канал для уведомлений об исчерпании лимитов
 DOCS_URL = "https://rid3-quiz.web.app/docs/g-bot.html"
 
 app = Client("google_bot", session_string=SESSION_STRING, api_id=API_ID, api_hash=API_HASH)
 
-# --- ФУНКЦИЯ УМНОГО ОТВЕТА (С РОТАЦИЕЙ КЛЮЧЕЙ) ---
-def get_ai_response(prompt, history_text):
-    # Перемешиваем ключи, чтобы распределить нагрузку
+# --- ФУНКЦИЯ УМНОГО ОТВЕТА (АСИНХРОННАЯ С РОТАЦИЕЙ КЛЮЧЕЙ) ---
+async def get_ai_response(client, prompt, history_text):
     shuffled_keys = list(AI_CHAT_KEYS)
     random.shuffle(shuffled_keys)
     
+    # Делаем поведение ИИ более "человечным" и естественным
     system_instruction = (
-        f"Ты Google Bot. Модель: Gemini 2.5 Flash. Создатель: {CREATOR}. "
-        "Ты можешь управлять чатом и генерировать фото/музыку. Используй историю для контекста."
+        "Ты — умный и живой ИИ-собеседник в Telegram (модель Gemini 2.5 Flash). "
+        "Общайся естественно, лаконично, как человек. Без лишней воды, без странного форматирования "
+        "и без приветствий в каждом сообщении. Отвечай строго по сути вопроса или поддерживай беседу."
     )
-    full_prompt = f"{system_instruction}\n\nКОНТЕКСТ:\n{history_text}\n\nЗАПРОС: {prompt}"
+    full_prompt = f"ИНСТРУКЦИЯ:\n{system_instruction}\n\nКОНТЕКСТ ДИАЛОГА:\n{history_text}\n\nНОВОЕ СООБЩЕНИЕ:\n{prompt}"
 
     for key in shuffled_keys:
         try:
             genai.configure(api_key=key)
-            # Используем 1.5-flash для стабильности, так как у 2.5 слишком малые лимиты (20/день)
             model = genai.GenerativeModel('gemini-2.5-flash') 
-            response = model.generate_content(full_prompt)
-            return response.text
-        except exceptions.ResourceExhausted:
-            print(f"⚠️ Ключ {key[:10]}... исчерпал лимит. Пробую следующий...")
-            continue
-        except Exception as e:
-            print(f"❌ Ошибка ключа: {e}")
-            continue
             
-    return "😴 Все мои API-ключи сейчас отдыхают (превышен лимит). Попробуй чуть позже!"
+            # Используем асинхронный вызов, чтобы бот не зависал для других пользователей
+            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
+            else:
+                print(f"❌ Ошибка API: {e}")
+                continue
+                
+    # Если цикл закончился и мы здесь — значит все ключи не сработали (лимит)
+    try:
+        await client.send_message(INFO_CHANNEL, "⚠️ **Внимание!** Лимит всех API-ключей Google Bot на сегодня полностью исчерпан.")
+    except Exception as e:
+        print(f"Не удалось написать в канал: {e}")
+        
+    return "Мои вычислительные ресурсы на сегодня исчерпаны (лимиты API). Я уже уведомил об этом в своем канале 😴"
 
+# --- ПОЛУЧЕНИЕ ИСТОРИИ ---
 async def fetch_history(chat_id):
     messages = []
     try:
-        async for msg in app.get_chat_history(chat_id, limit=15):
-            sender = msg.from_user.first_name if msg.from_user else "Система"
-            content = msg.text or "[Медиа]"
+        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 "[Медиа/Стикер]"
             messages.append(f"{sender}: {content}")
-    except: pass
+    except: 
+        pass
     return "\n".join(reversed(messages))
 
 # --- КОМАНДЫ ---
 @app.on_message(filters.command("рестарт", prefixes=".") & filters.me)
 async def restart_bot(client, message):
-    await message.edit("🔄 **Перезагрузка Google Bot 2.5...**")
+    await message.edit("🔄 **Перезагрузка ИИ...**")
     os.execl(sys.executable, sys.executable, *sys.argv)
 
 @app.on_message(filters.text & ~filters.me)
 async def handle_bot(client, message):
     text = message.text
-    is_group = message.chat.type in [enums.ChatType.GROUP, enums.ChatType.SUPERGROUP]
+    chat_type = message.chat.type
+    is_group = chat_type in [enums.ChatType.GROUP, enums.ChatType.SUPERGROUP]
+    
+    user_query = text
+    should_respond = False
     
-    if is_group:
-        if not text.lower().startswith("@g"): return
-        user_query = text[2:].strip()
+    # ТРИГГЕРЫ ДЛЯ ГРУПП
+    triggers = ["@g ", "гугл ", "google ", "бот ", "bot "]
+
+    if not is_group:
+        # В личных сообщениях отвечаем всегда
+        should_respond = True
     else:
-        user_query = text
+        # 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
+                user_query = text.strip()
+
+    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
+        except: pass
+        
+    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
+        except: pass
 
-    if not user_query: return
+    # Если это была просто короткая команда ("закрепи" или "удали"), не дергаем нейросеть
+    if action_taken and len(user_query.split()) <= 2:
+        return
 
     # Эффект печати
     await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
     
+    # Получаем контекст и генерируем ответ
     history = await fetch_history(message.chat.id)
-    loop = asyncio.get_event_loop()
-    ai_text = await loop.run_in_executor(None, get_ai_response, user_query, history)
-
-    # Фото-генерация
-    if any(word in user_query.lower() for word in ["нарисуй", "фото", "изображение"]):
-        photo_url = f"https://image.pollinations.ai/prompt/{user_query.replace(' ', '%20')}?width=1024&height=1024&nologo=true"
-        await message.reply_photo(photo_url, caption=f"🎨 **Nano Banana 2**\n\n{ai_text}\n\n👤 Создатель: {CREATOR}")
-        return
-
-    # Управление чатом
-    if "закрепи" in user_query.lower():
+    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"
+        
+        # Telegram ограничивает подписи к фото (до 1024 символов)
+        caption = ai_text[:1000]
         try:
-            target = message.reply_to_message.id if message.reply_to_message else message.id
-            await client.pin_chat_message(message.chat.id, target)
-        except: pass
-    
-    if "удали" in user_query.lower() and message.reply_to_message:
-        try: await client.delete_messages(message.chat.id, message.reply_to_message.id)
-        except: pass
+            await message.reply_photo(photo_url, caption=caption)
+            return
+        except Exception:
+            # Если генерация фото не удалась, просто отправляем текст
+            pass 
+
+    # --- О��ПРАВКА ОТВЕТА ---
+    await message.reply_text(ai_text)
 
 
-# === FLASK ===
+# === FLASK ДЛЯ ПОДДЕРЖАНИЯ АКТИВНОСТИ (Replit, Render, HuggingFace) ===
 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()