Update app.py

app.py

@@ -73,6 +73,12 @@ if torch.cuda.is_available():
 
 @spaces.GPU(duration=300)
 def infer_t2i(prompt, seed=42, randomize_seed=False, width=1024, height=576, guidance_scale=5.0, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+    # 한글 입력 감지 및 번역
+    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
+        translated = translator(prompt, max_length=512)[0]['translation_text']
+        prompt = translated
+        print(f"Translated prompt: {prompt}")
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
@@ -88,7 +94,7 @@ def infer_t2i(prompt, seed=42, randomize_seed=False, width=1024, height=576, gui
         ).images[0]
     
     torch.cuda.empty_cache()
-    return image, seed
+    return image, seed, prompt  # 번역된 프롬프트도 반환
 
     
 @spaces.GPU(duration=300)
@@ -244,10 +250,15 @@ with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
             t2i_output_image = gr.Image(label="Generated Image", elem_id="t2i_output_img")
             t2i_output_seed = gr.Number(label="Used Seed", elem_id="t2i_output_seed")
 
-        t2i_generate_btn.click(
-            fn=infer_t2i,
-            inputs=[t2i_input_text, t2i_seed, t2i_randomize_seed, t2i_width, t2i_height, t2i_guidance_scale, t2i_num_inference_steps],
-            outputs=[t2i_output_image, t2i_output_seed]
-        )
+        t2i_generate_btn = gr.Button("Generate")
+        t2i_output_image = gr.Image(label="Generated Image", elem_id="t2i_output_img")
+        t2i_output_seed = gr.Number(label="Used Seed", elem_id="t2i_output_seed")
+        t2i_translated_prompt = gr.Text(label="Translated Prompt (if applicable)", elem_id="t2i_translated_prompt")
 
+    t2i_generate_btn.click(
+        fn=infer_t2i,
+        inputs=[t2i_input_text, t2i_seed, t2i_randomize_seed, t2i_width, t2i_height, t2i_guidance_scale, t2i_num_inference_steps],
+        outputs=[t2i_output_image, t2i_output_seed, t2i_translated_prompt]
+    )
+    
 dynamicrafter_iface.queue(max_size=12).launch(show_api=True)