use correct CLI flags (--text_prompt, --cuda -1, --output_dir)

app.py

@@ -14,37 +14,40 @@ DEVICE    = "cpu"                         # free HF Spaces have no GPU
 
 def generate_motion(prompt: str) -> str:
     """
-    Runs the MDM sampling script in a subprocess and returns the BVH
-    file path so Gradio can hand it to the user.
+    Calls the MDM sampling script and returns the generated BVH path.
     """
-    out_file = Path("/tmp") / f"{uuid.uuid4().hex}.bvh"
+    # create a unique temp directory for this run
+    out_dir = Path("/tmp") / f"mdm_{uuid.uuid4().hex}"
+    out_dir.mkdir(parents=True, exist_ok=True)
 
     cmd = [
         "python",
         "-m",
         "motion_diffusion_model.sample.generate",
         "--model_path", str(CKPT_PATH),
-        "--prompt",     prompt,
-        "--output",     str(out_file),
-        "--device",     DEVICE,
-        "--num_steps",  "50",            # matches the checkpoint
+        "--text_prompt", prompt,
+        "--cuda", "-1",                # -1 = cpu
+        "--num_samples", "1",
+        "--output_dir", str(out_dir),
     ]
 
-    # --- make sure the local repo root is on PYTHONPATH so
-    #     'utils.*' imports inside the script can be resolved
+    # add both repo roots to PYTHONPATH
     env = os.environ.copy()
     root = Path(__file__).parent
     repo_inner = root / "motion_diffusion_model"
-    env["PYTHONPATH"] = (
-        f"{env.get('PYTHONPATH', '')}:{root}:{repo_inner}"
-    )
-
+    env["PYTHONPATH"] = f"{env.get('PYTHONPATH', '')}:{root}:{repo_inner}"
 
     completed = subprocess.run(cmd, env=env, capture_output=True, text=True)
     if completed.returncode != 0:
         raise RuntimeError(f"Inference failed:\n{completed.stderr}")
 
-    return str(out_file)
+    # find the .bvh the script just wrote
+    bvh_files = list(out_dir.rglob("*.bvh"))
+    if not bvh_files:
+        raise RuntimeError("No BVH file produced.")
+    return str(bvh_files[0])
+
+
 
 
 # -----------------------  Gradio UI  ----------------------------------