Update app.py

app.py

@@ -309,6 +309,46 @@ footer {display: none}
 }
 """
 
+try:
+    prompt_enhancer = pipeline(
+        "text-generation", 
+        model="CohereForAI/c4ai-command-r-plus-08-2024",
+        token=HF_TOKEN
+    )
+except Exception as e:
+    print(f"Error initializing prompt enhancer: {str(e)}")
+    prompt_enhancer = None
+
+def enhance_prompt(prompt: str) -> str:
+    """프롬프트를 애니메이션 스타일로 증강"""
+    try:
+        if prompt_enhancer is None:
+            return prompt
+
+        base_prompt = f"Convert this into a detailed anime-style image prompt: {prompt}"
+        messages = [{"role": "user", "content": base_prompt}]
+        
+        response = prompt_enhancer(messages)
+        enhanced = response[0]['generated_text']
+        
+        # 기본 품질 향상 프롬프트 추가
+        enhancements = [
+            "masterpiece, best quality, highly detailed",
+            "anime style, animation style",
+            "vibrant colors, perfect lighting",
+            "professional composition",
+            "[trigger]"
+        ]
+        
+        # 최종 프롬프트 구성
+        final_prompt = f"{enhanced}, {', '.join(enhancements)}"
+        print(f"Enhanced prompt: {final_prompt}")
+        
+        return final_prompt
+    except Exception as e:
+        print(f"Prompt enhancement failed: {str(e)}")
+        return prompt
+
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -319,11 +359,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
     with gr.Row():
         with gr.Column(scale=1):
-            # 이미지 생성 섹션
             gen_prompt = gr.Textbox(
                 label="Generation Prompt",
                 placeholder="Enter your image generation prompt..."
             )
+            enhance_btn = gr.Button("✨ Enhance Prompt", variant="secondary")
+
             with gr.Row():
                 gen_width = gr.Slider(512, 1024, 768, step=64, label="Width")
                 gen_height = gr.Slider(512, 1024, 768, step=64, label="Height")
@@ -441,6 +482,17 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         outputs=output_image
     )
 
+    # 이벤트 바인딩 추가
+    def update_prompt(prompt):
+        enhanced = enhance_prompt(prompt)
+        return enhanced
+
+    enhance_btn.click(
+        fn=update_prompt,
+        inputs=[gen_prompt],
+        outputs=[gen_prompt]
+    )
+
 demo.queue(max_size=5)
 demo.launch(
     server_name="0.0.0.0",