Update app.py

app.py

@@ -85,24 +85,30 @@ def resize_image(image, resolution):
     new_h = int(h * ratio)
     return image.resize((new_w, new_h), Image.LANCZOS)
 
-def text_to_image(text, size=72, position="middle-center"):
+def get_font_files():
+    return [f for f in os.listdir() if f.endswith('.ttf')]
+
+def preview_font(font_name, preview_text):
+    font_size = 24
+    image = Image.new('RGB', (400, 100), color='white')
+    draw = ImageDraw.Draw(image)
+    try:
+        font = ImageFont.truetype(font_name, font_size)
+        draw.text((10, 10), f"{font_name}:", font=font, fill='black')
+        draw.text((10, 40), preview_text, font=font, fill='black')
+    except IOError:
+        draw.text((10, 10), f"Error loading font: {font_name}", fill='black')
+    return image
+
+def text_to_image(text, size=72, position="middle-center", font_name="Arial_Unicode.ttf"):
     width, height = 1024, 576
     image = Image.new("RGB", (width, height), "white")
     draw = ImageDraw.Draw(image)
 
-    font_files = ["Arial_Unicode.ttf"]
-    font = None
-    for font_file in font_files:
-        font_path = os.path.join(os.path.dirname(__file__), font_file)
-        if os.path.exists(font_path):
-            try:
-                font = ImageFont.truetype(font_path, size=size)
-                print(f"Using font: {font_file}")
-                break
-            except IOError:
-                print(f"Error loading font: {font_file}")
-    if font is None:
-        print("No suitable font found. Using default font.")
+    try:
+        font = ImageFont.truetype(font_name, size=size)
+    except IOError:
+        print(f"Error loading font: {font_name}. Using default font.")
         font = ImageFont.load_default()
 
     lines = text.split('\n')
@@ -153,7 +159,7 @@ def text_to_image(text, size=72, position="middle-center"):
     return image
 
 @spaces.GPU
-def infer_canny(prompt, text_for_image, text_position, font_size,
+def infer_canny(prompt, text_for_image, text_position, font_size, font_name,
           negative_prompt = "nsfw, facial shadows, low resolution, jpeg artifacts, blurry, bad quality, dark face, neon lights", 
           seed = 397886929, 
           randomize_seed = False,
@@ -171,8 +177,8 @@ def infer_canny(prompt, text_for_image, text_position, font_size,
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
 
-    # Generate text image
-    init_image = text_to_image(text_for_image, size=font_size, position=text_position)
+    # Generate text image with selected font
+    init_image = text_to_image(text_for_image, size=font_size, position=text_position, font_name=font_name)
     init_image = resize_image(init_image, MAX_IMAGE_SIZE)
 
     pipe = pipe_canny.to("cuda")
@@ -191,7 +197,7 @@ def infer_canny(prompt, text_for_image, text_position, font_size,
         num_images_per_prompt=1,
         generator=generator,
     ).images[0]
-    return image, seed  # CANNY 이미지 반환 제거
+    return image, seed
 
 def update_button_states(selected_position):
     return [
@@ -199,6 +205,20 @@ def update_button_states(selected_position):
         for pos in position_list
     ]
 
+
+def generate_font_preview(font_name):
+    font_size = 24
+    image = Image.new('RGB', (400, 100), color='white')
+    draw = ImageDraw.Draw(image)
+    try:
+        font = ImageFont.truetype(font_name, font_size)
+        draw.text((10, 10), f"{font_name}:", font=font, fill='black')
+        draw.text((10, 40), "AaBbCc 123 가나다", font=font, fill='black')
+    except IOError:
+        draw.text((10, 10), f"Error loading font: {font_name}", fill='black')
+    return image
+
+
 position_list = [
     "top-left", "top-left-center", "top-center", "top-right-center", "top-right",
     "upper-left", "upper-left-center", "upper-center", "upper-right-center", "upper-right",
@@ -207,7 +227,6 @@ position_list = [
     "bottom-left", "bottom-left-center", "bottom-center", "bottom-right-center", "bottom-right"
 ]
 
-
 css = """
 footer {
     visibility: hidden;
@@ -247,14 +266,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as Kolors:
                     label="Prompt",
                     placeholder="Enter your prompt",
                     lines=2,
-                    value="coffee in a cup bokeh --ar 85:128 --v 6.0 --style raw5, 4K, 리얼리티 사진"  # Default value added here
+                    value="coffee in a cup bokeh --ar 85:128 --v 6.0 --style raw5, 4K, 리얼리티 사진"
                 )
             with gr.Row():
                 text_for_image = gr.Textbox(
                     label="Text for Image Generation",
                     placeholder="Enter text to be converted into an image",
                     lines=3,
-                    value="대한 萬世 GO"  # Default value added here
+                    value="대한 萬世 GO"
                 )
             with gr.Row():
                 with gr.Column():
@@ -273,6 +292,15 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as Kolors:
                         step=1,
                         value=72
                     )
+            with gr.Row():
+                font_dropdown = gr.Dropdown(choices=get_font_files(), label="Select Font", value="Arial_Unicode.ttf")
+            
+            with gr.Row():
+                preview_text = gr.Textbox(label="Preview Text", value="Hello, World!")
+                preview_button = gr.Button("Preview Font")
+            
+            font_preview = gr.Image(label="Font Preview")
+
             with gr.Accordion("Advanced Settings", open=False):
                 negative_prompt = gr.Textbox(
                     label="Negative prompt",
@@ -329,18 +357,22 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as Kolors:
             with gr.Row():
                 canny_button = gr.Button("Start", elem_id="button")
 
-
         with gr.Column(elem_id="col-right"):
-            result = gr.Image(label="Result", show_label=False)  # Gallery에서 Image로 변경
+            result = gr.Image(label="Result", show_label=False)
             seed_used = gr.Number(label="Seed Used")
 
-    canny_button.click(
-        fn = infer_canny,
-        inputs = [prompt, text_for_image, text_position, font_size, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
-        outputs = [result, seed_used]
+    # Update font preview when button is clicked
+    preview_button.click(
+        fn=preview_font,
+        inputs=[font_dropdown, preview_text],
+        outputs=font_preview
     )
-        
 
+    canny_button.click(
+        fn=infer_canny,
+        inputs=[prompt, text_for_image, text_position, font_size, font_dropdown, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
+        outputs=[result, seed_used]
+    )
 
     # Set initial button states
     Kolors.load(update_button_states, inputs=[text_position], outputs=position_buttons)