Update app.py

app.py

@@ -1,4 +1,5 @@
 from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, CompositeAudioClip
+import tempfile
 import streamlit as st
 import cv2
 import base64
@@ -92,18 +93,21 @@ def merge_audio_video(video_filename, audio_filename, output_filename, overlay_a
     video_clip = VideoFileClip(video_filename)
     audio_clip = AudioFileClip(audio_filename)
 
+    audio_clips = [audio_clip]  # Start with the main audio clip
+
+    # If there's an overlay audio file, add it to the composite
     if overlay_audio_path:
         overlay_clip = AudioFileClip(overlay_audio_path).volumex(0.2)
-        audio_clip = CompositeAudioClip([audio_clip, overlay_clip.set_duration(audio_clip.duration)])
-
-    # Continue with the rest of your function...
+        audio_clips.append(overlay_clip.set_duration(audio_clip.duration))
 
+    composite_audio_clip = CompositeAudioClip(audio_clips)
 
-    if audio_clip.duration > video_clip.duration:
-        last_frame = video_clip.to_ImageClip(t=video_clip.duration-1).set_duration(audio_clip.duration - video_clip.duration)
+    if composite_audio_clip.duration > video_clip.duration:
+        # Extend video with the last frame if audio is longer
+        last_frame = video_clip.to_ImageClip(t=video_clip.duration-1).set_duration(composite_audio_clip.duration - video_clip.duration)
         video_clip = concatenate_videoclips([video_clip, last_frame])
 
-    final_clip = video_clip.set_audio(audio_clip)
+    final_clip = video_clip.set_audio(composite_audio_clip)
     final_clip.write_videofile(output_filename, codec='libx264', audio_codec="aac")
     video_clip.close()
     audio_clip.close()
@@ -130,11 +134,16 @@ overlay_audio_file = st.file_uploader("Upload overlay audio (optional)", type=["
 
 # After uploading, write it to a temporary file
 if overlay_audio_file is not None:
-    overlay_audio_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3").name
-    with open(overlay_audio_path, 'wb') as f:
-        f.write(overlay_audio_file.getvalue())
-    # Now overlay_audio_path can be safely used with MoviePy
-
+    # Create a temporary file
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
+        overlay_audio_path = temp_file.name
+        # Read content from the uploaded file
+        content = overlay_audio_file.read()
+        # Write content to the temporary file
+        temp_file.write(content)
+    
+    # Now overlay_audio_path contains the path to the temporary file
+    # and can be safely used with MoviePy
 
     voice_options = {'Echo (Male)': 'echo', 'Fable (Male)': 'fable', 'Onyx (Male)': 'onyx', 'Nova (Female)': 'nova', 'Shimmer (Female)': 'shimmer', 'Alloy (Female)': 'alloy'}
     option = st.selectbox('Choose the voice you want', list(voice_options.keys()))