Update app.py

app.py

@@ -755,7 +755,39 @@ def merge_section_videos(section1, section2, section3, section4, section5):
     except Exception as e:
         raise gr.Error(f"비디오 결합 중 오류 발생: {e}")
 
-
+def generate_script(topic):
+    """주제에 맞는 스크립트 생성"""
+    if not topic:
+        return "주제를 입력해주세요."
+        
+    messages = [
+        {"role": "system", "content": """당신은 영상 스크립트 작성 전문가입니다.
+주어진 주제로 다음 구조에 맞는 5개 섹션의 스크립트를 작성해주세요:
+
+1. 배경 및 필요성: 주제 소개와 시청자의 흥미 유발
+2. 흥미 유발: 구체적인 내용 전개와 호기심 자극
+3. 해결책 제시: 핵심 내용과 해결방안 제시
+4. 본론: 상세한 설명과 장점 부각
+5. 결론: 핵심 메시지 강조와 행동 유도
+
+각 섹션은 자연스럽게 연결되어야 하며, 
+전체적으로 일관된 톤과 분위기를 유지하면서도 
+시청자의 관심을 끝까지 유지할 수 있도록 작성해주세요."""},
+        {"role": "user", "content": f"다음 주제로 영상 스크립트를 작성해주세요: {topic}"}
+    ]
+    
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4-1106-preview",
+            messages=messages,
+            max_tokens=2000,
+            temperature=0.7
+        )
+        return response.choices[0].message.content.strip()
+    except Exception as e:
+        print(f"Error during script generation: {e}")
+        return "스크립트 생성 중 오류가 발생했습니다."
+        
 # Gradio Interface Definition
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     with gr.Tabs():
@@ -861,11 +893,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                 with gr.Column():
                     img2vid_output = gr.Video(label="생성된 비디오")
 
-
-        # Scenario to Video Tab (Modified)
         with gr.TabItem("시나리오로 비디오 만들기(숏폼)"):
             with gr.Row():
                 with gr.Column(scale=1):
+            # 스크립트 생성 섹션 추가
+                    script_topic = gr.Textbox(
+                        label="스크립트 생성",
+                        placeholder="겨울 일본 온천 여행을 주제로 밝은 느낌으로 스크립트 생성하라",
+                        lines=2
+                    )
+                    generate_script_btn = gr.Button("스크립트 생성", variant="primary")
+            
                     scenario_input = gr.Textbox(
                         label="영상 스크립트 입력",
                         placeholder="전체 시나리오를 입력하세요...",
@@ -877,6 +915,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         label="화면 크기 선택"
                     )
                     analyze_btn = gr.Button("시나리오 분석 및 프롬프트 생성", variant="primary")
+            
+
                 
                 with gr.Column(scale=2):
                     with gr.Row():
@@ -1120,6 +1160,12 @@ 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(