find mp4

app.py

@@ -4,6 +4,7 @@ import subprocess
 import tempfile
 from huggingface_hub import snapshot_download
 import gradio as gr
+import glob
 
 # ---------------- Step 1: Download Model ----------------
 repo_id = "Wan-AI/Wan2.2-TI2V-5B"
@@ -20,6 +21,14 @@ 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:
+        return None
+    latest_file = max(mp4_files, key=os.path.getmtime)
+    return latest_file
+
 # ---------------- Step 3: Generation Functions ----------------
 @spaces.GPU(duration=get_duration)
 def generate_t2v(prompt, size="1280*704", duration_seconds=5, steps=25):
@@ -27,7 +36,6 @@ def generate_t2v(prompt, size="1280*704", duration_seconds=5, steps=25):
         return None, None, "Please enter a prompt."
 
     temp_dir = tempfile.mkdtemp()
-    output_path = os.path.join(temp_dir, "output.mp4")
 
     cmd = [
         "python", "generate.py",
@@ -46,13 +54,15 @@ def generate_t2v(prompt, size="1280*704", duration_seconds=5, steps=25):
     except subprocess.CalledProcessError as e:
         return None, None, f"Error during T2V generation: {e}"
 
-    if os.path.exists("output.mp4"):
-        os.rename("output.mp4", output_path)
-    if not os.path.exists(output_path):
-        return None, None, "Generation finished but output file not found."
+    video_file = find_latest_mp4()
+    if video_file is None:
+        return None, None, "Generation finished but no video file found."
+
+    dest_path = os.path.join(temp_dir, os.path.basename(video_file))
+    os.rename(video_file, dest_path)
 
-    download_link = f"<a href='file/{output_path}' download>📥 Download Video</a>"
-    return output_path, download_link, "Text-to-Video generated successfully!"
+    download_link = f"<a href='file/{dest_path}' download>📥 Download Video</a>"
+    return dest_path, download_link, "Text-to-Video generated successfully!"
 
 @spaces.GPU(duration=get_duration)
 def generate_i2v(image, prompt, size="1280*704", duration_seconds=5, steps=25):
@@ -63,8 +73,6 @@ def generate_i2v(image, prompt, size="1280*704", duration_seconds=5, steps=25):
     image_path = os.path.join(temp_dir, "input.jpg")
     image.save(image_path)
 
-    output_path = os.path.join(temp_dir, "output.mp4")
-
     cmd = [
         "python", "generate.py",
         "--task", "ti2v-5B",
@@ -83,13 +91,15 @@ def generate_i2v(image, prompt, size="1280*704", duration_seconds=5, steps=25):
     except subprocess.CalledProcessError as e:
         return None, None, f"Error during I2V generation: {e}"
 
-    if os.path.exists("output.mp4"):
-        os.rename("output.mp4", output_path)
-    if not os.path.exists(output_path):
-        return None, None, "Generation finished but output file not found."
+    video_file = find_latest_mp4()
+    if video_file is None:
+        return None, None, "Generation finished but no video file found."
+
+    dest_path = os.path.join(temp_dir, os.path.basename(video_file))
+    os.rename(video_file, dest_path)
 
-    download_link = f"<a href='file/{output_path}' download>📥 Download Video</a>"
-    return output_path, download_link, "Image-to-Video generated successfully!"
+    download_link = f"<a href='file/{dest_path}' download>📥 Download Video</a>"
+    return dest_path, download_link, "Image-to-Video generated successfully!"
 
 # ---------------- Step 4: Gradio UI ----------------
 with gr.Blocks() as demo: