Update app.py

app.py

@@ -35,6 +35,8 @@ from transformers import pipeline
 from transformers import AutoModelForImageSegmentation
 from torchvision import transforms
 from moviepy import VideoFileClip, vfx, concatenate_videoclips, ImageSequenceClip, concatenate_audioclips, AudioFileClip, CompositeAudioClip
+from moviepy.video.compositing.CompositeVideoClip import CompositeVideoClip
+from moviepy.video.VideoClip import ColorClip
 import time
 from concurrent.futures import ThreadPoolExecutor
 
@@ -628,12 +630,46 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
         
         status = f"{len(video_paths)}개의 비디오 로드 중..."
         
-        # 비디오 클립 로드
+        # 비디오 클립 로드 및 크기 확인
         video_clips = []
+        max_width = 0
+        max_height = 0
+        
         for i, video_path in enumerate(video_paths):
             status = f"비디오 {i+1}/{len(video_paths)} 로드 중: {os.path.basename(video_path)}"
             clip = VideoFileClip(video_path)
             video_clips.append(clip)
+            
+            # 최대 크기 추적
+            if clip.w > max_width:
+                max_width = clip.w
+            if clip.h > max_height:
+                max_height = clip.h
+        
+        # 임시 변수 정리를 위한 배경 클립 저장
+        background_clips = []
+        
+        # 모든 비디오를 최대 크기로 조정 (원본 품질 유지)
+        resized_clips = []
+        for clip in video_clips:
+            if clip.w != max_width or clip.h != max_height:
+                # 패딩 추가 방식 (원본 품질 유지)
+                # 검은색 배경 생성
+                background = ColorClip(size=(max_width, max_height), color=(0,0,0), duration=clip.duration)
+                background_clips.append(background)
+                
+                # 클립을 중앙에 배치
+                x_center = (max_width - clip.w) // 2
+                y_center = (max_height - clip.h) // 2
+                
+                resized_clip = CompositeVideoClip([
+                    background,
+                    clip.set_position((x_center, y_center))
+                ])
+                
+                resized_clips.append(resized_clip)
+            else:
+                resized_clips.append(clip)
         
         # 첫 번째 비디오의 FPS를 기본값으로 사용
         if output_fps == 0:
@@ -642,22 +678,31 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
         status = "비디오 병합 중..."
         
         # 비디오 병합
-        final_video = concatenate_videoclips(video_clips, method="compose")
+        final_video = concatenate_videoclips(resized_clips, method="compose")
         
         # 오디오 처리
         if audio_file:
             status = "오디오 처리 중..."
             
             try:
+                # 오디오 파일 경로 확인
+                if isinstance(audio_file, str):
+                    audio_path = audio_file
+                else:
+                    # gr.Audio에서 반환된 튜플인 경우
+                    audio_path = audio_file
+                
+                logging.info(f"Processing audio from: {audio_path}")
+                
                 # 오디오 로드
-                if audio_file.endswith(('.mp4', '.avi', '.mov', '.mkv')):
+                if audio_path.endswith(('.mp4', '.avi', '.mov', '.mkv')):
                     # 비디오 파일에서 오디오 추출
-                    temp_video = VideoFileClip(audio_file)
+                    temp_video = VideoFileClip(audio_path)
                     audio_clip = temp_video.audio
                     temp_video.close()
                 else:
                     # 오디오 파일 직접 로드
-                    audio_clip = AudioFileClip(audio_file)
+                    audio_clip = AudioFileClip(audio_path)
                 
                 if audio_clip is None:
                     raise ValueError("오디오를 로드할 수 없습니다.")
@@ -680,38 +725,59 @@ def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
                     looped_audio = concatenate_audioclips(audio_clips_list)
                     audio_clip = looped_audio.subclip(0, video_duration)
                 
-                # 기존 오디오와 새 오디오 합성 (기존 오디오가 있는 경우)
-                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)
+                # 기존 오디오 제거하고 새 오디오로 교체
+                # (기존 오디오와 합성하려면 아래 주석 해제)
+                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)
+                
+                logging.info("Audio successfully added to video")
                     
             except Exception as e:
-                logging.warning(f"오디오 처리 중 오류 발생: {str(e)}")
+                logging.error(f"오디오 처리 중 오류 발생: {str(e)}")
                 # 오디오 처리 실패해도 비디오는 계속 처리
+                status = f"오디오 처리 실패: {str(e)}, 비디오만 병합합니다."
         
         status = "비디오 저장 중..."
         
         # 임시 파일로 저장
         with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_file:
             temp_filepath = temp_file.name
+            
+            # 코덱 설정 - 원본 품질 유지
             final_video.write_videofile(
                 temp_filepath, 
                 fps=output_fps,
                 codec="libx264",
-                audio_codec="aac"
+                audio_codec="aac",
+                preset="medium",  # 품질 설정
+                bitrate="5000k",  # 비트레이트 설정으로 품질 유지
+                audio_bitrate="192k"
             )
         
         # 리소스 정리
         for clip in video_clips:
             clip.close()
+        for clip in background_clips:
+            clip.close()
+        for i, clip in enumerate(resized_clips):
+            if clip != video_clips[i] if i < len(video_clips) else True:  # 리사이즈된 경우만
+                clip.close()
+        if audio_file and 'audio_clip' in locals():
+            audio_clip.close()
         final_video.close()
         
-        return temp_filepath, f"✅ 성공적으로 {len(video_paths)}개의 비디오를 병합했습니다!"
+        return temp_filepath, f"✅ 성공적으로 {len(video_paths)}개의 비디오를 병합했습니다! (크기: {max_width}x{max_height})"
         
     except Exception as e:
         logging.error(f"Video merge error: {str(e)}")
+        import traceback
+        traceback.print_exc()
         return None, f"❌ 오류 발생: {str(e)}"
 
 # CSS
@@ -955,9 +1021,10 @@ with demo:
                         
                     with gr.Group(elem_classes="panel-box"):
                         gr.Markdown("### 🎵 오디오 설정 (선택)")
+                        gr.Markdown("**주의**: 업로드한 오디오가 비디오의 기존 오디오를 완전히 대체합니다.")
                         
                         audio_file = gr.Audio(
-                            label="오디오 파일",
+                            label="오디오 파일 (MP3, WAV, M4A 등)",
                             type="filepath",
                             sources=["upload"]
                         )
@@ -971,6 +1038,12 @@ with demo:
                             info="100% = 원본 볼륨"
                         )
                         
+                        gr.Markdown("""
+                        **오디오 옵션**:
+                        - 오디오가 비디오보다 짧으면 자동으로 반복됩니다
+                        - 오디오가 비디오보다 길면 비디오 길이에 맞춰 잘립니다
+                        """)
+                        
                     with gr.Group(elem_classes="panel-box"):
                         gr.Markdown("### ⚙️ 편집 설정")
                         
@@ -982,6 +1055,13 @@ with demo:
                             label="출력 FPS (0 = 첫 번째 비디오의 FPS 사용)"
                         )
                         
+                        gr.Markdown("""
+                        **크기 처리**:
+                        - 모든 비디오는 가장 큰 비디오의 크기에 맞춰집니다
+                        - 작은 비디오는 검은색 패딩이 추가되어 중앙에 배치됩니다
+                        - 원본 화질이 그대로 유지됩니다
+                        """)
+                        
                         merge_videos_btn = gr.Button("🎬 비디오 병합", variant="primary", elem_id="merge-btn")
                 
                 # 출력 컬럼
@@ -999,7 +1079,14 @@ with demo:
                         3. (선택) 오디오 파일을 추가하고 볼륨을 조절하세요
                         4. '비디오 병합' 버튼을 클릭하세요
                         
-                        **팁**: 파일명을 01.mp4, 02.mp4, 03.mp4 형식으로 지정하면 순서 관리가 쉽습니다.
+                        **특징**:
+                        - ✅ 작은 비디오는 패딩 추가로 원본 화질 유지
+                        - ✅ 업로드한 오디오가 전체 비디오에 적용됩니다
+                        - ✅ 원본 화질이 최대한 유지됩니다
+                        
+                        **팁**: 
+                        - 파일명을 01.mp4, 02.mp4, 03.mp4 형식으로 지정하면 순서 관리가 쉽습니다
+                        - 오디오를 추가하면 기존 비디오의 오디오는 대체됩니다
                         """)
         
         # 다섯 번째 탭: 비디오 배경제거/합성