Update app.py

app.py

@@ -491,6 +491,74 @@ def create_advanced_options():
             num_frames_slider,
         ]
 
+
+# 시나리오 분석 및 프롬프트 생성을 위한 시스템 프롬프트
+system_prompt_scenario = """당신은 영상 시나리오를 5개의 섹션으로 나누고 각각에 대한 비디오 생성 프롬프트를 작성하는 전문가입니다.
+주어진 시나리오를 다음 5개 섹션으로 나누어 각각의 프롬프트를 생성해주세요:
+
+1. 배경 및 필요성: 주제의 배경과 중요성을 시각적으로 표현
+2. 문제 제기 및 흥미 유발: 핵심 문제나 흥미로운 포인트를 드라마틱하게 표현
+3. 해결책 제시: 주요 해결방안이나 접근법을 시각적으로 제시
+4. 본론: 핵심 내용을 상세히 설명하는 시각적 표현
+5. 결론 및 강조: 주요 포인트를 다시 한번 강조하고 마무리
+
+각 섹션의 프롬프트는 다음 요소를 포함해야 합니다:
+- 주요 시각적 요소와 동작
+- 카메라 움직임과 앵글
+- 장면 전환과 효과
+- 분위기와 톤
+- 등장 요소들의 세부 묘사
+
+각 섹션은 10초 분량의 영상을 생성할 수 있도록 구체적이고 시각적인 설명을 포함해야 합니다."""
+
+def analyze_scenario(scenario):
+    """시나리오를 분석하여 5개의 섹션으로 나누고 각각의 프롬프트를 생성"""
+    messages = [
+        {"role": "system", "content": system_prompt_scenario},
+        {"role": "user", "content": 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")
+        
+        # 5개의 섹션으로 정리
+        section_prompts = []
+        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
+        if current_section:
+            section_prompts.append(current_section)
+            
+        # 정확히 5개의 섹션이 되도록 조정
+        while len(section_prompts) < 5:
+            section_prompts.append("추가 섹션이 필요합니다.")
+        return section_prompts[:5]
+    except Exception as e:
+        print(f"Error during scenario analysis: {e}")
+        return ["Error occurred during analysis"] * 5
+
+def generate_section_video(prompt, preset, progress=gr.Progress()):
+    """각 섹션의 비디오 생성"""
+    selected = next(item for item in preset_options if item["label"] == preset)
+    return generate_video_from_text(
+        prompt=prompt,
+        height=selected["height"],
+        width=selected["width"],
+        num_frames=selected["num_frames"],
+        progress=progress
+    )
+    
 # Gradio Interface Definition
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     with gr.Tabs():
@@ -518,14 +586,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         visible=False
                     )
 
-
                     txt2vid_preset = gr.Dropdown(
                         choices=[p["label"] for p in preset_options],
-                        value="[16:9] 512x320, 10.3초",  # 정확한 라벨로 수정
+                        value="[16:9] 512x320, 10.3초",
                         label="Step 2: 해상도 프리셋 선택",
                     )
 
-
                     txt2vid_frame_rate = gr.Slider(
                         label="Step 3: 프레임 레이트",
                         minimum=21,
@@ -573,14 +639,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         visible=False
                     )
 
-
                     img2vid_preset = gr.Dropdown(
                         choices=[p["label"] for p in preset_options],
-                        value="[16:9] 512x320, 10.3초",  # 정확한 라벨로 수정
+                        value="[16:9] 512x320, 10.3초",
                         label="Step 3: 해상도 프리셋 선택",
-                    )                    
-                    
-
+                    )
 
                     img2vid_frame_rate = gr.Slider(
                         label="Step 4: 프레임 레이트",
@@ -601,6 +664,72 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                 with gr.Column():
                     img2vid_output = gr.Video(label="생성된 비디오")
 
+        # Scenario to Video Tab (New)
+        with gr.TabItem("시나리오로 비디오 만들기(숏폼)"):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    scenario_input = gr.Textbox(
+                        label="영상 스크립트 입력",
+                        placeholder="전체 시나리오를 입력하세요...",
+                        lines=10
+                    )
+                    scenario_preset = gr.Dropdown(
+                        choices=[p["label"] for p in preset_options],
+                        value="[16:9] 512x320, 10.3초",
+                        label="화면 크기 선택"
+                    )
+                    analyze_btn = gr.Button("시나리오 분석 및 프롬프트 생성", variant="primary")
+                
+                with gr.Column(scale=2):
+                    with gr.Row():
+                        # 섹션 1
+                        with gr.Column():
+                            section1_prompt = gr.Textbox(
+                                label="1. 배경 및 필요성",
+                                lines=4
+                            )
+                            section1_generate = gr.Button("생성")
+                            section1_video = gr.Video(label="섹션 1 영상")
+                        
+                        # 섹션 2
+                        with gr.Column():
+                            section2_prompt = gr.Textbox(
+                                label="2. 문제 제기 및 흥미 유발",
+                                lines=4
+                            )
+                            section2_generate = gr.Button("생성")
+                            section2_video = gr.Video(label="섹션 2 영상")
+                    
+                    with gr.Row():
+                        # 섹션 3
+                        with gr.Column():
+                            section3_prompt = gr.Textbox(
+                                label="3. 해결책 제시",
+                                lines=4
+                            )
+                            section3_generate = gr.Button("생성")
+                            section3_video = gr.Video(label="섹션 3 영상")
+                        
+                        # 섹션 4
+                        with gr.Column():
+                            section4_prompt = gr.Textbox(
+                                label="4. 본론",
+                                lines=4
+                            )
+                            section4_generate = gr.Button("생성")
+                            section4_video = gr.Video(label="섹션 4 영상")
+                    
+                    with gr.Row():
+                        # 섹션 5
+                        with gr.Column():
+                            section5_prompt = gr.Textbox(
+                                label="5. 결론 및 강조",
+                                lines=4
+                            )
+                            section5_generate = gr.Button("생성")
+                            section5_video = gr.Video(label="섹션 5 영상")
+
+    # Event handlers
     txt2vid_preset.change(
         fn=preset_changed,
         inputs=[txt2vid_preset],
@@ -610,16 +739,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
             txt2vid_current_num_frames,
             *txt2vid_advanced[3:]
         ]
-    )    
-    
-
-
+    )
 
     txt2vid_enhance_toggle.change(
         fn=update_prompt_t2v,
         inputs=[txt2vid_prompt, txt2vid_enhance_toggle],
         outputs=txt2vid_prompt
     )
+
     txt2vid_generate.click(
         fn=generate_video_from_text,
         inputs=[
@@ -627,11 +754,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
             txt2vid_enhance_toggle,
             txt2vid_negative_prompt,
             txt2vid_frame_rate,
-            *txt2vid_advanced[:3],  # seed, inference_steps, guidance_scale
-            txt2vid_current_height,  # State 변수 참조
-            txt2vid_current_width,   # State 변수 참조
-            txt2vid_current_num_frames,  # State 변수 참조
-            ],
+            *txt2vid_advanced[:3],
+            txt2vid_current_height,
+            txt2vid_current_width,
+            txt2vid_current_num_frames,
+        ],
         outputs=txt2vid_output,
         concurrency_limit=1,
         concurrency_id="generate_video",
@@ -663,7 +790,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
             img2vid_enhance_toggle,
             img2vid_negative_prompt,
             img2vid_frame_rate,
-            *img2vid_advanced[:3],  # seed, inference_steps, guidance_scale
+            *img2vid_advanced[:3],
             img2vid_current_height,
             img2vid_current_width,
             img2vid_current_num_frames,
@@ -674,6 +801,46 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         queue=True,
     )
 
+    # Scenario tab event handlers
+    analyze_btn.click(
+        fn=analyze_scenario,
+        inputs=[scenario_input],
+        outputs=[
+            section1_prompt, section2_prompt, section3_prompt,
+            section4_prompt, section5_prompt
+        ]
+    )
+
+    section1_generate.click(
+        fn=generate_section_video,
+        inputs=[section1_prompt, scenario_preset],
+        outputs=section1_video
+    )
+
+    section2_generate.click(
+        fn=generate_section_video,
+        inputs=[section2_prompt, scenario_preset],
+        outputs=section2_video
+    )
+
+    section3_generate.click(
+        fn=generate_section_video,
+        inputs=[section3_prompt, scenario_preset],
+        outputs=section3_video
+    )
+
+    section4_generate.click(
+        fn=generate_section_video,
+        inputs=[section4_prompt, scenario_preset],
+        outputs=section4_video
+    )
+
+    section5_generate.click(
+        fn=generate_section_video,
+        inputs=[section5_prompt, scenario_preset],
+        outputs=section5_video
+    )
+
 if __name__ == "__main__":
     iface.queue(max_size=64, default_concurrency_limit=1, api_open=False).launch(
         share=True, show_api=False