Update app.py

app.py

@@ -525,7 +525,8 @@ def add_text_to_image(
     x_position,
     y_position,
     thickness,
-    text_position_type
+    text_position_type,
+    font_choice  # 새로운 파라미터 추가
 ):
     """
     Add text to an image with customizable properties
@@ -557,9 +558,17 @@ def add_text_to_image(
         draw = ImageDraw.Draw(txt_overlay)
         
         # 폰트 설정
+        font_files = {
+            "Default": "DejaVuSans.ttf",
+            "Korean Regular": "ko-Regular.ttf",
+            "Korean Son": "ko-son.ttf"
+        }
+        
         try:
-            font = ImageFont.truetype("DejaVuSans.ttf", int(font_size))
-        except:
+            font_file = font_files.get(font_choice, "DejaVuSans.ttf")
+            font = ImageFont.truetype(font_file, int(font_size))
+        except Exception as e:
+            print(f"Font loading error ({font_choice}): {str(e)}")
             try:
                 font = ImageFont.truetype("arial.ttf", int(font_size))
             except:
@@ -617,6 +626,7 @@ def add_text_to_image(
     except Exception as e:
         print(f"Error in add_text_to_image: {str(e)}")
         return input_image
+
     
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
@@ -710,6 +720,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     
                     with gr.Row():
                         with gr.Column(scale=1):
+                            # 폰트 선택 Dropdown 추가
+                            font_choice = gr.Dropdown(
+                                choices=["Default", "Korean Regular", "Korean Son"],
+                                value="Default",
+                                label="Font Selection",
+                                interactive=True
+                            )
                             font_size = gr.Slider(
                                 minimum=10,
                                 maximum=200,
@@ -819,7 +836,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=True
     )
 
-    # 텍스트 추가 버튼 이벤트 연결
+    # 텍스트 추가 버튼 이벤트 연결 수정
     add_text_btn.click(
         fn=add_text_to_image,
         inputs=[
@@ -831,7 +848,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             x_position,
             y_position,
             thickness,
-            text_position_type
+            text_position_type,
+            font_choice  # 새로운 입력 추가
         ],
         outputs=combined_image
     )