Update app.py

app.py

@@ -722,7 +722,8 @@ def process_video_bg(vid, bg_type="색상", bg_image=None, bg_video=None, color=
         yield None, None, f"비디오 처리 오류: {e}. 경과 시간: {elapsed_time:.2f}초"
 
 @spaces.GPU(duration=180)
-def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
+@spaces.GPU(duration=180)
+def merge_videos_with_audio(video_files, audio_file, audio_mode, audio_volume, original_audio_volume, output_fps):
     """여러 비디오를 병합하고 오디오를 추가하는 함수"""
     if not video_files:
         return None, "비디오 파일을 업로드해주세요."
@@ -773,20 +774,15 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
             logging.warning(f"비디오 크기가 서로 다릅니다. 첫 번째 비디오 크기({target_width}x{target_height})로 조정합니다.")
             
             # 크기가 다른 비디오들을 조정
-
             adjusted_clips = []
             for clip, size in zip(video_clips, clip_sizes):
                 if size != (target_width, target_height):
-                    # resize 함수가 있으면 사용, 없으면 대체 방법 사용
                     if resize is not None:
                         adjusted_clip = resize(clip, newsize=(target_width, target_height))
                     else:
-                        # resize가 없을 때 대체 방법
-                        # clip.resize() 메서드 사용 시도
                         if hasattr(clip, 'resize'):
                             adjusted_clip = clip.resize((target_width, target_height))
                         else:
-                            # 최후의 수단: 그대로 사용
                             adjusted_clip = clip
                             logging.warning(f"Cannot resize video. Using original size.")
                     adjusted_clips.append(adjusted_clip)
@@ -813,19 +809,17 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
                 if isinstance(audio_file, str):
                     audio_path = audio_file
                 else:
-                    # gr.Audio에서 반환된 튜플인 경우
                     audio_path = audio_file
                 
                 logging.info(f"Processing audio from: {audio_path}")
+                logging.info(f"Audio mode: {audio_mode}")
                 
                 # 오디오 로드
                 if audio_path.endswith(('.mp4', '.avi', '.mov', '.mkv')):
-                    # 비디오 파일에서 오디오 추출
                     temp_video = VideoFileClip(audio_path)
                     audio_clip = temp_video.audio
                     temp_video.close()
                 else:
-                    # 오디오 파일 직접 로드
                     audio_clip = AudioFileClip(audio_path)
                 
                 if audio_clip is None:
@@ -840,31 +834,39 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
                 audio_duration = audio_clip.duration
                 
                 if audio_duration > video_duration:
-                    # 오디오가 더 길면 잘라냄
                     audio_clip = audio_clip.subclip(0, video_duration)
                 elif audio_duration < video_duration:
-                    # 오디오가 더 짧으면 반복
                     loops_needed = int(video_duration / audio_duration) + 1
                     audio_clips_list = [audio_clip] * loops_needed
                     looped_audio = concatenate_audioclips(audio_clips_list)
                     audio_clip = looped_audio.subclip(0, video_duration)
                 
-                # 기존 오디오 제거하고 새 오디오로 교체
-                # (기존 오디오와 합성하려면 아래 주석 해제)
-                final_video = final_video.set_audio(audio_clip)
-                
-                # 기존 오디오와 새 오디오 합성을 원하는 경우:
-                # if final_video.audio:
-                #     final_audio = CompositeAudioClip([final_video.audio, audio_clip])
-                #     final_video = final_video.set_audio(final_audio)
-                # else:
-                #     final_video = final_video.set_audio(audio_clip)
+                # 오디오 모드에 따른 처리
+                if audio_mode == "백그라운드 뮤직":
+                    # 백그라운드 뮤직 모드: 기존 오디오와 합성
+                    if final_video.audio:
+                        # 원본 오디오 볼륨 조절
+                        original_audio = final_video.audio
+                        if original_audio_volume != 100:
+                            original_audio = original_audio.volumex(original_audio_volume / 100)
+                        
+                        # 두 오디오 합성
+                        final_audio = CompositeAudioClip([original_audio, audio_clip])
+                        final_video = final_video.set_audio(final_audio)
+                        logging.info("Background music mode: Mixed original and new audio")
+                    else:
+                        # 원본 오디오가 없으면 그냥 추가
+                        final_video = final_video.set_audio(audio_clip)
+                        logging.info("No original audio found, adding new audio only")
+                else:
+                    # 대체 모드: 기존 오디오를 완전히 교체
+                    final_video = final_video.set_audio(audio_clip)
+                    logging.info("Replace mode: Replaced original audio")
                 
-                logging.info("Audio successfully added to video")
+                logging.info("Audio successfully processed")
                     
             except Exception as e:
                 logging.error(f"오디오 처리 중 오류 발생: {str(e)}")
-                # 오디오 처리 실패해도 비디오는 계속 처리
                 status = f"오디오 처리 실패: {str(e)}, 비디오만 병합합니다."
         
         status = "비디오 저장 중..."
@@ -879,8 +881,8 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
                 fps=output_fps,
                 codec="libx264",
                 audio_codec="aac",
-                preset="medium",  # 품질 설정
-                bitrate="5000k",  # 비트레이트 설정으로 품질 유지
+                preset="medium",
+                bitrate="5000k",
                 audio_bitrate="192k"
             )
         
@@ -895,7 +897,15 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
             audio_clip.close()
         final_video.close()
         
-        return temp_filepath, f"✅ 성공적으로 {len(video_paths)}개의 비디오를 병합했습니다! (크기: {target_width}x{target_height})"
+        # 상태 메시지 생성
+        if audio_file and audio_mode == "백그라운드 뮤직":
+            mode_msg = "백그라운드 뮤직 추가됨"
+        elif audio_file:
+            mode_msg = "오디오 대체됨"
+        else:
+            mode_msg = "오디오 없음"
+        
+        return temp_filepath, f"✅ 성공적으로 {len(video_paths)}개의 비디오를 병합했습니다! (크기: {target_width}x{target_height}, {mode_msg})"
         
     except Exception as e:
         logging.error(f"Video merge error: {str(e)}")
@@ -1126,7 +1136,8 @@ with demo:
                             interactive=False
                         )
         
-        # 네 번째 탭: 비디오 편집
+
+# 네 번째 탭: 비디오 편집 부분을 다음과 같이 수정
         with gr.Tab("비디오 편집", elem_classes="tabitem"):
             with gr.Row(equal_height=True):
                 # 입력 컬럼
@@ -1134,57 +1145,59 @@ with demo:
                     with gr.Group(elem_classes="panel-box"):
                         gr.Markdown("### 🎥 비디오 업로드 (최대 10개)")
                         gr.Markdown("**파일명이 작을수록 우선순위가 높습니다** (예: 1.mp4, 2.mp4, 3.mp4)")
-                        
+                
                         video_files = gr.File(
                             label="비디오 파일들",
                             file_count="multiple",
                             file_types=["video"],
                             type="filepath"
                         )
-                        
+                
                     with gr.Group(elem_classes="panel-box"):
                         gr.Markdown("### 🎵 오디오 설정 (선택)")
-                        gr.Markdown("**주의**: 업로드한 오디오가 비디오의 기존 오디오를 완전히 대체합니다.")
-                        
+                
+                # 오디오 모드 선택 추가
+                        audio_mode = gr.Radio(
+                            ["대체", "백그라운드 뮤직"],
+                            label="오디오 모드",
+                            value="대체",
+                            info="대체: 기존 오디오를 완전히 교체 | 백그라운드 뮤직: 기존 오디오와 함께 재생"
+                        )
+                
                         audio_file = gr.Audio(
                             label="오디오 파일 (MP3, WAV, M4A 등)",
                             type="filepath",
                             sources=["upload"]
                         )
-                        
+                
                         audio_volume = gr.Slider(
                             minimum=0,
                             maximum=200,
                             value=100,
                             step=1,
-                            label="오디오 볼륨 (%)",
+                            label="추가 오디오 볼륨 (%)",
                             info="100% = 원본 볼륨"
                         )
-                        
-                        gr.Markdown("""
-                        **오디오 옵션**:
-                        - 오디오가 비디오보다 짧으면 자��으로 반복됩니다
-                        - 오디오가 비디오보다 길면 비디오 길이에 맞춰 잘립니다
-                        """)
-                        
-                    with gr.Group(elem_classes="panel-box"):
-                        gr.Markdown("### ⚙️ 편집 설정")
-                        
-                        output_fps = gr.Slider(
+                
+                # 백그라운드 모드일 때만 보이는 원본 오디오 볼륨 조절
+                        original_audio_volume = gr.Slider(
                             minimum=0,
-                            maximum=60,
-                            value=0,
+                            maximum=200,
+                            value=100,
                             step=1,
-                            label="출력 FPS (0 = 첫 번째 비디오의 FPS 사용)"
+                            label="원본 오디오 볼륨 (%)",
+                            info="백그라운드 뮤직 모드에서 원본 비디오 오디오의 볼륨",
+                            visible=False
                         )
-                        
+                
                         gr.Markdown("""
-                        **크기 처리**:
-                        - 첫 번째 비디오의 크기가 기준이 됩니다
-                        - 다른 크기의 비디오는 첫 번째 비디오 크기로 조정됩니다
-                        - 최상의 결과를 위해 같은 크기의 비디오를 사용하세요
+                        **오디오 옵션**:
+                        - **대체 모드**: 업로드한 오디오가 비디오의 기존 오디오를 완전히 대체합니다
+                        - **백그라운드 뮤직 모드**: 업로드한 오디오가 기존 오디오와 함께 재생됩니다
+                        - 오디오가 비디오보다 짧으면 자동으로 반복됩니다
+                        - 오디오가 비디오보다 길면 비디오 길이에 맞춰 잘립니다
                         """)
-                        
+
                         merge_videos_btn = gr.Button("🎬 비디오 병합", variant="primary", elem_id="merge-btn")
                 
                 # 출력 컬럼
@@ -1362,12 +1375,22 @@ with demo:
     )
     
     # 이벤트 연결 - 네 번째 탭 (비디오 편집)
+
+    def toggle_original_volume(mode):
+        return gr.update(visible=(mode == "백그라운드 뮤직"))
+
+    audio_mode.change(
+        toggle_original_volume,
+        inputs=[audio_mode],
+        outputs=[original_audio_volume]
+    )
+
+
     merge_videos_btn.click(
         merge_videos_with_audio,
-        inputs=[video_files, audio_file, audio_volume, output_fps],
+        inputs=[video_files, audio_file, audio_mode, audio_volume, original_audio_volume, output_fps],
         outputs=[merged_video, merge_status]
-    )
-    
+    )    
     # 이벤트 연결 - 다섯 번째 탭 (비디오 배경제거/합성)
     def update_bg_visibility(bg_type):
         if bg_type == "색상":