Update app.py

app.py

@@ -857,10 +857,11 @@ from moviepy.audio.AudioClip import CompositeAudioClip  # For combining audio tr
 
 # --- Functions ---
 
-def merge_videos(video_files, logo_path1, logo_path2, music_path, output_path):
+def merge_videos(video_files, logo_path1, logo_path2, music_path, output_path, orig_vol=1.0, music_vol=0.5):
     """
     Merge video clips, overlay two logos, and combine with background music.
     The original video audio is preserved and mixed with the background music.
+    The user can control the volumes of the original audio and background music.
     The merged video is written with the composite audio.
     """
     # Load video clips (each with its own audio)
@@ -868,14 +869,14 @@ def merge_videos(video_files, logo_path1, logo_path2, music_path, output_path):
     # Concatenate clips (preserving original audio)
     final_clip = concatenate_videoclips(clips)
     # Optional: Resize to a fixed resolution if needed
-    # final_clip = final_clip.resized(newsize=(1080, 720))
+    # final_clip = final_clip.resize(newsize=(1080, 720))
     
     # Determine final duration
     final_duration = final_clip.duration or sum(clip.duration for clip in clips)
     
     # Calculate positions based on frame dimensions
     final_width = final_clip.w  # current width of the video
-    # Adjust positions: left logo inset a bit, right logo inset from right edge (logo width=70)
+    # Adjust positions: left logo inset a bit, right logo inset from right edge (logo width=70 assumed)
     left_pos = (int(final_width * 0.02), 20)
     right_pos = (int(final_width * 0.98) - 70, 20)
     
@@ -889,17 +890,18 @@ def merge_videos(video_files, logo_path1, logo_path2, music_path, output_path):
              .with_position(right_pos)
              .with_duration(final_duration))
     
-    # Load background music, adjust volume, and match duration
-    music = AudioFileClip(music_path).with_volume_scaled(0.3).with_duration(final_duration) # Background music volume
+    # Load background music, adjust volume using slider value, and match duration
+    music = AudioFileClip(music_path).with_volume_scaled(music_vol).with_duration(final_duration)
     
-    # Composite the video with logos; original audio is preserved
-    composite_clip = CompositeVideoClip([final_clip, logo1, logo2])
-    original_audio = final_clip.audio
+    # Adjust original video audio using slider value
+    original_audio = final_clip.audio.with_volume_scaled(orig_vol)
+    
+    # Composite the video with logos and audio tracks
     composite_audio = CompositeAudioClip([original_audio, music])
-    composite_clip = composite_clip.with_audio(composite_audio).with_volume_scaled(0.6) # Background original video volume
+    composite_clip = CompositeVideoClip([final_clip, logo1, logo2]).with_audio(composite_audio)
     
     # Write the merged video with composite audio
-    composite_clip.write_videofile(output_path, codec="libx264", fps=30)
+    composite_clip.write_videofile(output_path, codec="libx264", fps=30, audio_codec="aac")
     
     # Close resources
     composite_clip.close()
@@ -932,11 +934,11 @@ def add_captions_to_video(video_path, captions, output_path):
     
     # Font parameters
     font = cv2.FONT_HERSHEY_SIMPLEX
-    font_scale = 0.8
+    font_scale = 0.5
     thickness = 1
     text_color = (0, 0, 0)      # Black text
     box_color = (0, 255, 255)   # Yellow box (BGR)
-    margin = 5                # Margin around text for background box
+    margin = 3                # Margin around text for background box
     
     frame_count = 0
     while True:
@@ -997,6 +999,10 @@ logo_file1 = st.file_uploader("Upload Logo Image 1 (Top Left)", type=["png", "jp
 logo_file2 = st.file_uploader("Upload Logo Image 2 (Top Right)", type=["png", "jpg", "jpeg"])
 music_file = st.file_uploader("Upload Music File", type=["mp3", "wav"])
 
+# Sliders for controlling audio volumes
+original_audio_vol = st.slider("Original Video Audio Volume", min_value=0.0, max_value=1.0, value=1.0, step=0.05)
+music_audio_vol = st.slider("Background Music Volume", min_value=0.0, max_value=1.0, value=0.5, step=0.05)
+
 # Create a persistent temporary directory in session state
 if 'temp_dir' not in st.session_state:
     st.session_state.temp_dir = tempfile.mkdtemp()
@@ -1009,7 +1015,7 @@ if 'intermediate_path' in st.session_state:
 # Step 1: Merge Videos Button
 if st.button("Merge Videos"):
     if video_files and logo_file1 and logo_file2 and music_file:
-        # Save uploaded files into session_state.temp_dir
+        # Save uploaded files into st.session_state.temp_dir
         video_paths = []
         for video in video_files:
             path = os.path.join(st.session_state.temp_dir, video.name)
@@ -1031,7 +1037,8 @@ if st.button("Merge Videos"):
         
         # Define the intermediate output file path
         intermediate_path = os.path.join(st.session_state.temp_dir, "intermediate_output.mp4")
-        merge_videos(video_paths, logo_path1, logo_path2, music_path, intermediate_path)
+        merge_videos(video_paths, logo_path1, logo_path2, music_path, intermediate_path,
+                     orig_vol=original_audio_vol, music_vol=music_audio_vol)
         
         # Determine total length of the merged video
         clip = VideoFileClip(intermediate_path)