Update app.py

app.py

@@ -86,17 +86,27 @@ def augment_prompt_with_llm(prompt):
         print(f"Error calling LLM API: {e}")
         return prompt
 
-def get_korean_font(font_size):
-    """Load NanumGothic font from the same directory as app.py"""
+def get_korean_font(font_name, font_size):
+    """Load Korean font from the same directory as app.py"""
+    font_files = {
+        "나눔고딕": "NanumGothic-Regular.ttf",
+        "흑백사진": "BlackAndWhitePicture-Regular.ttf",
+        "치론성": "ChironSungHK-VariableFont_wght.ttf",
+        "독도": "Dokdo-Regular.ttf",
+        "싱글데이": "SingleDay-Regular.ttf"
+    }
+    
+    # Get the font file name
+    font_file = font_files.get(font_name, "NanumGothic-Regular.ttf")
+    
     try:
-        # Try to load NanumGothic-Regular.ttf from the same directory
-        font_path = "NanumGothic-Regular.ttf"
-        return ImageFont.truetype(font_path, font_size)
+        # Try to load font from the same directory
+        return ImageFont.truetype(font_file, 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"),
+            f"./{font_file}",
+            os.path.join(os.path.dirname(__file__), font_file),
         ]
         
         for path in alternative_paths:
@@ -105,12 +115,16 @@ def get_korean_font(font_size):
             except:
                 continue
         
-        # Final fallback to default font
-        print("Warning: NanumGothic-Regular.ttf not found. Using default font.")
-        return ImageFont.load_default()
+        # If specific font not found, try to load NanumGothic as fallback
+        try:
+            return ImageFont.truetype("NanumGothic-Regular.ttf", font_size)
+        except:
+            # Final fallback to default font
+            print(f"Warning: {font_file} not found. Using default font.")
+            return ImageFont.load_default()
 
 def add_text_overlay(image, title_ko, author_ko, 
-                    title_position, author_position, text_color,
+                    title_position, author_position, text_color, font_name,
                     title_size, author_size):
     """Add Korean text overlay to the generated image"""
     # Create a copy of the image to work with
@@ -118,8 +132,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(title_size)
-    author_font = get_korean_font(author_size)
+    title_font = get_korean_font(font_name, title_size)
+    author_font = get_korean_font(font_name, author_size)
     
     # Get image dimensions
     img_width, img_height = img_with_text.size
@@ -217,6 +231,7 @@ 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),
@@ -238,7 +253,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,
+                               title_position, author_position, text_color, font_name,
                                title_size, author_size)
     
     # Save the generated image
@@ -454,6 +469,7 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False) as demo:
             title_position,
             author_position,
             text_color,
+            font_name,
             title_size,
             author_size,
         ],