Update app.py

app.py

@@ -17,16 +17,6 @@ B_KEY = os.getenv("B_KEY")
 API_URL = os.getenv("API_URL")
 UPLOAD_URL = os.getenv("UPLOAD_URL")
 
-def upload_file(file_path):
-    with open(file_path, 'rb') as file:
-        files = {'fileToUpload': (os.path.basename(file_path), file)}
-        data = {'reqtype': 'fileupload'}
-        response = requests.post(UPLOAD_URL, files=files, data=data)
-    
-    if response.status_code == 200:
-        return response.text.strip()
-    return None
-
 def lipsync_api_call(video_url, audio_url):
     headers = {
         "Content-Type": "application/json",
@@ -95,9 +85,9 @@ def combine_audio_video(video_path, audio_path, output_path):
 
     subprocess.run(cmd, check=True)
 
-def process_video(video_url, audio_path, progress=gr.Progress()):
-    if not audio_path:
-        return None, "No audio file provided"
+def process_video(video_url, audio_url, progress=gr.Progress()):
+    if not audio_url:
+        return None, "No audio URL provided"
     if not video_url:
         return None, "No video URL provided"
         
@@ -106,12 +96,6 @@ def process_video(video_url, audio_path, progress=gr.Progress()):
     progress(0.2, desc="Processing video...")
     
     try:
-        progress(0.3, desc="Uploading audio file...")
-        audio_url = upload_file(audio_path)
-        
-        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)
         
@@ -143,11 +127,18 @@ def process_video(video_url, audio_path, progress=gr.Progress()):
             with open(temp_video_path, "wb") as f:
                 f.write(video_response.content)
 
+            # Download the audio from the URL
+            audio_response = requests.get(audio_url)
+            temp_audio_path = f"temp_audio_{session_id}.mp3"
+            with open(temp_audio_path, "wb") as f:
+                f.write(audio_response.content)
+
             output_path = f"output_{session_id}.mp4"
-            combine_audio_video(temp_video_path, audio_path, output_path)
+            combine_audio_video(temp_video_path, temp_audio_path, output_path)
             
-            # Clean up temporary video file
+            # Clean up temporary files
             os.remove(temp_video_path)
+            os.remove(temp_audio_path)
             
             progress(1.0, desc="Complete!")
             return output_path, f"Used fallback method. Original error: {str(e)}"
@@ -160,22 +151,22 @@ def create_interface():
         with gr.Row():
             with gr.Column():
                 video_url_input = gr.Textbox(label="Video URL")
-                audio_input = gr.Audio(label="Upload Audio File", type="filepath")
+                audio_url_input = gr.Textbox(label="Audio URL")
                 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(video_url, audio_file):
-            if not audio_file:
-                return None, "Please upload an audio file."
+        def on_generate(video_url, audio_url):
+            if not audio_url:
+                return None, "Please provide an audio URL."
             if not video_url:
                 return None, "Please provide a video URL."
-            return process_video(video_url, audio_file)
+            return process_video(video_url, audio_url)
         
         generate_btn.click(
             fn=on_generate,
-            inputs=[video_url_input, audio_input],
+            inputs=[video_url_input, audio_url_input],
             outputs=[video_output, status_output]
         )