Update app.py

app.py

@@ -1,40 +1,27 @@
 import gradio as gr
-import subprocess
-import os
+import subprocess, os
 
-# Ensure checkpoints folder exists
-assert os.path.exists("Wav2Lip/checkpoints/wav2lip.pth"), "checkpoint not found"
+assert os.path.exists("checkpoints/wav2lip.pth")
 
-def generate(script, mic_audio, webcam_video):
-    # Save script to temp for TTS (if needed)
-    with open("script.txt", "w") as f:
-        f.write(script)
-
-    # Assume mic_audio & webcam_video are files
-    audio_path = mic_audio.name
-    video_path = webcam_video.name
-
-    # Run Wav2Lip with your existing command
-    cmd = [
+def generate(mic_path: str, video_path: str):
+    output_path = "results/output.mp4"
+    os.makedirs("results", exist_ok=True)
+    subprocess.call([
         "python", "inference.py",
         "--checkpoint_path", "checkpoints/wav2lip.pth",
         "--face", video_path,
-        "--audio", audio_path,
-        "--outfile", "results/output.mp4"
-    ]
-    subprocess.call(cmd)
-
-    return "results/output.mp4"
+        "--audio", mic_path,
+        "--outfile", output_path
+    ])
+    return output_path
 
-# Build Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎙️ Wav2Lip Lip-Sync Video Generator")
-    script_input = gr.Textbox(lines=4, placeholder="Paste your script here")
+    gr.Markdown("## 🎬 AI Lip‑Sync Video Recorder")
     mic = gr.Audio(sources=["microphone"], type="filepath", label="Record Audio")
-    webcam = gr.Video(sources=["webcam"], label="Record Video")
-    generate_btn = gr.Button("Generate Video")
-    output_video = gr.Video(label="Generated Video")
+    webcam = gr.Video(sources=["webcam"], type="filepath", label="Record Video")
+    btn = gr.Button("Generate Video")
+    out = gr.Video(label="Generated Video")
 
-    generate_btn.click(fn=generate, inputs=[script_input, mic, webcam], outputs=output_video)
+    btn.click(generate, inputs=[mic, webcam], outputs=out)
 
-demo.launch()
+demo.launch(share=True)