Update app.py

app.py

@@ -1,33 +1,59 @@
-import gradio as gr, subprocess, os
+import gradio as gr
+import subprocess, os
 
-assert os.path.exists("Wav2Lip/checkpoints/wav2lip.pth")
+# Ensure checkpoint is in the repo
+assert os.path.exists("Wav2Lip/checkpoints/wav2lip.pth"), "Checkpoint not found!"
 
-def generate(script, mic_path, video_path):
-    # Basic input checks
-    if not script or not mic_path or not video_path:
-        return None, "⚠️ Please provide all script, audio, and video."
-    
-    os.makedirs("results", exist_ok=True)
+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)
+
+    os.makedirs("Wav2Lip/results", exist_ok=True)
+    output_path = "Wav2Lip/results/output.mp4"
+
+    # Run Wav2Lip
     subprocess.call([
         "python", "Wav2Lip/inference.py",
         "--checkpoint_path", "Wav2Lip/checkpoints/wav2lip.pth",
         "--face", video_path,
         "--audio", mic_path,
-        "--outfile", "Wav2Lip/results/output.mp4"
+        "--outfile", output_path
     ])
-    return "results/output.mp4", ""
 
+    # 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")
-    
-    script_input = gr.Textbox(lines=3, placeholder="Paste script here", label="1) Script")
-    mic = gr.Audio(sources=["microphone"], type="filepath", label="2) Record Audio")
-    webcam = gr.Video(sources=["webcam"], label="3) Record Video")
-    
+    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")
-    status = gr.Text(label="Status Message")
-    
-    btn.click(fn=generate, inputs=[script_input, mic, webcam], outputs=[output_video, status])
+    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()