update

app.py

@@ -4,82 +4,100 @@ import gdown
 import gradio as gr
 import zipfile
 
-# -----------------------------
-# DOWNLOAD SADTALKER FROM DRIVE
-# -----------------------------
+# ------------------------------------------------------
+# 1) DOWNLOAD SadTalker ZIP FROM GOOGLE DRIVE
+# ------------------------------------------------------
 
-FOLDER_ID = "1fsB8ug8zSabdtFecCpL20ApGLjz2xgvy"
+# ← Put your ZIP FILE ID here
+SADTALKER_ZIP_ID = "1fsB8ug8zSabdtFecCpL20ApGLjz2xgvy"
 
 if not os.path.exists("SadTalker"):
-    print("⬇️ Downloading SadTalker folder from Drive...")
+    print("⬇️ Downloading SadTalker.zip...")
+    gdown.download(f"https://drive.google.com/uc?id={SADTALKER_ZIP_ID}", "sad.zip", quiet=False)
 
-    # Download folder as SadTalker_drive
-    gdown.download_folder(
-        id=FOLDER_ID,
-        output="SadTalker",
-        quiet=False,
-        use_cookies=False
-    )
+    print("📦 Extracting ZIP...")
+    with zipfile.ZipFile("sad.zip", 'r') as zip_ref:
+        zip_ref.extractall("./")
 
-print("✔ SadTalker loaded successfully!")
+    # Detect extracted folder name
+    if os.path.exists("SadTalker-main"):
+        os.rename("SadTalker-main", "SadTalker")
+    elif os.path.exists("SadTalker"):
+        pass
 
+print("✔ SadTalker ready!")
 os.chdir("SadTalker")
 
-# -----------------------------
-# GENERATE TALKING VIDEO
-# -----------------------------
+# ------------------------------------------------------
+# 2) DOWNLOAD CHECKPOINT MODELS (YOUR OLD CODE)
+# ------------------------------------------------------
+
+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}")
+        gdown.download(url, out, quiet=False)
+    else:
+        print(f"✔ Already exists: {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")
+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")
+gd("1aUrFSdMHpBbH1CBMz-Y5bw0kCXGhs0Mu", "checkpoints/hubert-soft.pt")   # optional
+
+print("✔ All checkpoints ready!")
+
+# ------------------------------------------------------
+# 3) GENERATE VIDEO (Auto-detect MP4)
+# ------------------------------------------------------
 
 def generate_video(image, audio):
-    image_path = "input.png"
-    image.save(image_path)
-    audio_path = audio
+    image.save("input.png")
 
-    # Run SadTalker (GFPGAN OFF for CPU Spaces)
     cmd = [
-    "python3", "inference.py",
-    "--driven_audio", audio_path,
-    "--source_image", image_path,
-    "--result_dir", "results",
-    "--still",
-    "--expression_scale", "1.8",
-    "--preprocess", "full",
-    "--enhancer", "None"
+        "python3", "inference.py",
+        "--driven_audio", audio,
+        "--source_image", "input.png",
+        "--result_dir", "results",
+        "--still",
+        "--enhancer", "None",
+        "--expression_scale", "1.8",
+        "--preprocess", "full"
     ]
 
-
-    print("▶ Running SadTalker inference...")
+    print("▶ Running inference...")
     subprocess.run(cmd)
 
-    # Auto-find final mp4
-    mp4_list = []
+    # Auto find latest mp4
+    mp4s = []
     for root, dirs, files in os.walk("results"):
         for f in files:
             if f.endswith(".mp4"):
-                mp4_list.append(os.path.join(root, f))
+                mp4s.append(os.path.join(root, f))
 
-    if not mp4_list:
+    if not mp4s:
         return "❌ No video generated."
 
-    # Latest video
-    mp4_list.sort(key=lambda x: os.path.getmtime(x), reverse=True)
-    final_video = mp4_list[0]
-
-    print("🎉 Final video:", final_video)
-    return final_video
+    mp4s.sort(key=lambda x: os.path.getmtime(x), reverse=True)
+    return mp4s[0]
 
+# ------------------------------------------------------
+# 4) GRADIO UI
+# ------------------------------------------------------
 
-# -----------------------------
-# GRADIO UI
-# -----------------------------
 demo = gr.Interface(
     fn=generate_video,
-    inputs=[
-        gr.Image(type="pil", label="Source Image"),
-        gr.Audio(type="filepath", label="Input Audio")
-    ],
-    outputs=gr.Video(label="Generated Video"),
-    title="SadTalker (Google Drive Version)",
-    description="Loads SadTalker directly from your Google Drive (No GitHub cloning)"
+    inputs=[gr.Image(type="pil"), gr.Audio(type="filepath")],
+    outputs=gr.Video(),
+    title="SadTalker from Google Drive (Full Version)",
+    description="Loads SadTalker + all checkpoints + auto video detection"
 )
 
 demo.launch()