Update app.py

app.py

@@ -17,9 +17,6 @@ B_KEY = os.getenv("B_KEY")
 API_URL = os.getenv("API_URL")
 UPLOAD_URL = os.getenv("UPLOAD_URL")
 
-def get_video_models():
-    return [f for f in os.listdir("models") if f.endswith((".mp4", ".avi", ".mov"))]
-
 def upload_file(file_path):
     with open(file_path, 'rb') as file:
         files = {'fileToUpload': (os.path.basename(file_path), file)}
@@ -98,22 +95,22 @@ def combine_audio_video(video_path, audio_path, output_path):
 
     subprocess.run(cmd, check=True)
 
-def process_video(model, audio_path, progress=gr.Progress()):
+def process_video(video_url, audio_path, progress=gr.Progress()):
     if not audio_path:
         return None, "No audio file provided"
+    if not video_url:
+        return None, "No video URL provided"
         
     session_id = str(uuid.uuid4())  # Generate a unique session ID
     
     progress(0.2, desc="Processing video...")
-    video_path = os.path.join("models", model)
     
     try:
-        progress(0.3, desc="Uploading files...")
-        video_url = upload_file(video_path)
+        progress(0.3, desc="Uploading audio file...")
         audio_url = upload_file(audio_path)
         
-        if not video_url or not audio_url:
-            raise Exception("Failed to upload files")
+        if not audio_url:
+            raise Exception("Failed to upload audio file")
         
         progress(0.4, desc="Initiating lipsync...")
         job_data = lipsync_api_call(video_url, audio_url)
@@ -140,35 +137,45 @@ def process_video(model, audio_path, progress=gr.Progress()):
     except Exception as e:
         progress(0.8, desc="Falling back to simple combination...")
         try:
+            # Download the video from the URL
+            video_response = requests.get(video_url)
+            temp_video_path = f"temp_video_{session_id}.mp4"
+            with open(temp_video_path, "wb") as f:
+                f.write(video_response.content)
+
             output_path = f"output_{session_id}.mp4"
-            combine_audio_video(video_path, audio_path, output_path)
+            combine_audio_video(temp_video_path, audio_path, output_path)
+            
+            # Clean up temporary video file
+            os.remove(temp_video_path)
+            
             progress(1.0, desc="Complete!")
             return output_path, f"Used fallback method. Original error: {str(e)}"
         except Exception as fallback_error:
             return None, f"All methods failed. Error: {str(fallback_error)}"
 
 def create_interface():
-    models = get_video_models()
-    
     with gr.Blocks() as app:
         gr.Markdown("# Video Lip Sync")
         with gr.Row():
             with gr.Column():
-                model_dropdown = gr.Dropdown(choices=models, label="Select Video Model", value=models[0] if models else None)
+                video_url_input = gr.Textbox(label="Video URL")
                 audio_input = gr.Audio(label="Upload Audio File", type="filepath")
                 generate_btn = gr.Button("Generate Video")
             with gr.Column():
                 video_output = gr.Video(label="Generated Video")
                 status_output = gr.Textbox(label="Status", interactive=False)
         
-        def on_generate(model_name, audio_file):
+        def on_generate(video_url, audio_file):
             if not audio_file:
                 return None, "Please upload an audio file."
-            return process_video(model_name, audio_file)
+            if not video_url:
+                return None, "Please provide a video URL."
+            return process_video(video_url, audio_file)
         
         generate_btn.click(
             fn=on_generate,
-            inputs=[model_dropdown, audio_input],
+            inputs=[video_url_input, audio_input],
             outputs=[video_output, status_output]
         )