add app.py

app.py

@@ -0,0 +1,76 @@
+import os
+import uuid
+import gradio as gr
+from wyn_wav2lip.wav2lip import Wav2Lip
+
+MEDIA_DIR = "media"
+os.makedirs(MEDIA_DIR, exist_ok=True)
+
+wav2lip = Wav2Lip()
+wav2lip.setup()
+
+def lipsync_func(image, audio):
+    # image: PIL.Image
+    # audio: tuple (sample_rate, data) or filepath depending on Gradio component
+
+    # Save inputs
+    image_path = os.path.join(MEDIA_DIR, f"{uuid.uuid4()}.png")
+    audio_path = os.path.join(MEDIA_DIR, f"{uuid.uuid4()}.wav")
+
+    image.save(image_path)
+
+    import soundfile as sf
+    sr, data = audio  # if using gr.Audio(type="numpy")
+    sf.write(audio_path, data, sr)
+
+    # Run Wav2Lip like we did before
+    # (Use the same logic as in the Colab example)
+    video_path = run_wav2lip(image_path, audio_path)
+
+    return video_path
+
+def run_wav2lip(image_path, audio_path):
+    # similar to the Colab sample before
+    existing_mp4 = {
+        f for f in os.listdir(MEDIA_DIR)
+        if f.lower().endswith(".mp4")
+    }
+
+    import os
+    from os import getcwd, chdir
+    
+    old = getcwd()
+    chdir(MEDIA_DIR)
+    try:
+        wav2lip.run(
+            video_file=os.path.basename(image_path),
+            vocal_file=os.path.basename(audio_path),
+        )
+    finally:
+        chdir(old)
+
+    new_mp4 = [
+        f for f in os.listdir(MEDIA_DIR)
+        if f.lower().endswith(".mp4") and f not in existing_mp4
+    ]
+    if not new_mp4:
+        mp4_candidates = [
+            os.path.join(MEDIA_DIR, f)
+            for f in os.listdir(MEDIA_DIR)
+            if f.lower().endswith(".mp4")
+        ]
+        if not mp4_candidates:
+            raise RuntimeError("No video created")
+        return max(mp4_candidates, key=os.path.getmtime)
+    return os.path.join(MEDIA_DIR, new_mp4[0])
+
+
+demo = gr.Interface(
+    fn=lipsync_func,
+    inputs=[gr.Image(type="pil"), gr.Audio(type="numpy")],
+    outputs=gr.Video(),
+    title="Wav2Lip Lipsync Service"
+)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,2 @@
+wyn-wav2lip
+gradio