Update app.py

app.py

@@ -4,19 +4,15 @@ import os
 from pathlib import Path
 
 def setup():
-    """One-time setup"""
     if Path("setup_done.txt").exists():
         return
     
-    print("Installing dependencies...")
-    # Install compatible versions
-    os.system("pip install -q numpy==1.23.5")
-    os.system("pip install -q gfpgan realesrgan")
+    print("Setting up Video-Retalking...")
     
-    print("Downloading model...")
+    # Clone repo
     os.system("git clone https://github.com/OpenTalker/video-retalking.git vrt")
     
-    # Download checkpoints
+    # Download models
     os.chdir("vrt")
     os.system("bash scripts/download_models.sh")
     os.chdir("..")
@@ -26,7 +22,7 @@ def setup():
 
 def generate(image, audio):
     if not image or not audio:
-        return None, "Upload both files!"
+        return None, "❌ Upload both image and audio!"
     
     try:
         setup()
@@ -46,19 +42,20 @@ def generate(image, audio):
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
-# Simple UI
+# Gradio Interface
 demo = gr.Interface(
     fn=generate,
     inputs=[
-        gr.Image(type="filepath", label="Face Image"),
-        gr.Audio(type="filepath", label="Audio File")
+        gr.Image(type="filepath", label="📷 Face Image"),
+        gr.Audio(type="filepath", label="🎵 Audio File")
     ],
     outputs=[
-        gr.Video(label="Result"),
-        gr.Textbox(label="Status")
+        gr.Video(label="📹 Generated Video"),
+        gr.Textbox(label="Status", lines=2)
     ],
-    title="AI Lip Sync Generator",
-    description="Upload face image and audio"
+    title="🎬 Video-Retalking Lip Sync",
+    description="Upload a face image and audio to generate lip-synced video"
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()