Update app.py

app.py

@@ -50,6 +50,7 @@ print("Checking available fonts...")
 available_fonts = check_available_fonts()
 if not available_fonts:
     print("WARNING: No Korean fonts found! Please ensure font files are in the same directory as app.py")
+    available_fonts = ["나눔고딕"]  # Default fallback
 else:
     print(f"Available fonts: {', '.join(available_fonts)}")
 
@@ -365,15 +366,6 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
                         max_lines=1,
                         placeholder="Enter your prompt",
                         container=False,
-                    # Font change event handler for debugging
-                    def on_font_change(font_name):
-                        print(f"Font changed to: {font_name}")
-                        return font_name
-                    
-                    font_name.change(
-                        fn=on_font_change,
-                        inputs=[font_name],
-                        outputs=[font_name]
                     )
                     augment_button = gr.Button("증강", scale=0)
                     run_button = gr.Button("Run", scale=0)
@@ -398,7 +390,7 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
                                 label="제목 글자 크기",
                                 minimum=20,
                                 maximum=100,
-                                value=48,
+                                value=100,
                                 step=2
                             )
                         with gr.Column():
@@ -412,20 +404,49 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
                                 label="지은이 글자 크기",
                                 minimum=16,
                                 maximum=60,
-                                value=32,
+                                value=24,
                                 step=2
                             )
                     
                     with gr.Row():
-                        font_name = gr.Dropdown(
-                            label="폰트 선택",
-                            choices=["나눔고딕", "나눔명조", "맑은 고딕", "바탕", "돋움", "기본"],
-                            value="나눔고딕"
-                        )
-                        text_color = gr.ColorPicker(
-                            label="글자 색상",
-                            value="#FFFFFF"
-                        )
+                        with gr.Column():
+                            font_name = gr.Dropdown(
+                                label="폰트 선택",
+                                choices=available_fonts,
+                                value=available_fonts[0] if available_fonts else "나눔고딕"
+                            )
+                        with gr.Column():
+                            gr.Markdown("### 글자 색상 선택")
+                            color_buttons = gr.Radio(
+                                label="미리 정의된 색상",
+                                choices=[
+                                    ("흰색", "#FFFFFF"),
+                                    ("검정색", "#000000"),
+                                    ("빨간색", "#FF0000"),
+                                    ("파란색", "#0000FF"),
+                                    ("초록색", "#00FF00"),
+                                    ("노란색", "#FFFF00"),
+                                    ("주황색", "#FFA500"),
+                                    ("보라색", "#800080"),
+                                    ("회색", "#808080"),
+                                    ("금색", "#FFD700"),
+                                    ("은색", "#C0C0C0"),
+                                    ("커스텀", "custom")
+                                ],
+                                value="#FFFFFF",
+                                type="value"
+                            )
+                            
+                            custom_color = gr.ColorPicker(
+                                label="커스텀 색상 선택",
+                                value="#FFFFFF",
+                                visible=False
+                            )
+                            
+                            text_color = gr.Textbox(
+                                value="#FFFFFF",
+                                visible=False
+                            )
 
                 with gr.Accordion("Advanced Settings", open=False):
                     seed = gr.Slider(
@@ -497,6 +518,22 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
     # Event handlers
     def refresh_gallery():
         return load_generated_images()
+    
+    def update_color_visibility(selected_color):
+        if selected_color == "custom":
+            return gr.update(visible=True), gr.update(value="#FFFFFF")
+        else:
+            return gr.update(visible=False), gr.update(value=selected_color)
+    
+    def update_final_color(selected_color, custom_color_value):
+        if selected_color == "custom":
+            return custom_color_value
+        else:
+            return selected_color
+    
+    def on_font_change(font_name):
+        print(f"Font changed to: {font_name}")
+        return font_name
 
     refresh_btn.click(
         fn=refresh_gallery,
@@ -504,25 +541,29 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
         outputs=generated_gallery,
     )
     
-    # Augment button handler
     augment_button.click(
         fn=augment_prompt,
         inputs=[prompt],
         outputs=[prompt],
     )
-
-    # Auto-augment Korean prompts
-    def handle_prompt_change(prompt_text):
-        if is_korean_only(prompt_text):
-            return augment_prompt_with_llm(prompt_text)
-        return prompt_text
-    
-    # Optional: Auto-augment on prompt change (commented out to avoid too many API calls)
-    # prompt.change(
-    #     fn=handle_prompt_change,
-    #     inputs=[prompt],
-    #     outputs=[prompt]
-    # )
+    
+    font_name.change(
+        fn=on_font_change,
+        inputs=[font_name],
+        outputs=[font_name]
+    )
+    
+    color_buttons.change(
+        fn=update_color_visibility,
+        inputs=[color_buttons],
+        outputs=[custom_color, text_color]
+    )
+    
+    custom_color.change(
+        fn=update_final_color,
+        inputs=[color_buttons, custom_color],
+        outputs=[text_color]
+    )
 
     gr.on(
         triggers=[run_button.click, prompt.submit],