Update app.py

app.py

@@ -19,7 +19,7 @@ from transformers import pipeline
 # 비디오 배경제거를 위한 추가 import
 from transformers import AutoModelForImageSegmentation
 from torchvision import transforms
-from moviepy import VideoFileClip, vfx, concatenate_videoclips, ImageSequenceClip
+from moviepy import VideoFileClip, vfx, concatenate_videoclips, ImageSequenceClip, concatenate_audioclips, AudioFileClip, CompositeAudioClip
 import time
 from concurrent.futures import ThreadPoolExecutor
 
@@ -564,6 +564,106 @@ def process_video_bg(vid, bg_type="색상", bg_image=None, bg_video=None, color=
         yield gr.update(visible=False), gr.update(visible=True), f"비디오 처리 오류: {e}. 경과 시간: {elapsed_time:.2f}초"
         yield None, f"비디오 처리 오류: {e}", f"비디오 처리 오류: {e}. 경과 시간: {elapsed_time:.2f}초"
 
+@spaces.GPU
+def merge_videos_with_audio(video_files, audio_file, audio_volume, output_fps):
+    """여러 비디오를 병합하고 오디오를 추가하는 함수"""
+    if not video_files:
+        return None, "비디오 파일을 업로드해주세요."
+    
+    if len(video_files) > 10:
+        return None, "최대 10개의 비디오만 업로드 가능합니다."
+    
+    try:
+        # 상태 업데이트
+        status = "비디오 파일 정렬 중..."
+        
+        # 파일 경로와 파일명을 튜플로 저장하고 파일명으로 정렬
+        video_paths = []
+        if isinstance(video_files, list):
+            for video_file in video_files:
+                if video_file is not None:
+                    video_paths.append(video_file)
+        else:
+            video_paths.append(video_files)
+        
+        # 파일명으로 정렬 (경로에서 파일명만 추출하여 정렬)
+        video_paths.sort(key=lambda x: os.path.basename(x))
+        
+        status = f"{len(video_paths)}개의 비디오 로드 중..."
+        
+        # 비디오 클립 로드
+        video_clips = []
+        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)
+        
+        # 첫 번째 비디오의 FPS를 기본값으로 사용
+        if output_fps == 0:
+            output_fps = video_clips[0].fps
+        
+        status = "비디오 병합 중..."
+        
+        # 비디오 병합
+        final_video = concatenate_videoclips(video_clips, method="compose")
+        
+        # 오디오 처리
+        if audio_file:
+            status = "오디오 처리 중..."
+            
+            # 오디오 로드
+            audio_clip = VideoFileClip(audio_file).audio if audio_file.endswith(('.mp4', '.avi', '.mov')) else None
+            
+            if audio_clip is None:
+                # 오디오 파일인 경우
+                audio_clip = AudioFileClip(audio_file)
+            
+            # 볼륨 조절
+            if audio_volume != 100:
+                audio_clip = audio_clip.volumex(audio_volume / 100)
+            
+            # 오디오를 비디오 길이에 맞춤
+            video_duration = final_video.duration
+            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_clip = concatenate_audioclips([audio_clip] * loops_needed).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)
+        
+        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"
+            )
+        
+        # 리소스 정리
+        for clip in video_clips:
+            clip.close()
+        final_video.close()
+        
+        return temp_filepath, f"✅ 성공적으로 {len(video_paths)}개의 비디오를 병합했습니다!"
+        
+    except Exception as e:
+        logging.error(f"Video merge error: {str(e)}")
+        return None, f"❌ 오류 발생: {str(e)}"
+
 # CSS
 css = """
 :root {
@@ -589,7 +689,7 @@ css = """
     padding: 20px !important;
     margin-bottom: 20px !important;
 }
-#generate-btn, #video-btn, #outpaint-btn, #preview-btn, #audio-btn, #bg-remove-btn {
+#generate-btn, #video-btn, #outpaint-btn, #preview-btn, #audio-btn, #bg-remove-btn, #merge-btn {
     background: linear-gradient(135deg, #ff9a9e, #fad0c4) !important;
     font-size: 1.1rem !important;
     padding: 12px 24px !important;
@@ -786,7 +886,72 @@ with demo:
                         if not MMAUDIO_MODEL_LOADED:
                             gr.Markdown("⚠️ MMAudio 모델을 로드하지 못했습니다. 이 기능은 사용할 수 없습니다.")
         
-        # 네 번째 탭: 비디오 배경제거/합성
+        # 네 번째 탭: 비디오 편집
+        with gr.Tab("비디오 편집", elem_classes="tabitem"):
+            with gr.Row(equal_height=True):
+                # 입력 컬럼
+                with gr.Column(scale=1):
+                    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("### 🎵 오디오 설정 (선택)")
+                        
+                        audio_file = gr.Audio(
+                            label="오디오 파일",
+                            type="filepath",
+                            sources=["upload"]
+                        )
+                        
+                        audio_volume = gr.Slider(
+                            minimum=0,
+                            maximum=200,
+                            value=100,
+                            step=1,
+                            label="오디오 볼륨 (%)",
+                            info="100% = 원본 볼륨"
+                        )
+                        
+                    with gr.Group(elem_classes="panel-box"):
+                        gr.Markdown("### ⚙️ 편집 설정")
+                        
+                        output_fps = gr.Slider(
+                            minimum=0,
+                            maximum=60,
+                            value=0,
+                            step=1,
+                            label="출력 FPS (0 = 첫 번째 비디오의 FPS 사용)"
+                        )
+                        
+                        merge_videos_btn = gr.Button("🎬 비디오 병합", variant="primary", elem_id="merge-btn")
+                
+                # 출력 컬럼
+                with gr.Column(scale=1):
+                    with gr.Group(elem_classes="panel-box"):
+                        gr.Markdown("### 🎬 병합 결과")
+                        
+                        merge_status = gr.Textbox(label="처리 상태", interactive=False)
+                        merged_video = gr.Video(label="병합된 비디오")
+                        
+                        gr.Markdown("""
+                        ### ℹ️ 사용 방법
+                        1. 여러 비디오 파일을 업로드하세요 (최대 10개)
+                        2. 파일명이 작은 순서대로 자동 정렬됩니다
+                        3. (선택) 오디오 파일을 추가하고 볼륨을 조절하세요
+                        4. '비디오 병합' 버튼을 클릭하세요
+                        
+                        **팁**: 파일명을 01.mp4, 02.mp4, 03.mp4 형식으로 지정하면 순서 관리가 쉽습니다.
+                        """)
+        
+        # 다섯 번째 탭: 비디오 배경제거/합성
         with gr.Tab("비디오 배경제거/합성", elem_classes="tabitem"):
             with gr.Row(equal_height=True):
                 # 입력 컬럼
@@ -927,7 +1092,14 @@ with demo:
         [output_video_with_audio]
     )
     
-    # 이벤트 연결 - 네 번째 탭
+    # 이벤트 연결 - 네 번째 탭 (비디오 편집)
+    merge_videos_btn.click(
+        merge_videos_with_audio,
+        inputs=[video_files, audio_file, audio_volume, output_fps],
+        outputs=[merged_video, merge_status]
+    )
+    
+    # 이벤트 연결 - 다섯 번째 탭 (비디오 배경제거/합성)
     def update_bg_visibility(bg_type):
         if bg_type == "색상":
             return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)