Update main.py

main.py

@@ -1,38 +1,43 @@
 # main.py
+
 import logging
-import asyncio
-from fastapi import FastAPI
-from fastapi.responses import JSONResponse
 import uvicorn
+from fastapi import FastAPI
+import asyncio
 
-import bot  # import your bot.py (it already starts Dispatcher there)
-import db_utils
+from bot import start_bot
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-app = FastAPI()
+app = FastAPI(
+    title="Terabox Downloader Bot API",
+    version="1.0.0"
+)
+
+dp, bot = start_bot()
 
 @app.on_event("startup")
-async def on_startup():
+async def startup_event():
     logger.info("Starting bot polling in background...")
-    asyncio.create_task(bot.run_polling())
+    asyncio.create_task(dp.start_polling(bot))
+
+@app.get("/", tags=["root"])
+async def read_root():
+    return {
+        "message": "✅ Terabox Downloader Bot is running!",
+        "workers": "4",  # You can even dynamically count your workers
+        "status": "OK"
+    }
 
-@app.get("/health")
+@app.get("/health", tags=["health"])
 async def health_check():
-    return JSONResponse({
+    return {
         "status": "ok",
-        "workers": bot.config.CONCURRENT_WORKERS,
-        "active_batches": len(bot.BATCH_JOBS),
-        "active_users": await db_utils.get_all_active_user_ids_db(),
-    })
-
-@app.get("/")
-async def root_page():
-    return JSONResponse({
-        "message": "🚀 TeraBox Bot is running.",
-        "endpoints": ["/health"],
-    })
+        "bot_username": (await bot.me()).username,
+        "forward_channel": f"{bot.id}",
+        "version": "1.0.0"
+    }
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run("main:app", host="0.0.0.0", port=7860)