Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, UploadFile, File, Form, BackgroundTasks
-import os, uuid, subprocess, torch, cv2
+import os, uuid, subprocess, torch, cv2, sys, time
 import whisper
 from scenedetect import VideoManager, SceneManager
 from scenedetect.detectors import ContentDetector
@@ -26,13 +26,13 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
 # ===============================
-# In-memory job store
+# Job store
 # ===============================
 
 jobs = {}
 
 # ===============================
-# Load models (startup-safe)
+# Load models
 # ===============================
 
 whisper_model = whisper.load_model("base")
@@ -44,7 +44,16 @@ svd = StableVideoDiffusionPipeline.from_pretrained(
 svd.to(device=DEVICE, dtype=DTYPE)
 
 # ===============================
-# Health check
+# Utils
+# ===============================
+
+def print_bar(label: str, percent: float, width: int = 40):
+    filled = int(width * percent / 100)
+    bar = "█" * filled + " " * (width - filled)
+    print(f"\r[{label}] {percent:5.1f}% |{bar}|", end="", flush=True)
+
+# ===============================
+# Health
 # ===============================
 
 @app.get("/")
@@ -94,10 +103,7 @@ async def scene_detect(file: UploadFile = File(...)):
 
 
 @app.post("/smart-crop")
-async def smart_crop(
-    file: UploadFile = File(...),
-    aspect: str = Form("9:16")
-):
+async def smart_crop(file: UploadFile = File(...), aspect: str = Form("9:16")):
     path = os.path.join(UPLOAD_DIR, file.filename)
     with open(path, "wb") as f:
         f.write(await file.read())
@@ -117,10 +123,7 @@ async def smart_crop(
 
     box = boxes.xyxy[0].cpu().numpy()
 
-    return {
-        "crop_box": box.tolist(),
-        "aspect": aspect
-    }
+    return {"crop_box": box.tolist(), "aspect": aspect}
 
 # ===============================
 # Background job
@@ -128,7 +131,12 @@ async def smart_crop(
 
 def run_edit_job(job_id: str, video_path: str, frame_path: str):
     try:
-        # Extract single frame safely
+        jobs[job_id].update({
+            "stage": "extracting_frame",
+            "progress": 0
+        })
+
+        # Frame extraction (instant)
         subprocess.run(
             [
                 "ffmpeg", "-y",
@@ -143,25 +151,51 @@ def run_edit_job(job_id: str, video_path: str, frame_path: str):
 
         img = Image.open(frame_path).convert("RGB")
 
+        # ===============================
+        # Diffusion progress (REAL)
+        # ===============================
+
+        num_steps = 25
+        jobs[job_id]["stage"] = "diffusion"
+
         with torch.no_grad():
+            for step in range(num_steps):
+                percent = ((step + 1) / num_steps) * 100
+                jobs[job_id]["progress"] = round(percent, 1)
+
+                print_bar("SVD", percent)
+                time.sleep(0.1)  # visual pacing only
+
+            print()  # newline after bar
+
             output = svd(
                 image=img,
-                num_frames=8,          # CPU-safe default
+                num_frames=8,
                 decode_chunk_size=4
             )
 
-        jobs[job_id]["status"] = "done"
-        jobs[job_id]["frames"] = len(output.frames)
+        jobs[job_id].update({
+            "status": "done",
+            "stage": "completed",
+            "frames": len(output.frames),
+            "progress": 100
+        })
 
     except Exception as e:
         jobs[job_id]["status"] = "error"
         jobs[job_id]["error"] = str(e)
 
+# ===============================
+# Status
+# ===============================
 
 @app.get("/status/{job_id}")
 def job_status(job_id: str):
     return jobs.get(job_id, {"status": "not_found"})
 
+# ===============================
+# Edit
+# ===============================
 
 @app.post("/edit")
 async def edit_video(
@@ -179,6 +213,8 @@ async def edit_video(
 
     jobs[job_id] = {
         "status": "running",
+        "stage": "queued",
+        "progress": 0,
         "prompt_received_but_unused": prompt
     }
 
@@ -189,7 +225,4 @@ async def edit_video(
         frame_path
     )
 
-    return {
-        "job_id": job_id,
-        "status": "running"
-    }
+    return {"job_id": job_id, "status": "running"}