Update app.py

app.py

@@ -39,6 +39,26 @@ def predict(inputs, top_p, temperature, openai_api_key):
     else:
         return "Error: Unable to generate response."
 
+def create_detailed_image_prompts(script):
+    # 스크립트의 각 줄을 분석하여 프롬프트 생성
+    lines = script.split('\n')
+    detailed_prompts = []
+    for line in lines:
+        # 여기서는 예시로 몇 가지 간단한 패턴을 사용합니다.
+        # 실제 구현에서는 더 복잡한 자연어 처리를 적용해야 할 수 있습니다.
+        if "야구장에 갔다" in line:
+            action = "go to the baseball (background)"
+            prompt = f"3d style, 4K, like Brad Pitt young boy, {action}"
+        elif "숲을 탐험했다" in line:
+            action = "explore the forest (background)"
+            prompt = f"3d style, 4K, like Brad Pitt young boy, {action}"
+        else:
+            # 일치하는 패턴이 없을 경우 기본 프롬프트 사용
+            prompt = f"3d style, 4K, like Brad Pitt young boy, adventure"
+        detailed_prompts.append(prompt)
+    return "\n".join(detailed_prompts)
+
+# Gradio 인터페이스 수정
 with gr.Blocks() as demo:
     gr.Markdown("<h1 align='center'>토리의 모험: 3D 애니메이션 생성기</h1>")
     with gr.Row():
@@ -51,7 +71,10 @@ with gr.Blocks() as demo:
     submit_button = gr.Button("Generate Script")
     prompts_button = gr.Button("Generate Image Prompts")
 
+    def generate_prompts(script):
+        return create_detailed_image_prompts(script)
+
     submit_button.click(fn=predict, inputs=[inputs, top_p, temperature, openai_api_key], outputs=output)
-    prompts_button.click(fn=create_image_prompts, inputs=[output, openai_api_key], outputs=prompts_output)
+    prompts_button.click(fn=generate_prompts, inputs=[output], outputs=prompts_output)
 
-demo.launch()
+demo.launch()