Update app.py

app.py

@@ -22,7 +22,7 @@ import os
 import gc
 from openai import OpenAI
 import re
-
+import time
 # Load system prompts
 system_prompt_t2v = """당신은 비디오 생성을 위한 프롬프트 전문가입니다. 
 주어진 프롬프트를 다음 구조에 맞게 개선해주세요:
@@ -551,42 +551,56 @@ system_prompt_scenario = """당신은 영상 스크립트에 맞는 배경 영
 
 
 def analyze_scenario(scenario):
-    """시나리오를 분석하여 배경 영상용 프롬프트 생성"""
-    messages = [
-        {"role": "system", "content": system_prompt_scenario},
-        {"role": "user", "content": f"""
-다음 스크립트의 분위기와 감성을 표현할 수 있는 배경 영상 프롬프트를 생성해주세요:
+    """시나리오를 분석하여 각 섹션별 배경 영상용 프롬프트 생성"""
+    try:
+        # 각 섹션별 프롬프트 생성을 위한 메시지 구성
+        section_prompts = []
+        
+        for section_num in range(1, 6):
+            section_descriptions = {
+                1: "배경 및 필요성: 주제의 전반적인 분위기를 표현하는 배경 씬",
+                2: "흥미 유발: 긴장감이나 갈등을 암시하는 분위기 있는 배경",
+                3: "해결책 제시: 희망적이고 밝은 톤의 배경 전환",
+                4: "본론: 안정감 있고 신뢰도를 높이는 배경",
+                5: "결론: 임팩트 있는 마무리를 위한 역동적인 배경"
+            }
+            
+            messages = [
+                {"role": "system", "content": system_prompt_scenario},
+                {"role": "user", "content": f"""
+다음 스크립트의 {section_num}번째 섹션({section_descriptions[section_num]})에 대한 
+배경 영상 프롬프트를 생성해주세요.
 
+스크립트:
 {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")
-        
-        # 프롬프트 처리 로직은 동일
-        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)
+주의사항:
+1. 해당 섹션의 특성({section_descriptions[section_num]})에 맞는 분위기와 톤을 반영하세요.
+2. 직접적인 제품/서비스 묘사는 피하고, 감성적이고 은유적인 배경 영상에 집중하세요.
+3. 다음 구조를 반드시 포함하세요:
+   - 주요 동작을 명확한 한 문장으로 시작
+   - 구체적인 동작과 제스처를 시간 순서대로 설명
+   - 배경과 환경 세부 사항을 구체적으로 포함
+   - 카메라 각도와 움직임을 명시
+   - 조명과 색상을 자세히 설명
+   - 변화나 갑작스러운 사건을 자연스럽게 포함"""}
+            ]
+
+            response = client.chat.completions.create(
+                model="gpt-4-1106-preview",
+                messages=messages,
+                max_tokens=500,
+                temperature=0.7
+            )
+            
+            section_prompt = response.choices[0].message.content.strip()
+            section_prompts.append(f"{section_num}. {section_prompt}")
             
-        while len(section_prompts) < 5:
-            section_prompts.append("추가 섹션이 필요합니다.")
-        return section_prompts[:5]
+            # API 요청 사이에 짧은 딜레이 추가
+            time.sleep(1)
+        
+        return section_prompts
+        
     except Exception as e:
         print(f"Error during scenario analysis: {e}")
         return ["Error occurred during analysis"] * 5