Update app.py

app.py

@@ -37,9 +37,16 @@ logger = logging.getLogger(__name__)
 # ============================================================
 # КОНФИГУРАЦИЯ БОТА
 # ============================================================
-# РЕКОМЕНДАЦИЯ: В будущем лучше вынести токены в Secrets (переменные окружения) Hugging Face
 BOT_TOKEN = os.environ.get("BOT_TOKEN", "8605889873:AAE2gV2t0psXKlj-8h9ksyyoCrWa8Q-TdkA")
 
+# ВАЖНО: Укажите URL вашего Hugging Face Space
+# Формат: https://ВАШ_USERNAME-ВАШ_SPACE_NAME.hf.space
+WEBHOOK_URL = os.environ.get("WEBHOOK_URL", "https://rid3-aibottbh.hf.space
+")
+
+# Порт для веб-сервера (Hugging Face использует 7860)
+PORT = int(os.environ.get("PORT", 7860))
+
 # КЛЮЧИ GEMINI
 GEMINI_API_KEYS = [
     "AIzaSyDCL7CImQjZ3DlInbjmWDWw7UeQ5bQDUbU",
@@ -65,7 +72,7 @@ AI_TRIGGER_KEYWORDS = ["inai", "инэй", "инаи", "ии", "ai", "бот", "
 IMAGE_KEYWORDS = ["нарисуй", "рисуй", "draw", "картинка", "фото"]
 
 # ============================================================
-# РАБОТА С GEMINI И ИЗОБРАЖЕНИЯМИ (ЧЕРЕЗ AIOHTTP)
+# РАБОТА С GEMINI (ЧЕРЕЗ AIOHTTP)
 # ============================================================
 
 async def get_next_key():
@@ -95,24 +102,24 @@ async def call_gemini_api(prompt: str, system_prompt: str = None, history: list
             key, key_idx = await get_next_key()
             attempts += 1
             url = GEMINI_API_URL.format(model=GEMINI_MODEL, key=key)
-            
             try:
-                async with session.post(url, json=request_body, timeout=30) as response:
+                async with session.post(url, json=request_body, timeout=aiohttp.ClientTimeout(total=30)) as response:
                     if response.status == 200:
                         data = await response.json()
                         return data["candidates"][0]["content"]["parts"][0]["text"]
                     else:
-                        logger.warning(f"Ключ {key_idx} вернул статус {response.status}")
+                        error_text = await response.text()
+                        logger.warning(f"Ключ {key_idx} вернул статус {response.status}: {error_text}")
             except Exception as e:
-                logger.error(f"Ошибка API Gemini: {e}")
-                
+                logger.error(f"Ошибка API Gemini с ключом {key_idx}: {e}")
+
     return "❌ Ошибка сервиса Gemini. Попробуйте позже."
 
 async def generate_image(prompt: str) -> Optional[bytes]:
-    url = f"https://image.pollinations.ai/prompt/{urllib.parse.quote(prompt)}?nologo=true&seed={random.randint(1,999)}"
+    url = f"https://image.pollinations.ai/prompt/{urllib.parse.quote(prompt)}?nologo=true&seed={random.randint(1, 999)}"
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.get(url, timeout=40) as res:
+            async with session.get(url, timeout=aiohttp.ClientTimeout(total=60)) as res:
                 if res.status == 200:
                     return await res.read()
     except Exception as e:
@@ -133,7 +140,8 @@ def parse_ai_response(text: str):
         try:
             buttons = json.loads(btn_match.group(1))
             clean_text = clean_text.replace(btn_match.group(0), "").strip()
-        except: pass
+        except Exception as e:
+            logger.warning(f"Ошибка парсинга кнопок: {e}")
 
     draw_match = re.search(r'\[DRAW\]:\s*([^\n\[]+)', clean_text, re.IGNORECASE)
     if draw_match:
@@ -143,10 +151,16 @@ def parse_ai_response(text: str):
     return clean_text, buttons, draw_prompt
 
 def build_inline_keyboard(buttons: list) -> Optional[InlineKeyboardMarkup]:
-    if not buttons: return None
+    if not buttons:
+        return None
     keyboard = []
     for btn in buttons:
-        keyboard.append([InlineKeyboardButton(btn.get("text", "OK"), callback_data=f"ai_action:{btn.get('action', 'none')[:40]}")])
+        keyboard.append([
+            InlineKeyboardButton(
+                btn.get("text", "OK"),
+                callback_data=f"ai_action:{btn.get('action', 'none')[:40]}"
+            )
+        ])
     return InlineKeyboardMarkup(keyboard)
 
 # ============================================================
@@ -155,125 +169,198 @@ def build_inline_keyboard(buttons: list) -> Optional[InlineKeyboardMarkup]:
 chat_history: Dict[int, list] = {}
 
 def add_to_history(chat_id: int, role: str, text: str):
-    if chat_id not in chat_history: chat_history[chat_id] = []
+    if chat_id not in chat_history:
+        chat_history[chat_id] = []
     chat_history[chat_id].append({"role": role, "parts": [{"text": text}]})
     # Храним последние 10 сообщений
-    if len(chat_history[chat_id]) > 10: 
+    if len(chat_history[chat_id]) > 10:
         chat_history[chat_id].pop(0)
 
 # ============================================================
-# ОБРАБОТЧИКИ
+# ОБРАБОТЧИКИ TELEGRAM
 # ============================================================
 
 async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await update.message.reply_text("👋 Я **INAI #RID3**! Чем помочь?", parse_mode=ParseMode.MARKDOWN)
+    await update.message.reply_text(
+        "👋 Я **INAI #RID3**! Чем могу помочь?",
+        parse_mode=ParseMode.MARKDOWN
+    )
 
 async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if not update.message or not update.message.text: return
-    
+    if not update.message or not update.message.text:
+        return
+
     chat_id = update.effective_chat.id
     text = update.message.text
-    
+
     is_group = update.message.chat.type in ["group", "supergroup"]
-    is_reply = update.message.reply_to_message and update.message.reply_to_message.from_user.is_bot
-    
-    should_answer = not is_group or is_reply or any(k in text.lower() for k in AI_TRIGGER_KEYWORDS)
-    if not should_answer: return
+    is_reply = (
+        update.message.reply_to_message is not None
+        and update.message.reply_to_message.from_user is not None
+        and update.message.reply_to_message.from_user.is_bot
+    )
+
+    should_answer = (
+        not is_group
+        or is_reply
+        or any(k in text.lower() for k in AI_TRIGGER_KEYWORDS)
+    )
+    if not should_answer:
+        return
 
     await context.bot.send_chat_action(chat_id=chat_id, action=ChatAction.TYPING)
     status = await update.message.reply_text("💠 Думаю...")
-    
-    # Получаем историю без последнего сообщения пользователя
+
     history = chat_history.get(chat_id, [])
-    
+
     add_to_history(chat_id, "user", text)
     ai_resp = await call_gemini_api(text, history=history)
-    
+
     clean_text, buttons, draw_prompt = parse_ai_response(ai_resp)
     add_to_history(chat_id, "model", clean_text)
-    
+
     if draw_prompt or any(k in text.lower() for k in IMAGE_KEYWORDS):
         await status.edit_text("🎨 Рисую...")
-        img = await generate_image(draw_prompt if draw_prompt else text)
+        img_prompt = draw_prompt if draw_prompt else text
+        img = await generate_image(img_prompt)
         if img:
             await status.delete()
             await update.message.reply_photo(
-                photo=io.BytesIO(img), 
-                caption=clean_text[:1000], 
+                photo=io.BytesIO(img),
+                caption=clean_text[:1000] if clean_text else None,
                 reply_markup=build_inline_keyboard(buttons)
             )
             return
+        else:
+            await status.edit_text("❌ Не удалось сгенерировать изображение. Попробуйте ещё раз.")
+            return
 
-    # Защита от падений из-за неправильного Markdown, сгенерированного ИИ
     try:
         await status.edit_text(
-            clean_text[:4090], 
-            reply_markup=build_inline_keyboard(buttons), 
+            clean_text[:4090],
+            reply_markup=build_inline_keyboard(buttons),
             parse_mode=ParseMode.MARKDOWN
         )
     except Exception as e:
-        logger.warning(f"Ошибка Markdown разметки. Отправляю как текст: {e}")
-        await status.edit_text(
-            clean_text[:4090], 
-            reply_markup=build_inline_keyboard(buttons)
-        )
+        logger.warning(f"Ошибка Markdown разметки, отправляю как обычный текст: {e}")
+        try:
+            await status.edit_text(
+                clean_text[:4090],
+                reply_markup=build_inline_keyboard(buttons)
+            )
+        except Exception as e2:
+            logger.error(f"Критическая ошибка при отправке сообщения: {e2}")
 
 async def handle_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
     query = update.callback_query
     await query.answer()
     if query.data.startswith("ai_action:"):
-        await query.message.reply_text(f"Вы выбрали: {query.data.split(':')[-1]}. Опишите подробнее!")
+        action = query.data.split(":", 1)[-1]
+        await query.message.reply_text(
+            f"Вы выбрали: *{action}*. Опишите подробнее! 💬",
+            parse_mode=ParseMode.MARKDOWN
+        )
 
 async def error_handler(update: object, context: ContextTypes.DEFAULT_TYPE):
-    logger.error(f"Error: {context.error}")
+    logger.error(f"Произошла ошибка: {context.error}", exc_info=context.error)
 
 # ============================================================
-# ВЕБ-СЕРВЕР ДЛЯ HUGGING FACE (ПРЕДОТВРАЩАЕТ TIMEOUT ОШИБКИ)
+# ВЕБ-СЕРВЕР + WEBHOOK HANDLER (AIOHTTP)
 # ============================================================
-async def health_check(request):
-    return web.Response(text="Бот INAI успешно работает!")
 
-async def start_web_server():
-    app = web.Application()
-    app.router.add_get('/', health_check)
-    runner = web.AppRunner(app)
-    await runner.setup()
-    site = web.TCPSite(runner, '0.0.0.0', 7860)
-    await site.start()
-    logger.info("✅ Локальный веб-сервер запущен на порту 7860 (Hugging Face счастлив)")
+# Глобальная переменная для доступа к приложению из веб-сервера
+telegram_app: Optional[Application] = None
 
-# Хук, который запускает веб-сервер одновременно с ботом
-async def post_init(application: Application):
-    asyncio.create_task(start_web_server())
+async def health_check(request: web.Request) -> web.Response:
+    return web.Response(text="✅ INAI Bot работает!")
+
+async def webhook_handler(request: web.Request) -> web.Response:
+    """
+    Эта функция принимает все входящие POST-запросы от Telegram
+    и передаёт их в обработчик бота.
+    """
+    global telegram_app
+    try:
+        data = await request.json()
+        update = Update.de_json(data, telegram_app.bot)
+        await telegram_app.process_update(update)
+        return web.Response(text="OK")
+    except Exception as e:
+        logger.error(f"Ошибка в webhook_handler: {e}", exc_info=True)
+        return web.Response(status=500, text="Internal Server Error")
+
+async def build_web_app() -> web.Application:
+    """Создаём aiohttp приложение с маршрутами."""
+    app = web.Application()
+    app.router.add_get("/", health_check)
+    app.router.add_post(f"/webhook/{BOT_TOKEN}", webhook_handler)
+    return app
 
 # ============================================================
-# MAIN - ЗАПУСК
+# MAIN — ЗАПУСК ЧЕРЕЗ WEBHOOK (БЕЗ POLLING)
 # ============================================================
 
-def main():
-    print("🚀 Starting INAI Bot on Hugging Face...")
-    
-    # Настройка с МАКСИМАЛЬНЫМИ таймаутами
-    application = (
+async def main_async():
+    global telegram_app
+
+    print("🚀 Starting INAI Bot on Hugging Face (Webhook mode)...")
+
+    # Строим Telegram-приложение
+    telegram_app = (
         Application.builder()
         .token(BOT_TOKEN)
-        .post_init(post_init) # Запуск веб-сервера для HF
-        .connect_timeout(60.0)
-        .read_timeout(60.0)
-        .write_timeout(60.0)
-        .get_updates_connect_timeout(60.0)
-        .get_updates_read_timeout(60.0)
-        .pool_timeout(60.0)
+        .connect_timeout(30.0)
+        .read_timeout(30.0)
+        .write_timeout(30.0)
+        .pool_timeout(30.0)
         .build()
     )
 
-    application.add_handler(CommandHandler("start", start_command))
-    application.add_handler(CallbackQueryHandler(handle_callback))
-    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
-    application.add_error_handler(error_handler)
+    # Регистрируем обработчики
+    telegram_app.add_handler(CommandHandler("start", start_command))
+    telegram_app.add_handler(CallbackQueryHandler(handle_callback))
+    telegram_app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
+    telegram_app.add_error_handler(error_handler)
+
+    # Инициализируем бота
+    await telegram_app.initialize()
+
+    # Устанавливаем webhook
+    webhook_full_url = f"{WEBHOOK_URL.rstrip('/')}/webhook/{BOT_TOKEN}"
+    logger.info(f"Устанавливаю webhook: {webhook_full_url}")
+    await telegram_app.bot.set_webhook(
+        url=webhook_full_url,
+        drop_pending_updates=True,
+        allowed_updates=Update.ALL_TYPES,
+    )
+    logger.info("✅ Webhook успешно установлен!")
+
+    # Запускаем Telegram-приложение
+    await telegram_app.start()
+
+    # Запускаем aiohttp веб-сервер
+    web_app = await build_web_app()
+    runner = web.AppRunner(web_app)
+    await runner.setup()
+    site = web.TCPSite(runner, "0.0.0.0", PORT)
+    await site.start()
 
+    print(f"✅ Веб-сервер запущен на порту {PORT}")
+    print(f"✅ Webhook URL: {webhook_full_url}")
     print("✅ Bot is running. Waiting for messages...")
-    application.run_polling(drop_pending_updates=True)
+
+    # Держим бота запущенным вечно
+    try:
+        await asyncio.Event().wait()
+    finally:
+        logger.info("Останавливаю бота...")
+        await telegram_app.bot.delete_webhook()
+        await telegram_app.stop()
+        await telegram_app.shutdown()
+        await runner.cleanup()
+
+def main():
+    asyncio.run(main_async())
 
 if __name__ == "__main__":
     main()