Update app.py

app.py

@@ -1,59 +1,27 @@
-import gradio as gr
-import subprocess, os
+from huggingface_hub import hf_hub_download
+import gradio as gr, subprocess, os
 
-# Ensure checkpoint is in the repo
-assert os.path.exists("Wav2Lip/checkpoints/wav2lip.pth"), "Checkpoint not found!"
+# Replace these with your repo details
+REPO_ID = "pratikshahp/AI-Powered-Video-Generator"
+CHECKPOINT = "Wav2Lip/checkpoints/wav2lip.pth"
 
-def generate(script: str, mic_path: str, video_path: str):
-    # Validate inputs
-    if not (script and mic_path and video_path):
-        return None, "⚠️ Please provide script, audio, and video."
-
-    # Save the script (optional—useful for logs or TTS extension)
-    with open("Wav2Lip/script.txt", "w") as f:
-        f.write(script)
+# Download checkpoint once at startup
+hf_hub_download(repo_id=REPO_ID, filename=CHECKPOINT, repo_type="space")
 
+def generate(script, mic_path, video_path):
+    if not (script and mic_path and video_path):
+        return None, "⚠️ Provide script, audio, and video."
     os.makedirs("Wav2Lip/results", exist_ok=True)
-    output_path = "Wav2Lip/results/output.mp4"
-
-    # Run Wav2Lip
+    output = "Wav2Lip/results/output.mp4"
     subprocess.call([
         "python", "Wav2Lip/inference.py",
-        "--checkpoint_path", "Wav2Lip/checkpoints/wav2lip.pth",
+        "--checkpoint_path", CHECKPOINT,
         "--face", video_path,
         "--audio", mic_path,
-        "--outfile", output_path
+        "--outfile", output
     ])
+    # Download via Hub to ensure actual file is accessible
+    local_video = hf_hub_download(repo_id=REPO_ID, filename=output, repo_type="space")
+    return local_video, "✅ Done!"
 
-    # Return path to video and success status
-    return output_path, "✅ Your video is ready!"
-
-# Build Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎬 AI Lip‑Sync Video Generator (Wav2Lip)")
-
-    script_input = gr.Textbox(
-        lines=3, placeholder="Paste the script here...", label="1) Script Input"
-    )
-    mic = gr.Audio(
-        sources=["microphone"], type="filepath", label="2) Record Audio"
-    )
-    webcam = gr.Video(
-        sources=["webcam", "upload"], format="mp4", label="3) Record / Upload Video",
-        autoplay=False
-    )
-
-    btn = gr.Button("4) Generate Video")
-    output_video = gr.Video(
-        label="Generated Video", format="mp4", autoplay=True, show_download_button=True
-    )
-    status = gr.Textbox(label="Status", interactive=False)
-
-    btn.click(
-        fn=generate,
-        inputs=[script_input, mic, webcam],
-        outputs=[output_video, status]
-    )
-
-# Launch app
-demo.launch()
+# Gradio UI setup...