Update app.py

app.py

@@ -12,6 +12,47 @@ import requests
 import json
 import re
 
+# At the beginning of the script, after imports
+def check_available_fonts():
+    """Check which font files are available"""
+    font_files = {
+        "나눔고딕": "NanumGothic-Regular.ttf",
+        "흑백사진": "BlackAndWhitePicture-Regular.ttf",
+        "치론성": "ChironSungHK-VariableFont_wght.ttf",
+        "독도": "Dokdo-Regular.ttf",
+        "싱글데이": "SingleDay-Regular.ttf"
+    }
+    
+    available_fonts = []
+    for font_name, font_file in font_files.items():
+        paths = [
+            font_file,
+            f"./{font_file}",
+            os.path.join(os.getcwd(), font_file),
+            os.path.join(os.path.dirname(os.path.abspath(__file__)), font_file)
+        ]
+        
+        font_found = False
+        for path in paths:
+            if os.path.exists(path):
+                available_fonts.append(font_name)
+                print(f"✓ Found {font_name} at: {path}")
+                font_found = True
+                break
+        
+        if not font_found:
+            print(f"✗ {font_name} ({font_file}) not found")
+    
+    return available_fonts
+
+# Check fonts at startup
+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")
+else:
+    print(f"Available fonts: {', '.join(available_fonts)}")
+
 # Create permanent storage directory
 SAVE_DIR = "saved_images"  # Gradio will handle the persistence
 if not os.path.exists(SAVE_DIR):
@@ -98,35 +139,46 @@ def get_korean_font(font_name, font_size):
     
     # Get the font file name
     font_file = font_files.get(font_name, "NanumGothic-Regular.ttf")
+    print(f"Trying to load font: {font_name} -> {font_file}")  # Debug info
     
-    try:
-        # 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 = [
-            f"./{font_file}",
-            os.path.join(os.path.dirname(__file__), font_file),
-        ]
-        
-        for path in alternative_paths:
-            try:
-                return ImageFont.truetype(path, font_size)
-            except:
-                continue
-        
-        # If specific font not found, try to load NanumGothic as fallback
+    # List of paths to try
+    paths_to_try = [
+        font_file,  # Current directory
+        f"./{font_file}",  # Explicit current directory
+        os.path.join(os.getcwd(), font_file),  # Full path from current working directory
+        os.path.join(os.path.dirname(os.path.abspath(__file__)), font_file),  # Same directory as script
+    ]
+    
+    # Try each path
+    for path in paths_to_try:
+        try:
+            font = ImageFont.truetype(path, font_size)
+            print(f"Successfully loaded font from: {path}")  # Debug info
+            return font
+        except Exception as e:
+            print(f"Failed to load font from {path}: {e}")  # Debug info
+            continue
+    
+    # If specific font not found, try to load NanumGothic as fallback
+    print(f"Warning: {font_file} not found. Trying NanumGothic fallback...")  # Debug info
+    for path in ["NanumGothic-Regular.ttf", "./NanumGothic-Regular.ttf"]:
         try:
-            return ImageFont.truetype("NanumGothic-Regular.ttf", font_size)
+            font = ImageFont.truetype(path, font_size)
+            print(f"Loaded fallback font from: {path}")  # Debug info
+            return font
         except:
-            # Final fallback to default font
-            print(f"Warning: {font_file} not found. Using default font.")
-            return ImageFont.load_default()
+            continue
+    
+    # Final fallback to default font
+    print("Warning: No Korean fonts found. Using default font.")  # Debug info
+    return ImageFont.load_default()
 
 def add_text_overlay(image, title_ko, author_ko, 
                     title_position, author_position, text_color, font_name,
                     title_size, author_size):
     """Add Korean text overlay to the generated image"""
+    print(f"add_text_overlay called with font_name: {font_name}")  # Debug info
+    
     # Create a copy of the image to work with
     img_with_text = image.copy()
     draw = ImageDraw.Draw(img_with_text)
@@ -150,9 +202,13 @@ def add_text_overlay(image, title_ko, author_ko,
         title_x, title_y = position_coords[title_position]
         
         # Get text bbox for centering
-        bbox = draw.textbbox((0, 0), title_ko, font=title_font)
-        text_width = bbox[2] - bbox[0]
-        text_height = bbox[3] - bbox[1]
+        try:
+            bbox = draw.textbbox((0, 0), title_ko, font=title_font)
+            text_width = bbox[2] - bbox[0]
+            text_height = bbox[3] - bbox[1]
+        except:
+            # Fallback for older PIL versions
+            text_width, text_height = draw.textsize(title_ko, font=title_font)
         
         # Draw text with shadow for better visibility
         shadow_offset = 2
@@ -172,9 +228,13 @@ def add_text_overlay(image, title_ko, author_ko,
         author_x, author_y = position_coords[author_position]
         
         # Get text bbox for centering
-        bbox = draw.textbbox((0, 0), author_text, font=author_font)
-        text_width = bbox[2] - bbox[0]
-        text_height = bbox[3] - bbox[1]
+        try:
+            bbox = draw.textbbox((0, 0), author_text, font=author_font)
+            text_width = bbox[2] - bbox[0]
+            text_height = bbox[3] - bbox[1]
+        except:
+            # Fallback for older PIL versions
+            text_width, text_height = draw.textsize(author_text, font=author_font)
         
         # Draw text with shadow
         draw.text((author_x - text_width // 2 + shadow_offset, author_y - text_height // 2 + shadow_offset), 
@@ -236,6 +296,8 @@ def inference(
     author_size: int,
     progress: gr.Progress = gr.Progress(track_tqdm=True),
 ):
+    print(f"inference called with font_name: {font_name}")  # Debug info
+    
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
@@ -252,6 +314,7 @@ def inference(
     
     # Add text overlay if any Korean text is provided
     if title_ko or author_ko:
+        print(f"Adding text overlay with font: {font_name}")  # Debug info
         image = add_text_overlay(image, title_ko, author_ko,
                                title_position, author_position, text_color, font_name,
                                title_size, author_size)
@@ -302,6 +365,15 @@ 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)