Update app.py

app.py

@@ -20,8 +20,16 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
     response = generator(user_message, max_length=50, num_return_sequences=1, truncation=True)[0]['generated_text']
     await update.message.reply_text(response)
 
+async def error_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    print(f"Error occurred: {context.error}")
+    if "Conflict" in str(context.error):
+        await context.bot.send_message(
+            chat_id=update.effective_chat.id if update else None,
+            text="Bot conflict detected. Restarting..."
+        )
+
 # Run the Telegram bot
-def run_telegram_bot():
+async def run_telegram_bot():
     if not TELEGRAM_BOT_TOKEN:
         print("Telegram bot token not found. Please set TELEGRAM_BOT_TOKEN in the Space secrets.")
         return
@@ -31,11 +39,24 @@ def run_telegram_bot():
     # Add handlers
     application.add_handler(CommandHandler("start", start))
     application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
+    application.add_error_handler(error_handler)
+
+    # Clear any existing webhook and updates
+    print("Clearing any existing webhook and updates...")
+    try:
+        await application.bot.delete_webhook(drop_pending_updates=True)
+        # Fetch and discard any pending updates to clear the queue
+        updates = await application.bot.get_updates(timeout=1)
+        if updates:
+            last_update_id = updates[-1].update_id
+            await application.bot.get_updates(offset=last_update_id + 1, timeout=1)
+    except Exception as e:
+        print(f"Error clearing updates: {e}")
 
     # Start the bot in the main thread
     print("Starting Telegram bot...")
-    asyncio.run(application.run_polling(allowed_updates=Update.ALL_TYPES))
+    await application.run_polling(allowed_updates=Update.ALL_TYPES)
 
 # Run the bot
 if __name__ == "__main__":
-    run_telegram_bot()
+    asyncio.run(run_telegram_bot())