Update app.py

app.py

@@ -1,70 +1,58 @@
 from fastapi import FastAPI, Query
-from fastapi.responses import RedirectResponse, JSONResponse
+from fastapi.responses import StreamingResponse, JSONResponse
 import yt_dlp
+import requests
 
 app = FastAPI(
     title="TikTok Downloader API",
-    description="Streaming TikTok MP3 & MP4 (HF Space)",
-    version="1.0"
+    version="1.1"
 )
 
-YDL_BASE = {
+YDL_OPTS = {
     "quiet": True,
     "skip_download": True,
     "noplaylist": True,
 }
 
-@app.get("/")
-def home():
-    return {
-        "status": True,
-        "author": "Farel",
-        "endpoints": {
-            "mp4": "/api/mp4?url=https://tiktok.com/...",
-            "mp3": "/api/mp3?url=https://tiktok.com/..."
-        }
-    }
+HEADERS = {
+    "User-Agent": "Mozilla/5.0",
+    "Referer": "https://www.tiktok.com/"
+}
 
-@app.get("/api/mp4")
-def mp4(url: str = Query(..., description="TikTok video URL")):
-    opts = YDL_BASE | {
-        "format": "best"
-    }
+def proxy_stream(url: str, content_type: str):
+    r = requests.get(url, headers=HEADERS, stream=True, timeout=15)
+    r.raise_for_status()
+    return StreamingResponse(
+        r.iter_content(chunk_size=1024 * 64),
+        media_type=content_type
+    )
 
+@app.get("/api/mp4")
+def mp4(url: str = Query(...)):
     try:
-        with yt_dlp.YoutubeDL(opts) as ydl:
+        with yt_dlp.YoutubeDL(YDL_OPTS | {"format": "best"}) as ydl:
             info = ydl.extract_info(url, download=False)
 
         video_url = info.get("url")
         if not video_url:
-            raise Exception("MP4 URL not found")
+            raise Exception("Video URL not found")
 
-        return RedirectResponse(video_url)
+        return proxy_stream(video_url, "video/mp4")
 
     except Exception as e:
-        return JSONResponse(
-            status_code=500,
-            content={"status": False, "error": str(e)}
-        )
+        return JSONResponse(status_code=500, content={"error": str(e)})
 
 @app.get("/api/mp3")
-def mp3(url: str = Query(..., description="TikTok video URL")):
-    opts = YDL_BASE | {
-        "format": "bestaudio"
-    }
-
+def mp3(url: str = Query(...)):
     try:
-        with yt_dlp.YoutubeDL(opts) as ydl:
+        with yt_dlp.YoutubeDL(YDL_OPTS | {"format": "bestaudio"}) as ydl:
             info = ydl.extract_info(url, download=False)
 
         audio_url = info.get("url")
         if not audio_url:
-            raise Exception("MP3 URL not found")
+            raise Exception("Audio URL not found")
 
-        return RedirectResponse(audio_url)
+        return proxy_stream(audio_url, "audio/mpeg")
 
     except Exception as e:
-        return JSONResponse(
-            status_code=500,
-            content={"status": False, "error": str(e)}
-        )
+        return JSONResponse(status_code=500, content={"error": str(e)})