Update app.py

app.py

@@ -2,9 +2,9 @@ import spaces
 import os
 import subprocess
 import tempfile
+import glob
 from huggingface_hub import snapshot_download
 import gradio as gr
-import glob
 
 # ---------------- Step 1: Download Model ----------------
 repo_id = "Wan-AI/Wan2.2-TI2V-5B"
@@ -21,12 +21,12 @@ def get_duration(prompt, size, duration_seconds, steps):
         h, w = 704, 1280
     return int(duration_seconds) * int(steps) * 2.25 + 5
 
-# Helper to find the most recent .mp4 file in cwd
-def find_latest_mp4(search_path="."):
-    mp4_files = glob.glob(os.path.join(search_path, "*.mp4"))
-    if not mp4_files:
+# ---------------- Helper: Find latest .mp4 in cwd ----------------
+def find_latest_mp4():
+    files = glob.glob("*.mp4")
+    if not files:
         return None
-    latest_file = max(mp4_files, key=os.path.getmtime)
+    latest_file = max(files, key=os.path.getctime)
     return latest_file
 
 # ---------------- Step 3: Generation Functions ----------------