Update main.py

main.py

@@ -253,13 +253,16 @@ async def chatbot_talk(client: Client, message: Message):
 
 # Define the main coroutine
 async def main():
-    await db.connect()      # Connect to your database
-    LOGS.info("Connected to the database.")
-    await client.start()   # Start the Pyrogram client
-    LOGS.info("Bot started successfully.")
-    await idle()    # Keep the bot running until interrupted
-    LOGS.info("Bot stopping...")
-    await client.stop()    # Ensure the client stops gracefully
+    try:
+        await db.connect()      # Connect to your database
+        LOGS.info("Connected to the database.")
+        await client.start()   # Start the Pyrogram client
+        LOGS.info("Bot started successfully.")
+        await idle()    # Keep the bot running until interrupted
+        LOGS.info("Bot stopping...")
+        await client.stop()    # Ensure the client stops gracefully
+    except Exception as e:
+        LOGS.error(f"Unexpected error: {e}")
 
 # Entry point
 if __name__ == "__main__":