Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import subprocess
 import threading
 from pathlib import Path
@@ -8,10 +9,8 @@ SETUP_FLAG = Path("setup_done.txt")
 OUTPUT_DIR = Path("outputs")
 OUTPUT_DIR.mkdir(exist_ok=True)
 
-# Hugging Face mirror containing the checkpoint files (fast + reliable)
 HF_BASE = "https://huggingface.co/camenduru/video-retalking/resolve/main"
 
-# Files needed by Video-Retalking pipeline (common minimal set)
 CHECKPOINT_FILES = [
     "30_net_gen.pth",
     "BFM.zip",
@@ -29,8 +28,11 @@ CHECKPOINT_FILES = [
 
 _setup_lock = threading.Lock()
 
+def run(cmd, cwd=None, extra_env=None):
+    env = os.environ.copy()
+    if extra_env:
+        env.update(extra_env)
 
-def run(cmd, cwd=None):
     p = subprocess.run(
         cmd,
         cwd=str(cwd) if cwd else None,
@@ -38,47 +40,32 @@ def run(cmd, cwd=None):
         stderr=subprocess.STDOUT,
         text=True,
         check=False,
+        env=env,
     )
     if p.returncode != 0:
-        raise RuntimeError(
-            f"Command failed ({p.returncode}): {' '.join(cmd)}\n\n{p.stdout}"
-        )
+        raise RuntimeError(f"Command failed ({p.returncode}): {' '.join(cmd)}\n\n{p.stdout}")
     return p.stdout
 
-
 def download_file(url: str, dest: Path):
-    """
-    Download using curl with resume support (-C -) and follow redirects (-L).
-    """
     dest.parent.mkdir(parents=True, exist_ok=True)
-    # Skip if already present
     if dest.exists() and dest.stat().st_size > 0:
         return
     run(["bash", "-lc", f'curl -L -C - --retry 5 --retry-delay 2 -o "{dest}" "{url}"'])
 
-
 def ensure_checkpoints():
     ckpt_dir = REPO_DIR / "checkpoints"
     ckpt_dir.mkdir(parents=True, exist_ok=True)
 
-    # Download required checkpoint files
     for fname in CHECKPOINT_FILES:
-        url = f"{HF_BASE}/{fname}"
-        download_file(url, ckpt_dir / fname)
+        download_file(f"{HF_BASE}/{fname}", ckpt_dir / fname)
 
-    # Unzip BFM.zip -> checkpoints/BFM/
     bfm_zip = ckpt_dir / "BFM.zip"
     bfm_dir = ckpt_dir / "BFM"
     if not bfm_dir.exists():
         bfm_dir.mkdir(parents=True, exist_ok=True)
         run(["unzip", "-q", str(bfm_zip), "-d", str(bfm_dir)])
 
-
 def setup():
-    """
-    One-time setup: clone repo, pull LFS (best effort), download checkpoints.
-    Guarded by a lock to prevent multiple concurrent setup runs.
-    """
     with _setup_lock:
         if SETUP_FLAG.exists() and REPO_DIR.exists():
             return
@@ -88,11 +75,9 @@ def setup():
         if not REPO_DIR.exists():
             run(["git", "clone", "https://github.com/OpenTalker/video-retalking.git", str(REPO_DIR)])
 
-        # Best-effort: fetch any git-lfs files if repo uses them
         try:
             run(["git", "lfs", "pull"], cwd=REPO_DIR)
         except Exception:
-            # Not fatal; we'll fetch checkpoints from HF mirror anyway
             pass
 
         ensure_checkpoints()
@@ -100,7 +85,6 @@ def setup():
         SETUP_FLAG.touch()
         print("✅ Setup complete!")
 
-
 def generate(image_path, audio_path):
     if not image_path or not audio_path:
         return None, "❌ Upload both image and audio!"
@@ -111,15 +95,18 @@ def generate(image_path, audio_path):
         image_path = Path(image_path).resolve()
         audio_path = Path(audio_path).resolve()
 
-        if not image_path.exists():
-            return None, f"❌ Image not found: {image_path}"
-        if not audio_path.exists():
-            return None, f"❌ Audio not found: {audio_path}"
-
         out_path = (OUTPUT_DIR / "result.mp4").resolve()
         if out_path.exists():
             out_path.unlink()
 
+        # Fix OMP warning by forcing a valid integer value
+        safe_env = {
+            "OMP_NUM_THREADS": "1",
+            "MKL_NUM_THREADS": "1",
+            "OPENBLAS_NUM_THREADS": "1",
+            "NUMEXPR_NUM_THREADS": "1",
+        }
+
         cmd = [
             "python",
             "inference.py",
@@ -127,7 +114,7 @@ def generate(image_path, audio_path):
             "--audio", str(audio_path),
             "--outfile", str(out_path),
         ]
-        run(cmd, cwd=REPO_DIR)
+        run(cmd, cwd=REPO_DIR, extra_env=safe_env)
 
         if out_path.exists():
             return str(out_path), "✅ Video generated successfully!"
@@ -136,7 +123,6 @@ def generate(image_path, audio_path):
     except Exception as e:
         return None, f"❌ Error: {e}"
 
-
 demo = gr.Interface(
     fn=generate,
     inputs=[
@@ -145,7 +131,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Video(label="📹 Generated Video"),
-        gr.Textbox(label="Status", lines=4),
+        gr.Textbox(label="Status", lines=5),
     ],
     title="🎬 Video-Retalking Lip Sync",
     description="Upload a face image and audio to generate a lip-synced video.",