Update app.py

app.py

@@ -1,10 +1,15 @@
 import random
+import re
 import gradio as gr
 import numpy as np
 import spaces
 import torch
 from diffusers import AutoencoderKL
 from mixture_tiling_sdxl import StableDiffusionXLTilingPipeline
+from transformers import pipeline
+
+# 한글 입력 시 영어로 번역하기 위한 파이프라인 (CPU에서 동작)
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 MAX_SEED = np.iinfo(np.int32).max
 SCHEDULERS = [
@@ -36,7 +41,15 @@ pipe.enable_model_cpu_offload()  # VRAM이 제한된 경우 사용
 pipe.enable_vae_tiling()
 pipe.enable_vae_slicing()
 
-#region functions
+#region helper functions
+def translate_if_needed(text: str) -> str:
+    """텍스트에 한글이 포함된 경우 영어로 번역하여 반환."""
+    if re.search(r'[\uac00-\ud7a3]', text):
+        # 번역 결과는 리스트의 첫번째 요소의 "translation_text" 키에 있음
+        translated = translator(text)[0]["translation_text"]
+        return translated
+    return text
+
 def select_scheduler(scheduler_name):
     scheduler_parts = scheduler_name.split("-")
     scheduler_class_name = scheduler_parts[0]
@@ -63,6 +76,11 @@ def predict(left_prompt, center_prompt, right_prompt, negative_prompt, left_gs,
     pipe.scheduler = select_scheduler(scheduler)
     generator = torch.Generator("cuda").manual_seed(generation_seed)
     
+    # 한글 프롬프트가 입력된 경우 영어로 번역
+    left_prompt = translate_if_needed(left_prompt)
+    center_prompt = translate_if_needed(center_prompt)
+    right_prompt = translate_if_needed(right_prompt)
+    
     target_height = int(target_height)
     target_width = int(target_width)
     tile_height = int(tile_height)
@@ -221,7 +239,9 @@ button:hover, .btn:hover {
 
 title = """
 <h1 align="center" style="margin-bottom: 0.2em;"> 🤗 Panorama X3 Image </h1>
-
+<p align="center" style="font-size:1.1em; color:#555;">
+    Left/Center/Right prompts (English/Korean allowed)
+</p>
 """
 
 with gr.Blocks(css=css, title="SDXL Tiling Pipeline") as app:
@@ -234,21 +254,21 @@ with gr.Blocks(css=css, title="SDXL Tiling Pipeline") as app:
             with gr.Row():
                 with gr.Column(variant="panel"):
                     gr.Markdown("### Left Region")
-                    left_prompt = gr.Textbox(lines=4, placeholder="예: 울창한 숲과 햇살이 비추는 나무...", label="Left Prompt")
+                    left_prompt = gr.Textbox(lines=4, placeholder="e.g., 울창한 숲과 햇살이 비추는 나무...", label="Left Prompt (English/Korean allowed)")
                     left_gs = gr.Slider(minimum=0, maximum=15, value=7, step=1, label="Left CFG Scale")
                 with gr.Column(variant="panel"):
                     gr.Markdown("### Center Region")
-                    center_prompt = gr.Textbox(lines=4, placeholder="예: 잔잔한 호수와 반짝이는 수면...", label="Center Prompt")
+                    center_prompt = gr.Textbox(lines=4, placeholder="e.g., 잔잔한 호수와 반짝이는 수면...", label="Center Prompt (English/Korean allowed)")
                     center_gs = gr.Slider(minimum=0, maximum=15, value=7, step=1, label="Center CFG Scale")
                 with gr.Column(variant="panel"):
                     gr.Markdown("### Right Region")
-                    right_prompt = gr.Textbox(lines=4, placeholder="예: 웅장한 산맥과 하늘을 가르는 구름...", label="Right Prompt")
+                    right_prompt = gr.Textbox(lines=4, placeholder="e.g., 웅장한 산맥과 하늘을 가르는 구름...", label="Right Prompt (English/Korean allowed)")
                     right_gs = gr.Slider(minimum=0, maximum=15, value=7, step=1, label="Right CFG Scale")
             with gr.Row():
                 negative_prompt = gr.Textbox(
                     lines=2,
                     label="Negative Prompt",
-                    placeholder="예: blurry, low resolution, artifacts, poor details",
+                    placeholder="e.g., blurry, low resolution, artifacts, poor details",
                     value="blurry, low resolution, artifacts, poor details"
                 )
             with gr.Row():
@@ -296,7 +316,6 @@ with gr.Blocks(css=css, title="SDXL Tiling Pipeline") as app:
                         1024, 1920,
                         1280
                     ],
-
                     [
                         "Vibrant abstract strokes with fluid, swirling patterns in cool tones",
                         "Interlocking geometric shapes bursting with color and texture",