Update app.py

app.py

@@ -7,7 +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
+from typing import Optional
 
 # --- Configuration ---
 app = FastAPI()
@@ -31,13 +31,13 @@ def init_db():
                 task_id TEXT PRIMARY KEY,
                 status TEXT NOT NULL,
                 file_name TEXT,
-                error_message TEXT
+                error_message TEXT,
+                duration INTEGER 
             )
         ''')
         conn.commit()
 
 # --- Background Re-encoding Task ---
-# ✅✅✅ 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")
@@ -48,15 +48,13 @@ def process_video_task(url: str, cookie: Optional[str], task_id: str):
         db.execute("UPDATE jobs SET status = 'downloading' WHERE task_id = ?", (task_id,))
         db.commit()
         
-        # ✅✅✅ 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
+            headers["Cookie"] = 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:
@@ -67,7 +65,6 @@ def process_video_task(url: str, cookie: Optional[str], task_id: str):
         db.execute("UPDATE jobs SET status = 'encoding' WHERE task_id = ?", (task_id,))
         db.commit()
 
-        # ⚠️ THIS IS THE 100% CPU RE-ENCODE YOU WANTED ⚠️
         (
             ffmpeg
             .input(temp_in_path)
@@ -75,14 +72,23 @@ def process_video_task(url: str, cookie: Optional[str], task_id: str):
             .run(capture_stdout=True, capture_stderr=True)
         )
         
+        #
+        # ✅✅✅ FIX: Get duration using ffprobe ✅✅✅
+        #
+        probe = ffmpeg.probe(temp_out_path)
+        duration = int(float(probe['format']['duration'])) # Get duration in seconds
+        
         # 3. Update DB: Complete
         final_file_name = f"{task_id}_out.mp4"
-        db.execute("UPDATE jobs SET status = 'complete', file_name = ? WHERE task_id = ?", (final_file_name, task_id))
+        db.execute(
+            "UPDATE jobs SET status = 'complete', file_name = ?, duration = ? WHERE task_id = ?", 
+            (final_file_name, duration, task_id)
+        )
         db.commit()
 
     except Exception as e:
         error_msg = str(e)
-        if hasattr(e, 'stderr'): # Get error from ffmpeg
+        if hasattr(e, 'stderr'): 
             error_msg = e.stderr.decode()
         print(f"Error processing task {task_id}: {error_msg}")
         db.execute("UPDATE jobs SET status = 'error', error_message = ? WHERE task_id = ?", (error_msg, task_id))
@@ -95,7 +101,7 @@ def process_video_task(url: str, cookie: Optional[str], task_id: str):
 # --- API Endpoints ---
 class VideoRequest(BaseModel):
     url: str
-    cookie: Optional[str] = None # ✅✅✅ FIX: Accept 'cookie'
+    cookie: Optional[str] = None 
 
 @app.post("/process")
 async def start_processing(request: VideoRequest, background_tasks: BackgroundTasks):
@@ -106,7 +112,6 @@ async def start_processing(request: VideoRequest, background_tasks: BackgroundTa
         db.execute("INSERT INTO jobs (task_id, status) VALUES (?, 'queued')", (task_id,))
         db.commit()
         
-    # ✅✅✅ 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}
 
@@ -119,7 +124,7 @@ async def get_status(task_id: str):
     if not job:
         return JSONResponse(status_code=404, content={"status": "not_found"})
     
-    return dict(job) # Convert DB row to a dict
+    return dict(job) # This will now include 'duration'
 
 @app.get("/download/{file_name}")
 async def download_file(file_name: str):
@@ -132,7 +137,7 @@ async def download_file(file_name: str):
 
 @app.get("/")
 async def health_check():
-    """ Health check endpoint to fix 404 errors in log. """
+    """ Health check endpoint. """
     return {"status": "ok"}