Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 import time
 from diffusers import DiffusionPipeline, AutoencoderTiny
 from custom_pipeline import FluxWithCFGPipeline
+from transformers import pipeline
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
@@ -14,6 +15,9 @@ DEFAULT_WIDTH = 1024
 DEFAULT_HEIGHT = 768
 DEFAULT_INFERENCE_STEPS = 4
 
+# Initialize translator
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
 # Device and model setup
 dtype = torch.float16
 pipe = FluxWithCFGPipeline.from_pretrained(
@@ -23,15 +27,25 @@ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtyp
 pipe.to("cuda")
 torch.cuda.empty_cache()
 
+# Translation function
+def translate_to_english(text):
+    if any(ord('가') <= ord(char) <= ord('힣') for char in text):
+        translated = translator(text)[0]['translation_text']
+        return translated
+    return text
+
 # Inference function
 @spaces.GPU(duration=25)
 def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, randomize_seed=False, progress=gr.Progress(track_tqdm=True)):
+    # Translate prompt if Korean
+    english_prompt = translate_to_english(prompt)
+    
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(int(float(seed)))
 
     img = pipe.generate_images( 
-            prompt=prompt,
+            prompt=english_prompt,
             width=width,
             height=height,
             num_inference_steps=DEFAULT_INFERENCE_STEPS,
@@ -70,53 +84,4 @@ footer {visibility: hidden;}
 .generate-box .row {display: flex; align-items: center; margin-bottom: 10px;}
 .generate-box .row > * {margin-right: 10px;}
 .generate-box .row > *:last-child {margin-right: 0;}
-.advanced-options {background-color: #e0e0e0; border-radius: 10px; padding: 20px; margin-top: 20px;}
-.examples-gallery {margin-top: 30px;}
-"""
-
-# --- Gradio UI ---
-with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
-    with gr.Column(elem_id="container"):
-        gr.Markdown("# Open FLUX 1.1 Pro")
-        gr.Markdown("Flux Schnell-based with no commercial restrictions, 4-step fast image generation with quality enhancement, and improved memory efficiency (VAE).")        
-
-        with gr.Row():
-            with gr.Column(scale=2):
-                result = gr.Image(label="Generated Image", show_label=False, interactive=False, elem_classes="image-box")
-            with gr.Column(scale=1):
-                with gr.Column(elem_classes="generate-box"):
-                    prompt = gr.Text(
-                        label="Prompt",
-                        placeholder="sexy woman & man , under wear, full body, sunday",
-                        lines=3,
-                    )
-                    generateBtn = gr.Button("Generate Image", variant="primary")
-
-                    with gr.Column(elem_classes="advanced-options"):
-                        with gr.Row():
-                            seed = gr.Number(label="Seed", value=42)
-                            randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                        with gr.Row():
-                            width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
-                            height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
-
-        with gr.Column(elem_classes="examples-gallery"):
-            gr.Markdown("### Gallery")
-            gr.Examples(
-                examples=examples,
-                fn=generate_image,
-                inputs=[prompt],
-                outputs=[result, seed],
-                cache_examples="lazy" 
-            )
-
-    generateBtn.click(
-        fn=generate_image,
-        inputs=[prompt, seed, width, height, randomize_seed],
-        outputs=[result, seed],
-        show_progress="full",
-        api_name="GenerateImage",
-    )
-
-# Launch the app
-demo.launch()
+.advanced-options {background-color: #e0e0e0; border-radius