update

app.py

@@ -5,10 +5,9 @@ import gradio as gr
 import zipfile
 
 # ------------------------------------------------------
-# 1) DOWNLOAD SadTalker ZIP FROM GOOGLE DRIVE
+# 1) DOWNLOAD SadTalker ZIP FROM GOOGLE DRIVE (ONLY ONCE)
 # ------------------------------------------------------
 
-# ← Put your ZIP FILE ID here
 SADTALKER_ZIP_ID = "1ERCCvqt2YTNiwfqY1N95aQIIBm3maApu"
 
 if not os.path.exists("SadTalker"):
@@ -19,17 +18,15 @@ if not os.path.exists("SadTalker"):
     with zipfile.ZipFile("sad.zip", 'r') as zip_ref:
         zip_ref.extractall("./")
 
-    # Detect extracted folder name
+    # Rename folder if extracted as SadTalker-main
     if os.path.exists("SadTalker-main"):
         os.rename("SadTalker-main", "SadTalker")
-    elif os.path.exists("SadTalker"):
-        pass
 
 print("✔ SadTalker ready!")
 os.chdir("SadTalker")
 
 # ------------------------------------------------------
-# 2) DOWNLOAD CHECKPOINT MODELS (YOUR OLD CODE)
+# 2) DOWNLOAD CHECKPOINT MODELS (ONLY IF MISSING)
 # ------------------------------------------------------
 
 def gd(file_id, out):
@@ -42,7 +39,6 @@ def gd(file_id, out):
 
 os.makedirs("checkpoints", exist_ok=True)
 
-# Your original model IDs (replace with yours)
 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")
@@ -54,7 +50,7 @@ gd("1cag0u7e5RgdKoxBa7exY599VNGtYgNQb", "checkpoints/shape_predictor_68_face_lan
 print("✔ All checkpoints ready!")
 
 # ------------------------------------------------------
-# 3) GENERATE VIDEO (Auto-detect MP4)
+# 3) GENERATE VIDEO (FAST + ALWAYS RETURNS OUTPUT)
 # ------------------------------------------------------
 
 def generate_video(image, audio):
@@ -69,12 +65,13 @@ def generate_video(image, audio):
         "--size", "256",
         "--enhancer", "None",
         "--expression_scale", "1.8",
-        "--preprocess", "crop"
+        "--preprocess", "full"
     ]
 
+    print("▶ Running inference...")
     subprocess.run(cmd)
 
-    # find latest mp4
+    # Find all mp4 in results
     mp4s = []
     for root, dirs, files in os.walk("results"):
         for f in files:
@@ -84,12 +81,24 @@ def generate_video(image, audio):
     if not mp4s:
         return "❌ No video generated."
 
+    # Pick latest file
     mp4s.sort(key=lambda x: os.path.getmtime(x), reverse=True)
     latest = mp4s[0]
 
-    import shutil
-    shutil.copy(latest, "output.mp4")
+    print("🎬 Video generated:", latest)
+    return latest
 
-    return "output.mp4"
 
+# ------------------------------------------------------
+# 4) GRADIO UI
+# ------------------------------------------------------
+
+demo = gr.Interface(
+    fn=generate_video,
+    inputs=[gr.Image(type="pil"), gr.Audio(type="filepath")],
+    outputs=gr.Video(),
+    title="SadTalker (Google Drive Version)",
+    description="Fast loading + no duplicate downloads + auto video return"
+)
 
+demo.launch()