Update app.py

app.py

@@ -69,16 +69,54 @@ def contains_korean(text):
     korean_pattern = re.compile('[ㄱ-ㅎㅏ-ㅣ가-힣]')
     return bool(korean_pattern.search(text))
 
-def translate_korean_prompt(prompt):
+def translate_korean_prompt(prompt, max_length=450):
     """
     Translate Korean prompt to English if Korean text is detected
+    Split long text into chunks if necessary
     """
-    if contains_korean(prompt):
-        translated = translator(prompt)[0]['translation_text']
+    if not contains_korean(prompt):
+        return prompt
+        
+    # Split long text into chunks
+    def split_text(text, max_length):
+        words = text.split()
+        chunks = []
+        current_chunk = []
+        current_length = 0
+        
+        for word in words:
+            if current_length + len(word) + 1 > max_length:
+                chunks.append(' '.join(current_chunk))
+                current_chunk = [word]
+                current_length = len(word)
+            else:
+                current_chunk.append(word)
+                current_length += len(word) + 1
+                
+        if current_chunk:
+            chunks.append(' '.join(current_chunk))
+        return chunks
+
+    try:
+        if len(prompt) > max_length:
+            chunks = split_text(prompt, max_length)
+            translated_chunks = []
+            
+            for chunk in chunks:
+                translated = translator(chunk, max_length=512)[0]['translation_text']
+                translated_chunks.append(translated)
+                
+            final_translation = ' '.join(translated_chunks)
+        else:
+            final_translation = translator(prompt, max_length=512)[0]['translation_text']
+            
         print(f"Original Korean prompt: {prompt}")
-        print(f"Translated English prompt: {translated}")
-        return translated
-    return prompt
+        print(f"Translated English prompt: {final_translation}")
+        return final_translation
+        
+    except Exception as e:
+        print(f"Translation error: {e}")
+        return prompt  # Return original prompt if translation fails
 
 def enhance_prompt(prompt, type="t2v"):
     system_prompt = system_prompt_t2v if type == "t2v" else system_prompt_i2v
@@ -554,20 +592,25 @@ def analyze_scenario(scenario):
         return ["Error occurred during analysis"] * 5
 
 def generate_section_video(prompt, preset, section_number=1, base_seed=171198, progress=gr.Progress()):
-    """각 섹션의 비디오 생성 - 일관된 SEED 값 사용"""
-    selected = next(item for item in preset_options if item["label"] == preset)
-    
-    # 기본 SEED에 섹션 번호를 더해 약간의 변화를 주되 일관성 유지
-    section_seed = base_seed + section_number
-    
-    return generate_video_from_text(
-        prompt=prompt,
-        height=selected["height"],
-        width=selected["width"],
-        num_frames=selected["num_frames"],
-        seed=section_seed,  # 섹션별 고유 SEED 사용
-        progress=progress
-    )
+    """각 섹션의 비디오 생성 - 에러 처리 추가"""
+    try:
+        if not prompt or len(prompt.strip()) < 50:
+            raise gr.Error("프롬프트는 최소 50자 이상이어야 합니다.")
+            
+        selected = next(item for item in preset_options if item["label"] == preset)
+        section_seed = base_seed + section_number
+        
+        return generate_video_from_text(
+            prompt=prompt,
+            height=selected["height"],
+            width=selected["width"],
+            num_frames=selected["num_frames"],
+            seed=section_seed,
+            progress=progress
+        )
+    except Exception as e:
+        print(f"Error in section {section_number}: {e}")
+        raise gr.Error(f"섹션 {section_number} 생성 중 오류: {str(e)}")
 
 
 # 개별 섹션 프롬프트 생성 함수 추가
@@ -930,37 +973,44 @@ 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
+# 섹션 생성 이벤트 핸들러
+    section1_generate.click(
+        fn=generate_section_video,
+        inputs=[section1_prompt, scenario_preset],
+        outputs=section1_video,
+        api_name=f"generate_section1"
     )
-    
-    section2_regenerate.click(
-        fn=lambda x: generate_single_section_prompt(x, 2),
-        inputs=[scenario_input],
-        outputs=section2_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"
     )
-    
-    section3_regenerate.click(
-        fn=lambda x: generate_single_section_prompt(x, 3),
-        inputs=[scenario_input],
-        outputs=section3_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"
     )
-    
-    section4_regenerate.click(
-        fn=lambda x: generate_single_section_prompt(x, 4),
-        inputs=[scenario_input],
-        outputs=section4_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"
     )
-    
-    section5_regenerate.click(
-        fn=lambda x: generate_single_section_prompt(x, 5),
-        inputs=[scenario_input],
-        outputs=section5_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"
     )
 
+
+
     # 섹션 생성 이벤트 핸들러
     section1_generate.click(
         fn=lambda p, pr: generate_section_video(p, pr, 1),