Update app.py

app.py

@@ -8,11 +8,24 @@ API_URL = "https://api.openai.com/v1/chat/completions"
 # 번역 파이프라인 초기화
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
-def translate_text(text):
+def translate_and_generate_prompts(text):
     # 입력된 텍스트를 영어로 번역
     translation = translator(text, max_length=512)
     translated_text = translation[0]['translation_text']
-    return translated_text
+    
+    # 기본 키워드 추가
+    prompt = "3d style, 4k"
+    
+    # 지정 키워드 추가
+    if "토리" in text:
+        prompt += ", like Brad Pitt young boy"
+    elif "줄리" in text:
+        prompt += ", like Emma Watson young girl"
+    
+    # 번역된 텍스트 추가
+    prompt += f", {translated_text}"
+    
+    return prompt
 
 def predict(inputs, top_p, temperature, openai_api_key):
     narration_prompt = f"아동용 애니메이션 동영상에 사용할 스크립트를 작성하라. 반드시 한글로 작성할것. 일체의 지문이나 지시, 기호 표시, 배경 설명 등을 노출 하거나 출력하지 말고 기승전결의 구조로 모험의 이유/위기/도전/문제해결/교훈을 포함하여 순수한 나레이션만 1줄씩 출력하여 최대 10줄 이내로 출력. 입력: '{inputs}'"
@@ -37,9 +50,9 @@ def predict(inputs, top_p, temperature, openai_api_key):
         return "Error: Unable to generate response."
 
 def generate_prompts(script):
-    # 스크립트의 각 줄을 영어로 번역하여 프롬프트 생성
+    # 스크립트의 각 줄을 영어로 번역하고 프롬프트 생성
     lines = script.split('\n')
-    translated_prompts = [translate_text(line) for line in lines if line.strip() != '']
+    translated_prompts = [translate_and_generate_prompts(line) for line in lines if line.strip() != '']
     return "\n".join(translated_prompts)
 
 with gr.Blocks() as demo: