Updated app.py to include global sesarch of MP4 file

app.py

@@ -6,7 +6,8 @@ import subprocess
 import gradio as gr
 
 # Optional: set this if you want GPU on Spaces
-USE_GPU = False  # we're on CPU Basic now
+# (We are on CPU Basic, so keep this False.)
+USE_GPU = False
 
 
 def generate_motion(prompt: str):
@@ -20,7 +21,7 @@ def generate_motion(prompt: str):
     run_id = str(uuid.uuid4())[:8]
 
     # Build command
-    gpu_flag = "-1"  # CPU
+    gpu_flag = "-1"  # CPU only on Spaces
     cmd = [
         "python",
         "gen_t2m.py",
@@ -35,21 +36,45 @@ def generate_motion(prompt: str):
     except subprocess.CalledProcessError as e:
         raise gr.Error(f"Generation failed. See logs. (Error: {e})")
 
-    # Find the generated mp4
-    anim_dir = os.path.join("generation", run_id, "animation")
-    mp4_files = glob.glob(os.path.join(anim_dir, "*.mp4"))
+    # ------------------------------------------------------------------
+    # Find the generated mp4 (be very flexible about where it is)
+    # ------------------------------------------------------------------
 
-    if not mp4_files:
+    # Find all mp4 files anywhere in the repo
+    all_mp4s = glob.glob("**/*.mp4", recursive=True)
+
+    # Prefer files whose path contains this run_id
+    mp4_candidates = [f for f in all_mp4s if run_id in f]
+
+    # If nothing has run_id, prefer files under "generation/"
+    if not mp4_candidates:
+        mp4_in_generation = [f for f in all_mp4s if "generation" in f]
+        if mp4_in_generation:
+            mp4_candidates = mp4_in_generation
+
+    # If still nothing, fall back to any mp4 at all
+    if not mp4_candidates and all_mp4s:
+        mp4_candidates = all_mp4s
+
+    # If we still have no mp4s, something went wrong in gen_t2m.py
+    if not mp4_candidates:
         raise gr.Error("No MP4 file was generated. Please check server logs.")
 
-    # Return the first video file (Gradio will display it)
-    return mp4_files[0]
+    # Pick the most recently modified mp4 as the output
+    mp4_path = max(mp4_candidates, key=os.path.getmtime)
+
+    # Return the video path (Gradio will display it)
+    return mp4_path
 
 
 # Build the Gradio interface
 demo = gr.Interface(
     fn=generate_motion,
-    inputs=gr.Textbox(lines=2, label="Text prompt", placeholder="A person is walking forward."),
+    inputs=gr.Textbox(
+        lines=2,
+        label="Text prompt",
+        placeholder="A person is walking forward.",
+    ),
     outputs=gr.Video(label="Generated motion"),
     title="MoMask: Text-to-Motion Generation",
     description=(