Update app.py

app.py

@@ -18,6 +18,21 @@ from refiners.fluxion.utils import no_grad
 from refiners.solutions import BoxSegmenter
 from transformers import GroundingDinoForObjectDetection, GroundingDinoProcessor
 from diffusers import FluxPipeline
+# 상단에 import 추가
+from transformers import pipeline
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
+# 번역 함수 추가
+def translate_to_english(text: str) -> str:
+    """한글 텍스트를 영어로 번역"""
+    if any(ord('가') <= ord(char) <= ord('힣') for char in text):
+        try:
+            translated = translator(text)[0]['translation_text']
+            return translated
+        except Exception as e:
+            print(f"Translation error: {e}")
+            return text
+    return text
 
 BoundingBox = tuple[int, int, int, int]
 
@@ -284,11 +299,18 @@ def on_change_bbox(prompts: dict[str, Any] | None):
 def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str | None = None):
     return gr.update(interactive=bool(img and prompt))
 
+
+# process_prompt 함수 수정
 def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None, aspect_ratio: str = "1:1") -> tuple[Image.Image, Image.Image]:
     try:
         if img is None or prompt.strip() == "":
             raise gr.Error("Please provide both image and prompt")
         
+        # 프롬프트 번역
+        prompt = translate_to_english(prompt)
+        if bg_prompt:
+            bg_prompt = translate_to_english(bg_prompt)
+        
         # Process the image
         results, _ = _process(img, prompt, bg_prompt, aspect_ratio)