Update main.py

main.py

@@ -1,53 +1,20 @@
 # main.py
 import asyncio
 import logging
-import os
-
-from fastapi import FastAPI, Request, HTTPException
-from starlette.responses import JSONResponse
 import uvicorn
-
-import bot
-import config
+from fastapi import FastAPI
+from bot import dp, bot, on_startup, main_loop_task
 
 app = FastAPI()
 
-# Health endpoint
+@app.on_event("startup")
+async def startup_event():
+    logging.info("Starting bot polling in background...")
+    asyncio.create_task(main_loop_task())
+
 @app.get("/health")
 async def health():
-    return {"status": "ok", "uptime": bot.get_uptime()}
-
-# Optional admin endpoint (secured with ADMIN_API_TOKEN)
-@app.get("/admin")
-async def admin(request: Request):
-    token = request.query_params.get("token")
-    if config.ADMIN_API_TOKEN and token != config.ADMIN_API_TOKEN:
-        raise HTTPException(status_code=403, detail="Forbidden")
-    
-    # Basic batch stats
-    batches = {
-        batch_id: {
-            "processed": batch["processed_links"],
-            "total": batch["total_links"],
-            "successful": len(batch["successful_downloads"]),
-            "failed": len(batch["failed_links"]),
-        }
-        for batch_id, batch in bot.BATCH_JOBS.items()
-    }
-    return {"active_batches": batches}
-
-async def main():
-    # Run both bot and web API
-    bot_task = asyncio.create_task(bot.run_bot())
-    await bot_task
+    return {"status": "ok"}
 
 if __name__ == "__main__":
-    # Run FastAPI + start asyncio bot inside it
-    import threading
-
-    def run_uvicorn():
-        uvicorn.run("main:app", host="0.0.0.0", port=7860, log_level="info")
-
-    threading.Thread(target=run_uvicorn, daemon=True).start()
-
-    asyncio.run(main())
+    uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=False)