Update app.py

app.py

@@ -1,4 +1,3 @@
-# HF Spaces
 import gradio as gr
 import asyncio
 import fal_client
@@ -9,6 +8,9 @@ import time
 import base64
 import json
 
+# 번역 모델 임포트
+from transformers import pipeline as translation_pipeline
+
 # Local Dev
 import os
 from dotenv import load_dotenv
@@ -16,11 +18,18 @@ from dotenv import load_dotenv
 load_dotenv()
 FAL_KEY = os.getenv("FAL_KEY")
 
+# 번역 모델 초기화
+translator = translation_pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
 with open("examples/examples.json") as f:
     examples = json.load(f)
 
 # IC Light, Replace Background
 async def submit_ic_light_bria(image_data, positive_prompt, negative_prompt, lightsource_start_color, lightsource_end_color):
+    # 프롬프트 번역
+    positive_prompt = translate_to_english(positive_prompt)
+    negative_prompt = translate_to_english(negative_prompt)
+
     if not lightsource_start_color.startswith("#"):
         lightsource_start_color = f"#{lightsource_start_color}"
     if not lightsource_end_color.startswith("#"):
@@ -68,6 +77,12 @@ async def submit_ic_light_bria(image_data, positive_prompt, negative_prompt, lig
             else:
                 return [f"Error: {str(e)}"], None
 
+def translate_to_english(text):
+    # 입력된 텍스트가 한글일 경우 번역 수행
+    if any('가' <= c <= '힣' for c in text):
+        return translator(text, max_length=512)[0]['translation_text']
+    return text
+
 # SDXL, Depth Anything, Replace Background
 async def submit_sdxl_rembg(image_data, positive_prompt, negative_prompt):
     retries = 3