Update app.py

app.py

@@ -120,9 +120,27 @@ class EventEmitter:
             {"document": [content], "metadata": [{"name": title, "source": url, "html": False}]},
         )
 
+async def check_cf_worker_health():
+    try:
+        url = f"{CF_WORKER_URL}/health"
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url) as response:
+                if response.ok:
+                    logging.info("Cloudflare Worker health check passed.")
+                    return True
+                else:
+                    error_text = await response.text()
+                    logging.error(f"Cloudflare Worker health check failed: {response.status}, {error_text}")
+                    return False
+    except Exception as e:
+        logging.error(f"Cloudflare Worker health check error: {e}")
+        return False
+
 
 async def set_bot_commands(bot: Bot):
-    try:
+    max_retries = 3
+    for attempt in range(max_retries):
+      try:
         url = f"{CF_WORKER_URL}/bot{TELEGRAM_BOT_TOKEN}/deleteMyCommands"
         async with aiohttp.ClientSession() as session:
            async with session.post(url) as response:
@@ -143,9 +161,12 @@ async def set_bot_commands(bot: Bot):
                     logging.error(f"Telegram set commands failed: {response.status}, {error_text}")
                     return
         logging.info("Telegram commands set successfully")
-    except Exception as e:
-        logging.error(f"Error setting Telegram commands: {e}")
-
+        return
+      except Exception as e:
+        logging.error(f"Error setting Telegram commands, attempt {attempt + 1}: {e}")
+        if attempt == max_retries - 1:
+            raise
+        await asyncio.sleep(2) # Wait for 2 second before retrying
 
 def parse_command(user_message):
     command = user_message.split(" ")[0]
@@ -535,10 +556,19 @@ def get_help_message():
 
 async def main():
     logging.basicConfig(level=logging.INFO)
+    
+    if not await check_cf_worker_health():
+        logging.error("Cloudflare Worker health check failed. Exiting.")
+        return
+
     application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
 
     # Set bot commands
-    await set_bot_commands(application.bot)
+    try:
+      await set_bot_commands(application.bot)
+    except Exception as e:
+      logging.error(f"Failed to set bot commands after multiple retries: {e}")
+      return
 
     # Command handlers
     application.add_handler(CommandHandler("start", handle_telegram_update))