Update app.py

app.py

@@ -1,10 +1,14 @@
 import os
+import sys
 import json
 import tempfile
 import subprocess
 import shutil
 from pathlib import Path
 
+# Root of the repo — wherever app.py lives
+REPO_ROOT = Path(__file__).parent.resolve()
+
 import gradio as gr
 import spaces
 import numpy as np
@@ -39,13 +43,23 @@ def _write_caption_txt(image_path: str, caption: str) -> str:
 
 
 def _run(cmd: list[str], desc: str = "") -> tuple[bool, str]:
-    """Run a shell command and return (success, stderr/stdout)."""
+    """Run a Python -m command using the current interpreter with PYTHONPATH set."""
+    # Replace bare "python" / "python3" with the real interpreter path
+    if cmd[0] in ("python", "python3"):
+        cmd = [sys.executable] + cmd[1:]
+
+    run_env = {
+        **os.environ,
+        "PYTHONPATH": str(REPO_ROOT),          # makes lyra_2._src.* importable
+        "PYTORCH_CUDA_ALLOC_CONF": "expandable_segments:True",
+    }
     print(f"[Lyra] {desc}: {' '.join(cmd)}")
     result = subprocess.run(
         cmd,
         capture_output=True,
         text=True,
-        env={**os.environ, "PYTORCH_CUDA_ALLOC_CONF": "expandable_segments:True"},
+        env=run_env,
+        cwd=str(REPO_ROOT),                    # run from repo root
     )
     log = result.stdout + "\n" + result.stderr
     return result.returncode == 0, log
@@ -97,8 +111,7 @@ def run_zoomgs(
         if use_dmd:
             cmd.append("--use_dmd")
 
-        ok, log = _run(["env", "PYTHONPATH=."] + cmd[1:] if cmd[0] == "python" else cmd,
-                       "ZoomGS video generation")
+        ok, log = _run(cmd, "ZoomGS video generation")
 
         # Locate output video
         video_path = output_dir / "0" / "videos" / "0.mp4"
@@ -130,7 +143,7 @@ def run_zoomgs(
 # Custom trajectory (Option 2)
 # ──────────────────────────────────────────────
 
-@spaces.GPU(duration=200)
+@spaces.GPU(duration=900)
 def run_custom_traj(
     image,
     trajectory_file,
@@ -521,4 +534,4 @@ huggingface-cli download nvidia/Lyra-2.0 \\
 
 if __name__ == "__main__":
     demo = build_app()
-    demo.launch(css=CSS, ssr_mode=False)
+    demo.launch(css=CSS)