Update app.py

app.py

@@ -30,27 +30,27 @@ def check_text(prompt, negative=""):
 # Kid-friendly styles
 style_list = [
     {
-        "name": "Cartoon",
+        "name": "만화",
         "prompt": "colorful cartoon {prompt}. vibrant, playful, friendly, suitable for children, highly detailed, bright colors",
         "negative_prompt": "scary, dark, violent, ugly, realistic",
     },
     {
-        "name": "Children's Illustration",
+        "name": "어린이 일러스트",
         "prompt": "children's illustration {prompt}. cute, colorful, fun, simple shapes, smooth lines, highly detailed, joyful",
         "negative_prompt": "scary, dark, violent, deformed, ugly",
     },   
     {
-        "name": "Sticker",
+        "name": "스티커",
         "prompt": "children's sticker of {prompt}. bright colors, playful, high resolution, cartoonish",
         "negative_prompt": "scary, dark, violent, ugly, low resolution",
     },
     {
-        "name": "Fantasy",
+        "name": "판타지",
         "prompt": "fantasy world for children with {prompt}. magical, vibrant, friendly, beautiful, colorful",
         "negative_prompt": "dark, scary, violent, ugly, realistic",
     },
     {
-        "name": "(No style)",
+        "name": "(스타일 없음)",
         "prompt": "{prompt}",
         "negative_prompt": "",
     },
@@ -58,22 +58,21 @@ style_list = [
 
 styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
-DEFAULT_STYLE_NAME = "Sticker"
+DEFAULT_STYLE_NAME = "스티커"
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     return p.replace("{prompt}", positive), n + negative
 
-DESCRIPTION = """## Children's Sticker Generator
-
-Generate fun and playful stickers for children using AI.
+DESCRIPTION = """## 어린이 스티커 생성기
+AI를 사용하여 재미있고 귀여운 어린이 스티커를 생성합니다.
 """
 
 if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>⚠️Running on CPU, This may not work on CPU.</p>"
+    DESCRIPTION += "\n<p>⚠️CPU에서 실행 중입니다. CPU에서는 제대로 작동하지 않을 수 있습니다.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
-CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
+CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -149,7 +148,7 @@ def generate(
     prompt = translate_if_korean(prompt)
     
     if check_text(prompt, negative_prompt):
-        raise ValueError("Prompt contains restricted words.")
+        raise ValueError("프롬프트에 제한된 단어가 포함되어 있습니다.")
     
     # Ensure prompt is 2-3 words long
     prompt = " ".join(re.findall(r'\w+', prompt)[:3])
@@ -198,56 +197,55 @@ footer {
 }
 """
 
-
 # Define the Gradio UI for the sticker generator
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(
-                label="Enter your prompt",
+                label="프롬프트 입력",
                 show_label=False,
                 max_lines=1,
-                placeholder="프롬프트에 입력을 하세요.",
+                placeholder="프롬프트를 입력하세요.",
                 container=False,
             )
-            run_button = gr.Button("Run")
-        result = gr.Gallery(label="Generated Stickers", columns=2, preview=True)
-    with gr.Accordion("Advanced options", open=False):
-        use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
+            run_button = gr.Button("실행")
+        result = gr.Gallery(label="생성된 스티커", columns=2, preview=True)
+    with gr.Accordion("고급 옵션", open=False):
+        use_negative_prompt = gr.Checkbox(label="네거티브 프롬프트 사용", value=True, visible=True)
         negative_prompt = gr.Text(
-            label="Negative prompt",
+            label="네거티브 프롬프트",
             max_lines=1,
-            placeholder="Enter a negative prompt",
-            value="(scary, violent, dark, ugly)",
+            placeholder="네거티브 프롬프트를 입력하세요",
+            value="(무서운, 폭력적인, 어두운, 추한)",
             visible=True,
         )
         seed = gr.Slider(
-            label="Seed",
+            label="시드",
             minimum=0,
             maximum=MAX_SEED,
             step=1,
             value=0,
             visible=True
         )
-        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+        randomize_seed = gr.Checkbox(label="랜덤 시드", value=True)
         size_selection = gr.Radio(
             choices=["75mm", "35mm"],
             value="75mm",
-            label="Sticker Size",
+            label="스티커 크기",
         )
         style_selection = gr.Radio(
             choices=STYLE_NAMES,
             value=DEFAULT_STYLE_NAME,
-            label="Image Style",
+            label="이미지 스타일",
         )
         background_selection = gr.Radio(
-            choices=["transparent", "white"],
-            value="white",
-            label="transparent",
+            choices=["투명", "흰색"],
+            value="흰색",
+            label="배경색",
         )
         guidance_scale = gr.Slider(
-            label="Guidance Scale",
+            label="가이던스 스케일",
             minimum=0.1,
             maximum=20.0,
             step=0.1,