Update main.py

main.py

@@ -1,29 +1,29 @@
 import asyncio
 import os
-import nest_asyncio
-from telethon import TelegramClient
-from telethon.sessions import MemorySession
 from dummy_server import run_server
-from bot import setup_handlers
+from bot import app
 from worker import run_worker
-
-nest_asyncio.apply()
+from pyrogram import idle
 
 async def main():
-    api_id = int(os.environ["API_ID"])
-    api_hash = os.environ["API_HASH"]
-    bot_token = os.environ["BOT_TOKEN"]
-
-    client = TelegramClient(MemorySession(), api_id, api_hash)
-    await client.start(bot_token=bot_token)
+    # 1. Start the dummy web server so Hugging Face keeps the Space awake on port 7860
+    print("🌐 Starting Dummy Server...")
+    asyncio.create_task(run_server())
+    
+    # 2. Start the Pyrogram Bot
+    print("🤖 Starting Pyrogram Bot...")
+    await app.start()
+    
+    # 3. Start the Sniper Workers
+    print("🚀 Firing up Sniper Workers...")
+    asyncio.create_task(run_worker())
     
-    setup_handlers(client)
+    # 4. Keep the process alive forever
+    await idle()
     
-    await asyncio.gather(
-        run_server(),
-        client.run_until_disconnected(),
-        run_worker(),
-    )
+    # Cleanup on shutdown
+    await app.stop()
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    # Pyrogram's app.run() handles the async event loop perfectly
+    app.run(main())