Update app.py

app.py

@@ -11,6 +11,30 @@ from PIL import Image
 from diffusers.utils import load_image
 from diffusers.utils import export_to_video
 import random
+from transformers import pipeline
+
+# 번역 모델 로드
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
+# 한글 메뉴 이름 dictionary
+korean_labels = {
+    "Prompt": "프롬프트",
+    "1st direction to steer": "첫 번째 방향",
+    "2nd direction to steer": "두 번째 방향",
+    "Strength": "강도",
+    "Generate directions": "방향 생성",
+    "Generated Images": "생성된 이미지",
+    "From 1st to 2nd direction": "첫 번째에서 두 번째 방향으로",
+    "Strip": "이미지 스트립",
+    "Looping video": "루프 비디오",
+    "Advanced options": "고급 옵션",
+    "Num of intermediate images": "중간 이미지 수",
+    "Num iterations for clip directions": "클립 방향 반복 횟수",
+    "Num inference steps": "추론 단계 수",
+    "Guidance scale": "가이던스 스케일",
+    "Randomize seed": "시드 무작위화",
+    "Seed": "시드"
+}
 
 # load pipelines
 base_model = "black-forest-labs/FLUX.1-schnell"
@@ -50,6 +74,11 @@ def convert_to_centered_scale(num):
         end = num // 2 
     return tuple(range(start, end + 1))
 
+def translate_if_korean(text):
+    if any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text):
+        return translator(text)[0]['translation_text']
+    return text
+
 @spaces.GPU(duration=85)
 def generate(prompt,
              concept_1,
@@ -67,6 +96,11 @@ def generate(prompt,
              total_images=[],
              progress=gr.Progress()
     ):
+    # 프롬프트와 컨셉 번역
+    prompt = translate_if_korean(prompt)
+    concept_1 = translate_if_korean(concept_1)
+    concept_2 = translate_if_korean(concept_2)
+    
     print(f"Prompt: {prompt}, ← {concept_2}, {concept_1} ➡️ . scale {scale}, interm steps {interm_steps}")
     slider_x = [concept_2, concept_1]
     # check if avg diff for directions need to be re-calculated
@@ -117,8 +151,6 @@ def update_pre_generated_images(slider_value, total_images):
 def reset_recalc_directions():
     return True
 
-
-
 examples = [["a dog in the park", "winter", "summer", 1.5], ["a house", "USA suburb", "Europe", 2.5], ["a tomato", "rotten", "super fresh", 2.5]]
 
 css = """
@@ -127,11 +159,7 @@ footer {
 }
 """
 
-
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
-
-
-    
     x_concept_1 = gr.State("")
     x_concept_2 = gr.State("")
     total_images = gr.Gallery(visible=False)
@@ -143,37 +171,37 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     with gr.Row():
         with gr.Column():
             with gr.Group():
-                prompt = gr.Textbox(label="Prompt", info="Describe what to be steered by the directions", placeholder="A dog in the park")
+                prompt = gr.Textbox(label=korean_labels["Prompt"], info="설명할 내용을 입력하세요", placeholder="공원에 있는 강아지")
                 with gr.Row():
-                    concept_1 = gr.Textbox(label="1st direction to steer", info="Starting state", placeholder="winter")
-                    concept_2 = gr.Textbox(label="2nd direction to steer", info="Finishing state", placeholder="summer")
-            x = gr.Slider(minimum=0, value=1.75, step=0.1, maximum=4.0, label="Strength", info="maximum strength on each direction (unstable beyond 2.5)")
-            submit = gr.Button("Generate directions")
+                    concept_1 = gr.Textbox(label=korean_labels["1st direction to steer"], info="시작 상태", placeholder="겨울")
+                    concept_2 = gr.Textbox(label=korean_labels["2nd direction to steer"], info="종료 상태", placeholder="여름")
+            x = gr.Slider(minimum=0, value=1.75, step=0.1, maximum=4.0, label=korean_labels["Strength"], info="각 방향의 최대 강도 (2.5 이상은 불안정)")
+            submit = gr.Button(korean_labels["Generate directions"])
         with gr.Column():
             with gr.Group(elem_id="group"):
-                post_generation_image = gr.Image(label="Generated Images", type="filepath", elem_id="interactive")
-                post_generation_slider = gr.Slider(minimum=-10, maximum=10, value=0, step=1, label="From 1st to 2nd direction")
+                post_generation_image = gr.Image(label=korean_labels["Generated Images"], type="filepath", elem_id="interactive")
+                post_generation_slider = gr.Slider(minimum=-10, maximum=10, value=0, step=1, label=korean_labels["From 1st to 2nd direction"])
     with gr.Row():
         with gr.Column(scale=4):
-            image_seq = gr.Image(label="Strip", elem_id="strip", height=80)
+            image_seq = gr.Image(label=korean_labels["Strip"], elem_id="strip", height=80)
         with gr.Column(scale=2, min_width=100):
-            output_image = gr.Video(label="Looping video", elem_id="video", loop=True, autoplay=True)
-    with gr.Accordion(label="Advanced options", open=False):
-        interm_steps = gr.Slider(label = "Num of intermediate images", minimum=3, value=7, maximum=65, step=2)
+            output_image = gr.Video(label=korean_labels["Looping video"], elem_id="video", loop=True, autoplay=True)
+    with gr.Accordion(label=korean_labels["Advanced options"], open=False):
+        interm_steps = gr.Slider(label=korean_labels["Num of intermediate images"], minimum=3, value=7, maximum=65, step=2)
         with gr.Row():
-            iterations = gr.Slider(label = "Num iterations for clip directions", minimum=0, value=200, maximum=400, step=1)
-            steps = gr.Slider(label = "Num inference steps", minimum=1, value=3, maximum=4, step=1)
+            iterations = gr.Slider(label=korean_labels["Num iterations for clip directions"], minimum=0, value=200, maximum=400, step=1)
+            steps = gr.Slider(label=korean_labels["Num inference steps"], minimum=1, value=3, maximum=4, step=1)
         with gr.Row():
             guidance_scale = gr.Slider(
-                label="Guidance scale",
+                label=korean_labels["Guidance scale"],
                 minimum=0.1,
                 maximum=10.0,
                 step=0.1,
                 value=3.5,
             )
             with gr.Column():
-                randomize_seed = gr.Checkbox(True, label="Randomize seed")
-                seed = gr.Slider(minimum=0, maximum=MAX_SEED, step=1, label="Seed", interactive=True, randomize=True)
+                randomize_seed = gr.Checkbox(True, label=korean_labels["Randomize seed"])
+                seed = gr.Slider(minimum=0, maximum=MAX_SEED, step=1, label=korean_labels["Seed"], interactive=True, randomize=True)
 
     examples_gradio = gr.Examples(
         examples=examples,