Update app.py

app.py

@@ -15,7 +15,6 @@ B_KEY = os.getenv("B_KEY")
 
 # URLs
 API_URL = os.getenv("API_URL")
-UPLOAD_URL = os.getenv("UPLOAD_URL")
 
 def lipsync_api_call(video_url, audio_url):
     headers = {
@@ -53,31 +52,27 @@ def check_job_status(job_id):
     return None
 
 def get_media_duration(file_path):
-    # Fetch media duration using ffprobe
     cmd = ['ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of', 'default=noprint_wrappers=1:nokey=1', file_path]
     result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     return float(result.stdout.strip())
 
 def combine_audio_video(video_path, audio_path, output_path):
-    # Get durations of both video and audio
     video_duration = get_media_duration(video_path)
     audio_duration = get_media_duration(audio_path)
 
     if video_duration > audio_duration:
-        # Trim video to match the audio length
         cmd = [
             'ffmpeg', '-i', video_path, '-i', audio_path,
-            '-t', str(audio_duration),  # Trim video to audio duration
+            '-t', str(audio_duration),
             '-map', '0:v', '-map', '1:a',
             '-c:v', 'copy', '-c:a', 'aac',
             '-y', output_path
         ]
     else:
-        # Loop video if it's shorter than audio
-        loop_count = int(audio_duration // video_duration) + 1  # Calculate how many times to loop
+        loop_count = int(audio_duration // video_duration) + 1
         cmd = [
             'ffmpeg', '-stream_loop', str(loop_count), '-i', video_path, '-i', audio_path,
-            '-t', str(audio_duration),  # Match the duration of the final video with the audio
+            '-t', str(audio_duration),
             '-map', '0:v', '-map', '1:a',
             '-c:v', 'copy', '-c:a', 'aac',
             '-shortest', '-y', output_path
@@ -91,7 +86,7 @@ def process_video(video_url, audio_url, progress=gr.Progress()):
     if not video_url:
         return None, "No video URL provided"
         
-    session_id = str(uuid.uuid4())  # Generate a unique session ID
+    session_id = str(uuid.uuid4())
     
     progress(0.2, desc="Processing video...")
     
@@ -121,13 +116,11 @@ def process_video(video_url, audio_url, 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)
 
-            # 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:
@@ -136,7 +129,6 @@ def process_video(video_url, audio_url, progress=gr.Progress()):
             output_path = f"output_{session_id}.mp4"
             combine_audio_video(temp_video_path, temp_audio_path, output_path)
             
-            # Clean up temporary files
             os.remove(temp_video_path)
             os.remove(temp_audio_path)
             
@@ -157,15 +149,8 @@ def create_interface():
                 video_output = gr.Video(label="Generated Video")
                 status_output = gr.Textbox(label="Status", interactive=False)
         
-        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_url)
-        
         generate_btn.click(
-            fn=on_generate,
+            fn=process_video,
             inputs=[video_url_input, audio_url_input],
             outputs=[video_output, status_output]
         )