Update main.py

main.py

@@ -29,6 +29,9 @@ hentai_app = HentaiApp()
 from app.ht_rd import HentaiRandomApp
 ht_random_app = HentaiRandomApp()
 
+from app.tik_dl import TikTokDownloader
+tik_app = TikTokDownloader()
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     await invite_app.start()
@@ -133,6 +136,24 @@ async def hentai_random(
     key = x_api_key or apikey
     return await ht_random_app.get_random(key, API_KEY, limit)
 
+@app.get("/api/v1/tiktok/download")
+async def tiktok_download(
+    url: str = Query(..., description="TikTok Video URL"),
+    hd: int = Query(1, description="HD mode: 1 for ON, 0 for OFF"),
+    apikey: Optional[str] = Query(None),
+    x_api_key: Optional[str] = Header(None, alias="X-API-Key")
+):
+    key = x_api_key or apikey
+    if not API_KEY or key != API_KEY:
+        raise HTTPException(status_code=403, detail="API key invalid. Stop sniffin' around.")
+
+    res = await tik_app.fetch_video(url, hd)
+    
+    if not res["ok"]:
+        raise HTTPException(status_code=400, detail=res["error"])
+    
+    return res
+
 @app.get("/")
 def home():
     uptime_seconds = int(time.time() - START_TIME)