Update app.py

app.py

@@ -87,32 +87,46 @@ async def smart_crop(
         "aspect": aspect
     }
 
-
 @app.post("/edit")
 async def edit_video(
     file: UploadFile = File(...),
-    prompt: str = Form(...)
+    prompt: str = Form(...)  # keep prompt for future use
 ):
-    path = os.path.join(UPLOAD_DIR, file.filename)
-    with open(path, "wb") as f:
+    uid = uuid.uuid4().hex
+    video_path = os.path.join(UPLOAD_DIR, f"{uid}.mp4")
+    frame_path = os.path.join(OUTPUT_DIR, f"{uid}.png")
+
+    # Save video
+    with open(video_path, "wb") as f:
         f.write(await file.read())
 
-    # Extract first frame
-    subprocess.run([
-        "ffmpeg", "-i", path, "-frames:v", "1",
-        "frame.png"
-    ])
+    # Extract first frame safely
+    subprocess.run(
+        [
+            "ffmpeg",
+            "-y",
+            "-i", video_path,
+            "-vf", "scale=512:512:force_original_aspect_ratio=decrease",
+            "-frames:v", "1",
+            frame_path
+        ],
+        check=True
+    )
 
     from PIL import Image
-    img = Image.open("frame.png").resize((512, 512))
+    img = Image.open(frame_path).convert("RGB")
+
+    # IMPORTANT: SVD does NOT take prompt
+    with torch.no_grad():
+        output = svd(
+            image=img,
+            num_frames=16,
+            decode_chunk_size=8
+        )
 
-    frames = svd(
-        image=img,
-        prompt=prompt,
-        num_frames=16
-    ).frames
+    frames = output.frames
 
     return {
-        "prompt": prompt,
+        "prompt_received_but_unused": prompt,
         "frames_generated": len(frames)
     }