Update app.py

app.py

@@ -86,71 +86,31 @@ def augment_prompt_with_llm(prompt):
         print(f"Error calling LLM API: {e}")
         return prompt
 
-def get_korean_font(font_name, font_size):
-    """Get Korean font based on font name"""
-    font_paths = {
-        "나눔고딕": [
-            "/usr/share/fonts/truetype/nanum/NanumGothic.ttf",
-            "NanumGothic.ttf",
-            "/System/Library/Fonts/AppleSDGothicNeo.ttc",  # macOS
-            "C:/Windows/Fonts/NanumGothic.ttf",  # Windows
-        ],
-        "나눔명조": [
-            "/usr/share/fonts/truetype/nanum/NanumMyeongjo.ttf",
-            "NanumMyeongjo.ttf",
-            "C:/Windows/Fonts/NanumMyeongjo.ttf",  # Windows
-        ],
-        "맑은 고딕": [
-            "/usr/share/fonts/truetype/malgun/malgun.ttf",
-            "malgun.ttf",
-            "/System/Library/Fonts/AppleMyungjo.ttf",  # macOS
-            "C:/Windows/Fonts/malgun.ttf",  # Windows
-        ],
-        "바탕": [
-            "/usr/share/fonts/truetype/batang/batang.ttf",
-            "batang.ttf",
-            "C:/Windows/Fonts/batang.ttc",  # Windows
-        ],
-        "돋움": [
-            "/usr/share/fonts/truetype/dotum/dotum.ttf",
-            "dotum.ttf",
-            "C:/Windows/Fonts/dotum.ttc",  # Windows
-        ],
-        "기본": []
-    }
-    
-    # Try to load the selected font
-    if font_name in font_paths:
-        for font_path in font_paths[font_name]:
+def get_korean_font(font_size):
+    """Load NanumGothic font from the same directory as app.py"""
+    try:
+        # Try to load NanumGothic-Regular.ttf from the same directory
+        font_path = "NanumGothic-Regular.ttf"
+        return ImageFont.truetype(font_path, font_size)
+    except:
+        # If font file is not found, try alternative paths
+        alternative_paths = [
+            "./NanumGothic-Regular.ttf",
+            os.path.join(os.path.dirname(__file__), "NanumGothic-Regular.ttf"),
+        ]
+        
+        for path in alternative_paths:
             try:
-                return ImageFont.truetype(font_path, font_size)
+                return ImageFont.truetype(path, font_size)
             except:
                 continue
-    
-    # Try common font directories
-    common_paths = [
-        f"/usr/share/fonts/truetype/{font_name.lower()}/{font_name}.ttf",
-        f"fonts/{font_name}.ttf",
-        f"{font_name}.ttf",
-    ]
-    
-    for path in common_paths:
-        try:
-            return ImageFont.truetype(path, font_size)
-        except:
-            continue
-    
-    # Fallback to default font with larger size
-    try:
-        return ImageFont.truetype("DejaVuSans.ttf", font_size)
-    except:
-        # Use default font but try to make it larger
-        default = ImageFont.load_default()
-        # PIL default font is very small, so we might need to scale the text
-        return default
+        
+        # Final fallback to default font
+        print("Warning: NanumGothic-Regular.ttf not found. Using default font.")
+        return ImageFont.load_default()
 
 def add_text_overlay(image, title_ko, author_ko, 
-                    title_position, author_position, text_color, font_name,
+                    title_position, author_position, text_color,
                     title_size, author_size):
     """Add Korean text overlay to the generated image"""
     # Create a copy of the image to work with
@@ -158,8 +118,8 @@ def add_text_overlay(image, title_ko, author_ko,
     draw = ImageDraw.Draw(img_with_text)
     
     # Load Korean fonts with custom sizes
-    title_font = get_korean_font(font_name, title_size)
-    author_font = get_korean_font(font_name, author_size)
+    title_font = get_korean_font(title_size)
+    author_font = get_korean_font(author_size)
     
     # Get image dimensions
     img_width, img_height = img_with_text.size
@@ -257,7 +217,6 @@ def inference(
     title_position: str,
     author_position: str,
     text_color: str,
-    font_name: str,
     title_size: int,
     author_size: int,
     progress: gr.Progress = gr.Progress(track_tqdm=True),
@@ -279,7 +238,7 @@ def inference(
     # Add text overlay if any Korean text is provided
     if title_ko or author_ko:
         image = add_text_overlay(image, title_ko, author_ko,
-                               title_position, author_position, text_color, font_name,
+                               title_position, author_position, text_color,
                                title_size, author_size)
     
     # Save the generated image
@@ -352,7 +311,7 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
                                 label="제목 글자 크기",
                                 minimum=20,
                                 maximum=100,
-                                value=100,
+                                value=48,
                                 step=2
                             )
                         with gr.Column():
@@ -366,7 +325,7 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
                                 label="지은이 글자 크기",
                                 minimum=16,
                                 maximum=60,
-                                value=24,
+                                value=32,
                                 step=2
                             )
                     
@@ -495,7 +454,6 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
             title_position,
             author_position,
             text_color,
-            font_name,
             title_size,
             author_size,
         ],