Update app.py

app.py

@@ -7,6 +7,7 @@ import uuid
 from fastapi import FastAPI, BackgroundTasks
 from pydantic import BaseModel
 from starlette.responses import FileResponse, JSONResponse
+from typing import Optional # <-- Import Optional
 
 # --- Configuration ---
 app = FastAPI()
@@ -36,7 +37,8 @@ def init_db():
         conn.commit()
 
 # --- Background Re-encoding Task ---
-def process_video_task(url: str, task_id: str):
+# ✅✅✅ FIX: Accept 'cookie' as an argument
+def process_video_task(url: str, cookie: Optional[str], task_id: str):
     db = get_db()
     temp_in_path = os.path.join(DOWNLOAD_DIR, f"{task_id}_in.mp4")
     temp_out_path = os.path.join(ENCODED_DIR, f"{task_id}_out.mp4")
@@ -46,7 +48,16 @@ def process_video_task(url: str, task_id: str):
         db.execute("UPDATE jobs SET status = 'downloading' WHERE task_id = ?", (task_id,))
         db.commit()
         
-        with httpx.stream("GET", url, follow_redirects=True, timeout=600.0) as r:
+        # ✅✅✅ FIX: Add browser headers
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36",
+            "Referer": "https://terabox.com/"
+        }
+        if cookie:
+            headers["Cookie"] = cookie # Add the cookie
+        
+        # ✅✅✅ FIX: Use the 'headers' when downloading
+        with httpx.stream("GET", url, headers=headers, follow_redirects=True, timeout=600.0) as r:
             r.raise_for_status()
             with open(temp_in_path, 'wb') as f:
                 for chunk in r.iter_bytes(chunk_size=8192):
@@ -84,6 +95,7 @@ def process_video_task(url: str, task_id: str):
 # --- API Endpoints ---
 class VideoRequest(BaseModel):
     url: str
+    cookie: Optional[str] = None # ✅✅✅ FIX: Accept 'cookie'
 
 @app.post("/process")
 async def start_processing(request: VideoRequest, background_tasks: BackgroundTasks):
@@ -94,7 +106,8 @@ async def start_processing(request: VideoRequest, background_tasks: BackgroundTa
         db.execute("INSERT INTO jobs (task_id, status) VALUES (?, 'queued')", (task_id,))
         db.commit()
         
-    background_tasks.add_task(process_video_task, request.url, task_id)
+    # ✅✅✅ FIX: Pass the cookie to the background task
+    background_tasks.add_task(process_video_task, request.url, request.cookie, task_id)
     return {"status": "queued", "task_id": task_id}
 
 @app.get("/status/{task_id}")
@@ -117,9 +130,6 @@ async def download_file(file_name: str):
     
     return FileResponse(file_path, media_type='video/mp4', filename=file_name)
 
-#
-# ✅✅✅ FIX IS HERE ✅✅✅
-#
 @app.get("/")
 async def health_check():
     """ Health check endpoint to fix 404 errors in log. """
@@ -134,3 +144,4 @@ async def startup_event():
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)
+