Update app.py

app.py

@@ -4,6 +4,7 @@ import google.generativeai as genai
 import logging
 import httpx
 from flask import Flask, request, jsonify
+import time
 
 # Suppress experimental warnings
 os.environ['HF_HUB_DISABLE_EXPERIMENTAL_WARNING'] = '1'
@@ -97,19 +98,30 @@ def handle_update():
 async def set_telegram_webhook():
     """Sets the webhook for the Telegram bot using a proxy."""
     webhook_url = f"https://measmonysuon-flyingbird.hf.space/webhooks/{WEBHOOK_SECRET}"
-    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.")
-            else:
-                logger.error(f"Failed to set webhook: {result}")
-    except httpx.RequestError as e:
-        logger.error(f"Request exception: {e}")
+    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.")
+                    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.")
+                    time.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
 
 def run_flask_app():
     """Launches the Flask app and sets the Telegram webhook."""