Update app.py

app.py

@@ -604,6 +604,51 @@ def generate_single_section_prompt(scenario, section_number):
         return "Error occurred during prompt generation"
 
 
+# 비디오 결합 함수 추가
+def combine_videos(video_paths, output_path):
+    """여러 비디오를 하나로 결합"""
+    if not all(video_paths):
+        raise gr.Error("모든 섹션의 영상이 생성되어야 합니다.")
+        
+    try:
+        # 첫 번째 비디오의 속성 가져오기
+        cap = cv2.VideoCapture(video_paths[0])
+        fps = int(cap.get(cv2.CAP_PROP_FPS))
+        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        cap.release()
+
+        # 출력 비디오 설정
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+
+        # 각 비디오 순차적으로 결합
+        for video_path in video_paths:
+            if video_path and os.path.exists(video_path):
+                cap = cv2.VideoCapture(video_path)
+                while True:
+                    ret, frame = cap.read()
+                    if not ret:
+                        break
+                    out.write(frame)
+                cap.release()
+
+        out.release()
+        return output_path
+    except Exception as e:
+        raise gr.Error(f"비디오 결합 중 오류 발생: {e}")
+
+def merge_section_videos(section1, section2, section3, section4, section5):
+    """섹션 비디오들을 하나로 결합"""
+    videos = [section1, section2, section3, section4, section5]
+    
+    if not all(videos):
+        raise gr.Error("모든 섹션의 영상이 먼저 생성되어야 합니다.")
+    
+    output_path = tempfile.mktemp(suffix=".mp4")
+    return combine_videos(videos, output_path)
+
+
 # Gradio Interface Definition
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     with gr.Tabs():
@@ -735,19 +780,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                                 lines=4
                             )
                             with gr.Row():
-                                section1_regenerate = gr.Button("🔄 프롬프트 새로 생성")
-                                section1_generate = gr.Button("생성")
+                                section1_regenerate = gr.Button("🔄 프롬프트 생성")
+                                section1_generate = gr.Button("🔄 영상 생성")
                             section1_video = gr.Video(label="섹션 1 영상")
                         
                         # 섹션 2
                         with gr.Column():
                             section2_prompt = gr.Textbox(
-                                label="2. 문제 제기 및 흥미 유발",
+                                label="2. 흥미 유발",
                                 lines=4
                             )
                             with gr.Row():
-                                section2_regenerate = gr.Button("🔄 프롬프트 새로 생성")
-                                section2_generate = gr.Button("생성")
+                                section2_regenerate = gr.Button("🔄 프롬프트 생성")
+                                section2_generate = gr.Button("🔄 영상 생성")
                             section2_video = gr.Video(label="섹션 2 영상")
                     
                     with gr.Row():
@@ -758,8 +803,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                                 lines=4
                             )
                             with gr.Row():
-                                section3_regenerate = gr.Button("🔄 프롬프트 새로 생성")
-                                section3_generate = gr.Button("생성")
+                                section3_regenerate = gr.Button("🔄 프롬프트 생성")
+                                section3_generate = gr.Button("🔄 영상 생성")
                             section3_video = gr.Video(label="섹션 3 영상")
                         
                         # 섹션 4
@@ -769,8 +814,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                                 lines=4
                             )
                             with gr.Row():
-                                section4_regenerate = gr.Button("🔄 프롬프트 새로 생성")
-                                section4_generate = gr.Button("생성")
+                                section4_regenerate = gr.Button("🔄 프롬프트 생성")
+                                section4_generate = gr.Button("🔄 영상 생성")
                             section4_video = gr.Video(label="섹션 4 영상")
                     
                     with gr.Row():
@@ -781,10 +826,28 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                                 lines=4
                             )
                             with gr.Row():
-                                section5_regenerate = gr.Button("🔄 프롬프트 새로 생성")
-                                section5_generate = gr.Button("생성")
+                                section5_regenerate = gr.Button("🔄 프롬프트 생성")
+                                section5_generate = gr.Button("🔄 영상 생성")
                             section5_video = gr.Video(label="섹션 5 영상")
 
+
+    
+    # 통합 영상 섹션 추가
+                    with gr.Row():
+                        with gr.Column(scale=1):
+            # 기존의 scenario_input과 analyze_btn 유지
+                            merge_videos_btn = gr.Button("통합 영상 생성", variant="primary", size="lg")
+        
+                        with gr.Column(scale=2):
+            # 기존의 섹션 1-5 유지
+            
+            # 통합 영상 출력 섹션 추가
+                            with gr.Row():
+                                merged_video_output = gr.Video(label="통합 영상")
+
+
+
+    
 # Event handlers
     txt2vid_preset.change(
         fn=preset_changed,
@@ -929,6 +992,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         outputs=section5_video
     )
 
+
+# 이벤트 핸들러 추가
+    merge_videos_btn.click(
+        fn=merge_section_videos,
+        inputs=[
+            section1_video,
+            section2_video,
+            section3_video,
+            section4_video,
+            section5_video
+        ],
+        outputs=merged_video_output
+    )
+
     
 if __name__ == "__main__":
     iface.queue(max_size=64, default_concurrency_limit=1, api_open=False).launch(