Update app.py

app.py

@@ -28,6 +28,7 @@ CHECKPOINT_FILES = [
 
 _setup_lock = threading.Lock()
 
+
 def run(cmd, cwd=None, extra_env=None):
     env = os.environ.copy()
     if extra_env:
@@ -43,15 +44,19 @@ def run(cmd, cwd=None, extra_env=None):
         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):
     dest.parent.mkdir(parents=True, exist_ok=True)
     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)
@@ -59,12 +64,14 @@ def ensure_checkpoints():
     for fname in CHECKPOINT_FILES:
         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():
     with _setup_lock:
         if SETUP_FLAG.exists() and REPO_DIR.exists():
@@ -73,8 +80,11 @@ def setup():
         print("Setting up Video-Retalking...")
 
         if not REPO_DIR.exists():
-            run(["git", "clone", "https://github.com/OpenTalker/video-retalking.git", str(REPO_DIR)])
+            run(
+                ["git", "clone", "https://github.com/OpenTalker/video-retalking.git", str(REPO_DIR)]
+            )
 
+        # Best effort
         try:
             run(["git", "lfs", "pull"], cwd=REPO_DIR)
         except Exception:
@@ -85,6 +95,7 @@ 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!"
@@ -99,7 +110,6 @@ def generate(image_path, audio_path):
         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",
@@ -110,9 +120,12 @@ def generate(image_path, audio_path):
         cmd = [
             "python",
             "inference.py",
-            "--face", str(image_path),
-            "--audio", str(audio_path),
-            "--outfile", str(out_path),
+            "--face",
+            str(image_path),
+            "--audio",
+            str(audio_path),
+            "--outfile",
+            str(out_path),
         ]
         run(cmd, cwd=REPO_DIR, extra_env=safe_env)
 
@@ -123,6 +136,7 @@ def generate(image_path, audio_path):
     except Exception as e:
         return None, f"❌ Error: {e}"
 
+
 demo = gr.Interface(
     fn=generate,
     inputs=[
@@ -131,7 +145,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Video(label="📹 Generated Video"),
-        gr.Textbox(label="Status", lines=5),
+        gr.Textbox(label="Status", lines=6),
     ],
     title="🎬 Video-Retalking Lip Sync",
     description="Upload a face image and audio to generate a lip-synced video.",