Update app.py

app.py

@@ -1,14 +1,22 @@
+import os
 import telebot
 import google.generativeai as genai
 import logging
 import requests
 import gradio as gr
+import httpx
+import asyncio
+from flask import Flask, request, jsonify
 
+# Suppress experimental warnings
+os.environ['HF_HUB_DISABLE_EXPERIMENTAL_WARNING'] = '1'
 
 # Configuration
 GOOGLE_API_KEY = 'AIzaSyAYXUMnwmR4nUGDCs97FJJsafcQAPAAuzE'
 BOT_TOKEN = '7484321656:AAFaswxTqaSHu_s4jd_pk2Q2OJJWYcWHwAM'
-WEBHOOK_URL = f"https://measmonysuon-flyingbird.hf.space/webhooks/handle_update"
+WEBHOOK_SECRET = 'A3%26c8%21jP%23xZ1v*Qw5kL%5E0tR%40u9%25yS6'  # URL-encoded secret
+POWER_USER_ID = 75516649
+PROXY_URL = 'http://eR3LhYeoZXNWhIp:[email protected]:58874'
 
 # Initialize the Telegram bot
 bot = telebot.TeleBot(BOT_TOKEN)
@@ -76,5 +84,92 @@ iface = gr.Interface(
     live=True
 )
 
+# Initialize Flask app
+app = Flask(__name__)
+
+@app.route(f'/webhooks/{WEBHOOK_SECRET}', methods=['POST'])
+def handle_update():
+    """Handles incoming updates from Telegram."""
+    payload = request.json
+    logger.debug(f"Received payload: {payload}")  # Log the incoming payload
+
+    if payload.get('message'):
+        message_text = payload['message'].get('text')
+        chat_id = payload['message']['chat']['id']
+
+        if message_text:
+            try:
+                # Forward message to Gradio and get the response
+                response = requests.post('http://localhost:7860/', json={'text': message_text})
+                if response.status_code == 200:
+                    response_text = response.json().get('data', 'Sorry, I cannot process this request.')
+                else:
+                    response_text = 'Error occurred while processing your request.'
+
+                # Log and send the response
+                logger.debug(f"Generated response: {response_text}")
+                asyncio.run(bot_send_message(chat_id, response_text))
+                logger.info(f"Response sent to chat_id {chat_id}")
+
+            except Exception as e:
+                logger.error(f"Error during processing: {e}")
+                error_message = "Sorry, I can't answer this query right now but I will improve from time to time."
+                asyncio.run(bot_send_message(chat_id, error_message))
+                logger.error(f"Error message sent to chat_id {chat_id}")
+
+    return jsonify({'status': 'ok'}), 200
+
+async def set_telegram_webhook():
+    """Sets the webhook for the Telegram bot with proxy configuration."""
+    webhook_url = f"https://measmonysuon-flyingbird.hf.space/webhooks/{WEBHOOK_SECRET}"
+    retry_attempts = 5
+    retry_delay = 1  # seconds
+    
+    for attempt in range(retry_attempts):
+        try:
+            async with httpx.AsyncClient(proxies=PROXY_URL) as client:
+                response = await client.post(
+                    f"https://api.telegram.org/bot{BOT_TOKEN}/setWebhook",
+                    data={"url": webhook_url},
+                )
+                result = response.json()
+                if result.get('ok'):
+                    logger.info("Webhook set successfully.")
+                    # Notify power user
+                    await bot_send_message(POWER_USER_ID, "🚀 The webhook has been successfully connected! 🎉")
+                    return
+                elif result.get('error_code') == 429:
+                    retry_after = result['parameters'].get('retry_after', retry_delay)
+                    logger.warning(f"Rate limit exceeded. Retrying after {retry_after} seconds.")
+                    await asyncio.sleep(retry_after)
+                else:
+                    logger.error(f"Failed to set webhook: {result}")
+                    return
+        except httpx.RequestError as e:
+            logger.error(f"Request exception: {e}")
+            return
+
+async def bot_send_message(chat_id, text):
+    """Send a message to the specified chat ID using the proxy configuration."""
+    async with httpx.AsyncClient(proxies=PROXY_URL) as client:
+        response = await client.post(
+            f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage",
+            data={"chat_id": chat_id, "text": text, "parse_mode": "Markdown"}
+        )
+        if response.status_code == 200:
+            logger.info(f"Message sent successfully to chat_id {chat_id}")
+        else:
+            logger.error(f"Failed to send message: {response.text}")
+
+def run_flask_app():
+    """Launches the Flask app and sets the Telegram webhook."""
+    asyncio.run(set_telegram_webhook())
+    app.run(host="0.0.0.0", port=5000, debug=True)
+
 if __name__ == "__main__":
-    iface.launch()  # Launch the Gradio app
+    # Launch the Gradio app in a separate thread/process
+    import threading
+    threading.Thread(target=iface.launch, kwargs={'server_name': '0.0.0.0', 'server_port': 7860}).start()
+
+    # Start Flask app
+    run_flask_app()