update

app.py

@@ -3,93 +3,87 @@ import subprocess
 import gdown
 import gradio as gr
 
-# ---------------------------------------------------
-# AUTO CLONE SADTALKER FROM GITHUB (only first time)
-# ---------------------------------------------------
-
+# -----------------------------------------
+# AUTO CLONE SADTALKER (only first time)
+# -----------------------------------------
 if not os.path.exists("SadTalker"):
-    print("Cloning SadTalker from GitHub...")
+    print("➡️ Cloning SadTalker...")
     subprocess.run(["git", "clone", "https://github.com/OpenTalker/SadTalker.git"])
 
-# Move inside SadTalker folder
 os.chdir("SadTalker")
 
-# ---------------------------------------------------
-# GOOGLE DRIVE DOWNLOADER
-# ---------------------------------------------------
-
+# -----------------------------------------
+# GOOGLE DRIVE CHECKPOINT DOWNLOADER
+# -----------------------------------------
 def gd(file_id, out):
     url = f"https://drive.google.com/uc?id={file_id}"
     if not os.path.exists(out):
-        print(f"Downloading {out} ...")
+        print(f"⬇️ Downloading: {out}")
         gdown.download(url, out, quiet=False)
     else:
-        print(f"{out} already exists.")
+        print(f"✔ Already exists: {out}")
 
 os.makedirs("checkpoints", exist_ok=True)
 
-# Download all required model files
-gd("1g3VIpU3yhpITMZtrWU2mbmyzLoF9K3Gz", "checkpoints/auido2exp_00300-model.pth")
-gd("1Jp4i_Qc-6qCms7v1kN61RE3qnT_9J8vg", "checkpoints/auido2pose_00140-model.pth")
+# Download your files
+gd("1g3VIpU3yhpITMZtrWU2mbmyzLoF9K3Gz", "checkpoints/audio2exp_00300-model.pth")
+gd("1Jp4i_Qc-6qCms7v1kN61RE3qnT_9J8vg", "checkpoints/audio2pose_00140-model.pth")
 gd("1pTbKmpOeWRSA1NYQ3DnWe3W-9Qeyy7PK", "checkpoints/mapping_00109-model.pth.tar")
 gd("1QJins0e_hvvZM4d47gR9eoCxa0L4LpWj", "checkpoints/mapping_00229-model.pth.tar")
 gd("1caz3ESiy-aEzI0ptff8qpjcxdTf-L26E", "checkpoints/SadTalker_V0.0.2_256.safetensors")
 gd("12mj9EEs4KIS6-9FjF-PjEegF-TDKvvi6", "checkpoints/SadTalker_V0.0.2_512.safetensors")
-gd("1cag0u7e5RgdKoxBa7exY599VNGtYgNQb", "checkpoints/shape_predictor_68_face_landmarks.dat")
-
-# ---------------------------------------------------
-# SADTALKER INFERENCE FUNCTION
-# ---------------------------------------------------
 
+# -----------------------------------------
+# SADTALKER INFERENCE (AUTO-FIND RESULT)
+# -----------------------------------------
 def generate_video(image, audio):
-    # Save image
+
+    # Save input image
     image_path = "input.png"
     image.save(image_path)
-
     audio_path = audio
 
-    # Run SadTalker
+    # Run SadTalker (GFPGAN off: no freezing!)
     cmd = [
         "python3", "inference.py",
         "--driven_audio", audio_path,
         "--source_image", image_path,
-        "--result_dir", "./",
-        "--enhancer", "gfpgan",
+        "--result_dir", "results",
         "--still",
+        "--enhancer", "None"
     ]
+
+    print("🎬 Running SadTalker...")
     subprocess.run(cmd)
 
-    # 🔎 Find all mp4 files inside SadTalker folder
+    # Auto-find latest MP4 inside results folder
     mp4_files = []
-    for root, dirs, files in os.walk("."):
+    for root, dirs, files in os.walk("results"):
         for f in files:
             if f.endswith(".mp4"):
                 mp4_files.append(os.path.join(root, f))
 
     if not mp4_files:
-        return "Error: No video generated"
+        return "❌ Error: No video generated."
 
-    # 💯 Latest generated video choose
     mp4_files.sort(key=lambda x: os.path.getmtime(x), reverse=True)
-    final_video = mp4_files[0]
+    final_mp4 = mp4_files[0]
 
-    print("Returning video:", final_video)
-    return final_video
+    print("🎉 Generated Video:", final_mp4)
+    return final_mp4
 
-
-# ---------------------------------------------------
+# -----------------------------------------
 # GRADIO UI
-# ---------------------------------------------------
-
+# -----------------------------------------
 demo = gr.Interface(
     fn=generate_video,
     inputs=[
         gr.Image(type="pil", label="Source Image"),
-        gr.Audio(type="filepath", label="Input Voice")
+        gr.Audio(type="filepath", label="Input Audio")
     ],
-    outputs=gr.Video(label="Generated Talking Video"),
-    title="SadTalker Online API (Auto-Clone + Google Drive Models)",
-    description="Upload image + audio → Generate talking video"
+    outputs=gr.Video(label="Generated Video"),
+    title="SadTalker (HF Space Auto-Working)",
+    description="Upload image + audio → AI talking video"
 )
 
 demo.launch()