Update app-backup1.py

app-backup1.py

@@ -557,36 +557,67 @@ def analyze_scenario(scenario):
         {"role": "user", "content": f"""
 다음 스크립트의 분위기와 감성을 표현할 수 있는 배경 영상 프롬프트를 생성해주세요:
 
+주어진 스크립트의 분위기와 맥락을 시각적 배경으로 표현하되, 다음 원칙을 반드시 준수하세요:
+
+1. 제품이나 서비스를 직접적으로 묘사하지 말 것
+2. 스크립트의 감성과 톤앤매너를 표현하는 배경 영상에 집중할 것
+3. 5개 섹션이 하나의 이야기처럼 자연스럽게 연결되도록 할 것
+4. 추상적이고 은유적인 시각 표현을 활용할 것
+
+각 섹션별 프롬프트 작성 가이드:
+1. 배경 및 필요성: 주제의 전반적인 분위기를 표현하는 배경 씬
+2. 흥미 유발: 긴장감이나 갈등을 암시하는 분위기 있는 배경
+3. 해결책 제시: 희망적이고 밝은 톤의 배경 전환
+4. 본론: 안정감 있고 신뢰도를 높이는 배경
+5. 결론: 임팩트 있는 마무리를 위한 역동적인 배경
+
+모든 섹션이 일관된 스타일과 톤을 유지하면서도 자연스럽게 이어지도록 구성하세요.
+
+각 섹션의 프롬프트 작성시 반드시 다음 구조에 맞게 개선해주세요:
+1. 주요 동작을 명확한 한 문장으로 시작
+2. 구체적인 동작과 제스처를 시간 순서대로 설명
+3. 캐릭터/객체의 외모를 상세히 묘사
+4. 배경과 환경 세부 사항을 구체적으로 포함
+5. 카메라 각도와 움직임을 명시
+6. 조명과 색상을 자세히 설명
+7. 변화나 갑작스러운 사건을 자연스럽게 포함
+모든 설명은 하나의 자연스러운 문단으로 작성하고, 
+촬영 감독이 촬영 목록을 설명하는 것처럼 구체적이고 시각적으로 작성하세요.
+200단어를 넘지 않도록 하되, 최대한 상세하게 작성하세요.
+
 {scenario}
 
 각 섹션별로 직접적인 제품 묘사는 피하고, 스크립트의 감성을 표현하는 배경 영상에 집중해주세요."""},
     ]
     
-    try:
+        
         response = client.chat.completions.create(
             model="gpt-4-1106-preview",
             messages=messages,
             max_tokens=2000,
         )
-        prompts = response.choices[0].message.content.strip().split("\n\n")
+        content = response.choices[0].message.content.strip()
         
-        # 프롬프트 처리 로직은 동일
-        section_prompts = []
+        # 섹션별로 분리
+        sections = []
         current_section = ""
-        for line in prompts:
-            if line.strip():
-                if any(section in line for section in ["1.", "2.", "3.", "4.", "5."]):
-                    if current_section:
-                        section_prompts.append(current_section)
-                    current_section = line
-                else:
-                    current_section += "\n" + line
+        for line in content.split('\n'):
+            if line.strip().startswith(('1.', '2.', '3.', '4.', '5.')):
+                if current_section:
+                    sections.append(current_section.strip())
+                current_section = line
+            else:
+                current_section += "\n" + line
+                
         if current_section:
-            section_prompts.append(current_section)
+            sections.append(current_section.strip())
             
-        while len(section_prompts) < 5:
-            section_prompts.append("추가 섹션이 필요합니다.")
-        return section_prompts[:5]
+        # 부족한 섹션 채우기
+        while len(sections) < 5:
+            sections.append("추가 섹션이 필요합니다.")
+            
+        return sections[:5]
+        
     except Exception as e:
         print(f"Error during scenario analysis: {e}")
         return ["Error occurred during analysis"] * 5
@@ -612,7 +643,6 @@ def generate_section_video(prompt, preset, section_number=1, base_seed=171198, p
         print(f"Error in section {section_number}: {e}")
         raise gr.Error(f"섹션 {section_number} 생성 중 오류: {str(e)}")
 
-
 # 개별 섹션 프롬프트 생성 함수 추가
 def generate_single_section_prompt(scenario, section_number):
     """개별 섹션에 대한 프롬프트 생성"""
@@ -683,13 +713,50 @@ def combine_videos(video_paths, output_path):
 
 def merge_section_videos(section1, section2, section3, section4, section5):
     """섹션 비디오들을 하나로 결합"""
-    videos = [section1, section2, section3, section4, section5]
+    videos = []
     
-    if not all(videos):
-        raise gr.Error("모든 섹션의 영상이 먼저 생성되어야 합니다.")
+    # 각 섹션 비디오 확인 및 처리
+    for i, video in enumerate([section1, section2, section3, section4, section5], 1):
+        if video and os.path.exists(video):
+            videos.append(video)
+        else:
+            raise gr.Error(f"섹션 {i}의 영상이 없거나 접근할 수 없습니다.")
     
-    output_path = tempfile.mktemp(suffix=".mp4")
-    return combine_videos(videos, output_path)
+    if not videos:
+        raise gr.Error("결합할 영상이 없습니다.")
+    
+    try:
+        output_path = tempfile.mktemp(suffix=".mp4")
+        
+        # 첫 번째 비디오의 속성 가져오기
+        cap = cv2.VideoCapture(videos[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 videos:
+            cap = cv2.VideoCapture(video_path)
+            while True:
+                ret, frame = cap.read()
+                if not ret:
+                    break
+                # 필요한 경우 프레임 크기 조정
+                if frame.shape[:2] != (height, width):
+                    frame = cv2.resize(frame, (width, height))
+                out.write(frame)
+            cap.release()
+
+        out.release()
+        return output_path
+        
+    except Exception as e:
+        raise gr.Error(f"비디오 결합 중 오류 발생: {e}")
 
 
 # Gradio Interface Definition
@@ -973,6 +1040,39 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         ]
     )
 
+
+# 각 섹션의 프롬프트 재생성 이벤트 핸들러 추가
+    section1_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 1),
+        inputs=[scenario_input],
+        outputs=section1_prompt
+    )
+    
+    section2_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 2),
+        inputs=[scenario_input],
+        outputs=section2_prompt
+    )
+    
+    section3_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 3),
+        inputs=[scenario_input],
+        outputs=section3_prompt
+    )
+    
+    section4_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 4),
+        inputs=[scenario_input],
+        outputs=section4_prompt
+    )
+    
+    section5_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 5),
+        inputs=[scenario_input],
+        outputs=section5_prompt
+    )
+
+
 # 섹션 생성 이벤트 핸들러
     section1_generate.click(
         fn=generate_section_video,
@@ -1011,36 +1111,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
 
 
 
-    # 섹션 생성 이벤트 핸들러
-    section1_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 1),
-        inputs=[section1_prompt, scenario_preset],
-        outputs=section1_video
-    )
-
-    section2_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 2),
-        inputs=[section2_prompt, scenario_preset],
-        outputs=section2_video
-    )
-
-    section3_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 3),
-        inputs=[section3_prompt, scenario_preset],
-        outputs=section3_video
-    )
-
-    section4_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 4),
-        inputs=[section4_prompt, scenario_preset],
-        outputs=section4_video
-    )
-
-    section5_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 5),
-        inputs=[section5_prompt, scenario_preset],
-        outputs=section5_video
-    )
 
 
 # 이벤트 핸들러 추가