Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 import subprocess, os
 
-assert os.path.exists("Wav2Lip/checkpoints/wav2lip.pth")
+assert os.path.exists("checkpoints/wav2lip.pth")
+
+def generate(script: str, mic_path: str, video_path: str):
+    # Save script if you want to use it for TTS later
+    with open("script.txt", "w") as f:
+        f.write(script)
+
+    os.makedirs("results", exist_ok=True)
+    output_path = os.path.join("results", "output.mp4")
 
-def generate(mic_path: str, video_path: str):
-    output_dir = "results"
-    os.makedirs(output_dir, exist_ok=True)
-    output_path = os.path.join(output_dir, "output.mp4")
     subprocess.call([
         "python", "inference.py",
         "--checkpoint_path", "checkpoints/wav2lip.pth",
@@ -19,12 +23,27 @@ def generate(mic_path: str, video_path: str):
 with gr.Blocks() as demo:
     gr.Markdown("## 🎬 AI Lip‑Sync Video Generator")
     
-    mic = gr.Audio(sources=["microphone"], type="filepath", label="Record Audio")
-    webcam = gr.Video(sources=["webcam"], label="Record Video")  # removed 'type'
-    
-    btn = gr.Button("Generate")
+    script_input = gr.Textbox(
+        lines=3, 
+        placeholder="Paste your script here...", 
+        label="1) Script Input"
+    )
+    mic = gr.Audio(
+        sources=["microphone"], 
+        type="filepath", 
+        label="2) Record Audio"
+    )
+    webcam = gr.Video(
+        sources=["webcam"], 
+        label="3) Record Video"
+    )
+    btn = gr.Button("4) Generate Video")
     output_video = gr.Video(label="Generated Video")
 
-    btn.click(fn=generate, inputs=[mic, webcam], outputs=output_video)
+    btn.click(
+        fn=generate, 
+        inputs=[script_input, mic, webcam], 
+        outputs=output_video
+    )
 
-demo.launch(share=True)
+demo.launch()