Update app.py

app.py

@@ -627,7 +627,6 @@ def generate_section_video(prompt, preset, section_number=1, base_seed=171198, p
         raise gr.Error(f"섹션 {section_number} 생성 중 오류: {str(e)}")
 
 
-# 개별 섹션 프롬프트 생성 함수 추가
 def generate_single_section_prompt(scenario, section_number):
     """개별 섹션에 대한 프롬프트 생성"""
     section_descriptions = {
@@ -642,23 +641,35 @@ def generate_single_section_prompt(scenario, section_number):
         {"role": "system", "content": system_prompt_scenario},
         {"role": "user", "content": f"""
 다음 스크립트의 {section_number}번째 섹션({section_descriptions[section_number]})에 대한 
-배경 영상 프롬프트만을 생성해주세요:
+배경 영상 프롬프트를 생성해주세요.
 
+스크립트:
 {scenario}
 
-직접적인 제품 묘사는 피하고, 스크립트의 주제와 감성을 표현하는 핵심 키워드를 반영한 배경 영상에 집중해주세요."""}
+주의사항:
+1. 해당 섹션의 특성({section_descriptions[section_number]})에 맞는 분위기와 톤을 반영하세요.
+2. 직접적인 제품/서비스 묘사는 피하고, 감성적이고 은유적인 배경 영상에 집중하세요.
+3. 다음 구조를 반드시 포함하세요:
+   - 주요 동작을 명확한 한 문장으로 시작
+   - 구체적인 동작과 제스처를 시간 순서대로 설명
+   - 배경과 환경 세부 사항을 구체적으로 포함
+   - 카메라 각도와 움직임을 명시
+   - 조명과 색상을 자세히 설명
+   - 변화나 갑작스러운 사건을 자연스럽게 포함"""}
     ]
     
     try:
         response = client.chat.completions.create(
             model="gpt-4-1106-preview",
             messages=messages,
-            max_tokens=500,
+            max_tokens=1000,  # 토큰 수 증가
+            temperature=0.7
         )
-        return response.choices[0].message.content.strip()
+        generated_prompt = response.choices[0].message.content.strip()
+        return f"{section_number}. {generated_prompt}"
     except Exception as e:
-        print(f"Error during prompt generation: {e}")
-        return "Error occurred during prompt generation"
+        print(f"Error during prompt generation for section {section_number}: {e}")
+        return f"Error occurred during prompt generation for section {section_number}"
 
 
 # 비디오 결합 함수 추가
@@ -996,6 +1007,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
 
     
 # Event handlers
+    # Text to Video Tab handlers
     txt2vid_preset.change(
         fn=preset_changed,
         inputs=[txt2vid_preset],
@@ -1031,6 +1043,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         queue=True,
     )
 
+    # Image to Video Tab handlers
     img2vid_preset.change(
         fn=preset_changed,
         inputs=[img2vid_preset],
@@ -1067,7 +1080,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         queue=True,
     )
 
-    # Scenario tab event handlers
+    # Scenario Tab handlers
+    # 스크립트 생성 버튼 핸들러
+    generate_script_btn.click(
+        fn=generate_script,
+        inputs=[script_topic],
+        outputs=[scenario_input]
+    )
+
+    # 시나리오 분석 버튼 핸들러
     analyze_btn.click(
         fn=analyze_scenario,
         inputs=[scenario_input],
@@ -1077,45 +1098,38 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         ]
     )
 
-# 섹션 생성 이벤트 핸들러
-    section1_generate.click(
-        fn=generate_section_video,
-        inputs=[section1_prompt, scenario_preset],
-        outputs=section1_video,
-        api_name=f"generate_section1"
+    # 섹션별 프롬프트 재생성 핸들러
+    section1_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 1),
+        inputs=[scenario_input],
+        outputs=section1_prompt
     )
 
-    section2_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 2),
-        inputs=[section2_prompt, scenario_preset],
-        outputs=section2_video,
-        api_name=f"generate_section2"
+    section2_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 2),
+        inputs=[scenario_input],
+        outputs=section2_prompt
     )
 
-    section3_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 3),
-        inputs=[section3_prompt, scenario_preset],
-        outputs=section3_video,
-        api_name=f"generate_section3"
+    section3_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 3),
+        inputs=[scenario_input],
+        outputs=section3_prompt
     )
 
-    section4_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 4),
-        inputs=[section4_prompt, scenario_preset],
-        outputs=section4_video,
-        api_name=f"generate_section4"
+    section4_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 4),
+        inputs=[scenario_input],
+        outputs=section4_prompt
     )
 
-    section5_generate.click(
-        fn=lambda p, pr: generate_section_video(p, pr, 5),
-        inputs=[section5_prompt, scenario_preset],
-        outputs=section5_video,
-        api_name=f"generate_section5"
+    section5_regenerate.click(
+        fn=lambda x: generate_single_section_prompt(x, 5),
+        inputs=[scenario_input],
+        outputs=section5_prompt
     )
 
-
-
-    # 섹션 생성 이벤트 핸들러
+    # 섹션별 비디오 생성 핸들러
     section1_generate.click(
         fn=lambda p, pr: generate_section_video(p, pr, 1),
         inputs=[section1_prompt, scenario_preset],
@@ -1146,8 +1160,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         outputs=section5_video
     )
 
-
-# 이벤트 핸들러 추가
+    # 통합 영상 생성 핸들러
     merge_videos_btn.click(
         fn=merge_section_videos,
         inputs=[
@@ -1159,13 +1172,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         ],
         outputs=merged_video_output
     )
-
-# 스크립트 생성 버튼 이벤트 핸들러
-    generate_script_btn.click(
-        fn=generate_script,
-        inputs=[script_topic],
-        outputs=[scenario_input]
-    )
     
 if __name__ == "__main__":
     iface.queue(max_size=64, default_concurrency_limit=1, api_open=False).launch(