Update app.py

app.py

@@ -108,14 +108,63 @@ def openai_translate(text: str, retries: int = 3) -> str:
             time.sleep(2)
     return text  # 번역 실패 시 원문 그대로
 
-def prepare_prompt(user_prompt: str, style_key: str) -> str:
-    """한글이면 번역하고, 선택한 스타일 프리셋을 붙여서 최종 프롬프트를 만든다."""
+def enhance_prompt(text: str, retries: int = 3) -> str:
+    """OpenAI를 통해 프롬프트를 증강하여 고품질 이미지 생성을 위한 상세한 설명으로 변환."""
+    if client is None:
+        print("Warning: OpenAI client not available, returning original text")
+        return text
+
+    for attempt in range(retries):
+        try:
+            res = client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {
+                        "role": "system",
+                        "content": """You are an expert prompt engineer for image generation models. Enhance the given prompt to create high-quality, detailed images.
+
+Guidelines:
+- Add specific visual details (lighting, composition, colors, textures)
+- Include technical photography terms (depth of field, focal length, etc.)
+- Add atmosphere and mood descriptors
+- Specify image quality terms (4K, ultra-detailed, professional, etc.)
+- Keep the core subject and meaning intact
+- Make it comprehensive but not overly long
+- Focus on visual elements that will improve image generation quality
+
+Example:
+Input: "A man giving a speech"
+Output: "A professional man giving an inspiring speech at a podium, dramatic lighting with warm spotlights, confident posture and gestures, high-resolution 4K photography, sharp focus, cinematic composition, bokeh background with audience silhouettes, professional event setting, detailed facial expressions, realistic skin texture"
+"""
+                    },
+                    {"role": "user", "content": f"Enhance this prompt for high-quality image generation: {text}"}
+                ],
+                temperature=0.7,
+                max_tokens=512,
+            )
+            return res.choices[0].message.content.strip()
+        except Exception as e:
+            print(f"[enhance] attempt {attempt + 1} failed: {e}")
+            time.sleep(2)
+    return text  # 증강 실패 시 원문 그대로
+
+def prepare_prompt(user_prompt: str, style_key: str, enhance_prompt_enabled: bool = False) -> str:
+    """한글이면 번역하고, 프롬프트 증강 옵션이 활성화되면 증강하고, 선택한 스타일 프리셋을 붙여서 최종 프롬프트를 만든다."""
+    # 1. 번역 (한글인 경우)
     prompt_en = openai_translate(user_prompt)
+    
+    # 2. 프롬프트 증강 (활성화된 경우)
+    if enhance_prompt_enabled:
+        prompt_en = enhance_prompt(prompt_en)
+        print(f"Enhanced prompt: {prompt_en}")
+    
+    # 3. 스타일 프리셋 적용
     style_suffix = STYLE_PRESETS.get(style_key, "")
     if style_suffix:
         final_prompt = f"{prompt_en}, {style_suffix}"
     else:
         final_prompt = prompt_en
+    
     return final_prompt
 
 # ===== 이미지 저장 =====
@@ -157,6 +206,7 @@ def run_pipeline(prompt: str, seed: int, width: int, height: int, guidance_scale
 def generate_image(
     user_prompt: str,
     style_key: str,
+    enhance_prompt_enabled: bool = False,
     seed: int = 42,
     randomize_seed: bool = True,
     width: int = 1024,
@@ -171,7 +221,7 @@ def generate_image(
             seed = random.randint(0, MAX_SEED)
 
         # 1) 번역 + 증강
-        final_prompt = prepare_prompt(user_prompt, style_key)
+        final_prompt = prepare_prompt(user_prompt, style_key, enhance_prompt_enabled)
         print(f"Final prompt: {final_prompt}")
 
         # 2) 파이프라인 호출
@@ -224,6 +274,38 @@ button:hover{transform:translateY(-2px); box-shadow:0 5px 15px rgba(0,0,0,.1);}
 .input-container{border-radius:10px; box-shadow:0 2px 8px rgba(0,0,0,.05); background:rgba(255,255,255,.6); padding:20px; margin-bottom:1rem;} 
 .advanced-settings{margin-top:1rem; padding:1rem; border-radius:10px; background:rgba(255,255,255,.6);} 
 .example-region{background:rgba(255,255,255,.5); border-radius:10px; padding:1rem; margin-top:1rem;} 
+
+/* 프롬프트 입력칸 크기 2배 증가 */
+.large-prompt textarea {
+    min-height: 120px !important;
+    font-size: 16px !important;
+    line-height: 1.5 !important;
+}
+
+/* 생성 버튼 작게 만들기 */
+.small-generate-btn {
+    max-width: 120px !important;
+    height: 40px !important;
+    font-size: 14px !important;
+    padding: 8px 16px !important;
+}
+
+/* 프롬프트 증강 섹션 스타일 */
+.prompt-enhance-section {
+    background: rgba(255,255,255,.7);
+    border-radius: 8px;
+    padding: 15px;
+    margin-top: 10px;
+    border-left: 3px solid var(--color-primary);
+}
+
+/* 스타일 프리셋 섹션 */
+.style-preset-section {
+    background: rgba(255,255,255,.6);
+    border-radius: 8px;
+    padding: 15px;
+    margin-top: 10px;
+}
 """
 
 # ===== Gradio UI =====
@@ -244,9 +326,36 @@ def create_interface():
         # ===== 메인 입력 =====
         with gr.Column():
             with gr.Row(elem_classes="input-container"):
-                user_prompt = gr.Text(label="Prompt", max_lines=1, value=examples[0])
-                style_select = gr.Radio(label="Style Preset", choices=list(STYLE_PRESETS.keys()), value="None", interactive=True)
-                run_button = gr.Button("Generate", variant="primary")
+                with gr.Column(scale=4):
+                    user_prompt = gr.Text(
+                        label="Prompt", 
+                        max_lines=5, 
+                        value=examples[0],
+                        elem_classes="large-prompt"
+                    )
+                with gr.Column(scale=1):
+                    run_button = gr.Button(
+                        "생성", 
+                        variant="primary",
+                        elem_classes="small-generate-btn"
+                    )
+            
+            # 프롬프트 증강 옵션 (생성 버튼 아래)
+            with gr.Group(elem_classes="prompt-enhance-section"):
+                enhance_prompt_checkbox = gr.Checkbox(
+                    label="🚀 프롬프트 증강 (AI로 프롬프트를 자동으로 개선하여 고품질 이미지 생성)", 
+                    value=False,
+                    info="OpenAI API를 사용하여 입력한 프롬프트를 더욱 상세하고 고품질의 이미지를 생성할 수 있도록 자동으로 증강합니다."
+                )
+            
+            # 스타일 프리셋 섹션
+            with gr.Group(elem_classes="style-preset-section"):
+                style_select = gr.Radio(
+                    label="🎨 Style Preset", 
+                    choices=list(STYLE_PRESETS.keys()), 
+                    value="None", 
+                    interactive=True
+                )
 
             result_image = gr.Image(label="Generated Image")
             seed_output = gr.Number(label="Seed")
@@ -274,6 +383,7 @@ def create_interface():
             inputs=[
                 user_prompt,
                 style_select,
+                enhance_prompt_checkbox,
                 seed,
                 randomize_seed,
                 width,