Update app.py

app.py

@@ -1,8 +1,11 @@
 import logging
+import time
+import asyncio
 from flask import Flask
 from threading import Thread
 from telegram import Update
 from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, ContextTypes, filters
+from telegram.error import NetworkError, TimedOut
 import google.generativeai as genai
 
 # ==== CONFIG ====
@@ -17,7 +20,6 @@ logger = logging.getLogger(__name__)
 
 # ==== GEMINI AI SETUP ====
 genai.configure(api_key=GEMINI_API_KEY)
-# Updated model name - gemini-pro is deprecated
 model = genai.GenerativeModel("gemini-1.5-flash")
 
 # ==== FLASK APP ====
@@ -27,6 +29,10 @@ flask_app = Flask(__name__)
 def home():
     return "✅ Telegram AI Chatbot is running on port 7860!"
 
+@flask_app.route("/health")
+def health():
+    return {"status": "healthy", "timestamp": time.time()}
+
 # ==== TELEGRAM BOT HANDLERS ====
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
     await update.message.reply_text("👋 Hi! I am Sumit, your AI buddy. How can I help you today?")
@@ -37,18 +43,64 @@ async def chat(update: Update, context: ContextTypes.DEFAULT_TYPE):
         response = model.generate_content(user_message)
         reply = response.text if response.text else "⚠️ I couldn't generate a reply."
     except Exception as e:
-        logger.error(e)
+        logger.error(f"Gemini error: {e}")
         reply = "❌ Something went wrong while generating response."
-    await update.message.reply_text(reply)
+    
+    try:
+        await update.message.reply_text(reply)
+    except Exception as e:
+        logger.error(f"Telegram reply error: {e}")
 
-# ==== MAIN ====
-if __name__ == "__main__":
-    # Run Flask in a separate thread
-    Thread(target=lambda: flask_app.run(host="0.0.0.0", port=7860), daemon=True).start()
+def start_bot_with_retry():
+    """Start bot with retry logic for network issues"""
+    max_retries = 5
+    retry_delay = 10
+    
+    for attempt in range(max_retries):
+        try:
+            logger.info(f"Attempting to start bot (attempt {attempt + 1}/{max_retries})")
+            
+            # Build application with custom settings for better reliability
+            app = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
+            app.add_handler(CommandHandler("start", start))
+            app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, chat))
+            
+            # Start polling with error handling
+            app.run_polling(
+                timeout=30,
+                bootstrap_retries=3,
+                read_timeout=30,
+                write_timeout=30,
+                connect_timeout=30,
+                pool_timeout=30
+            )
+            
+        except (NetworkError, TimedOut, OSError) as e:
+            logger.error(f"Network error on attempt {attempt + 1}: {e}")
+            if attempt < max_retries - 1:
+                logger.info(f"Retrying in {retry_delay} seconds...")
+                time.sleep(retry_delay)
+                retry_delay *= 2  # Exponential backoff
+            else:
+                logger.error("Max retries reached. Bot failed to start.")
+                raise
+        except Exception as e:
+            logger.error(f"Unexpected error: {e}")
+            raise
 
-    # Run Telegram bot (no asyncio.run!)
-    app = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
-    app.add_handler(CommandHandler("start", start))
-    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, chat))
+def run_flask():
+    """Run Flask app"""
+    flask_app.run(host="0.0.0.0", port=7860, debug=False)
 
-    app.run_polling()
+# ==== MAIN ====
+if __name__ == "__main__":
+    # Start Flask in a separate thread
+    flask_thread = Thread(target=run_flask, daemon=True)
+    flask_thread.start()
+    
+    # Wait a bit for Flask to start
+    time.sleep(2)
+    logger.info("Flask app started, now starting Telegram bot...")
+    
+    # Start bot with retry logic
+    start_bot_with_retry()