Update app.py

app.py

@@ -10,6 +10,7 @@ from PIL import Image
 import io
 import gradio as gr
 import threading
+from googletrans import Translator
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
@@ -24,6 +25,7 @@ intents.guild_messages = True
 # PaliGemma 모델 설정 (CPU 모드)
 model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cpu").eval()
 processor = PaliGemmaProcessor.from_pretrained("gokaygokay/sd3-long-captioner")
+translator = Translator()
 
 def modify_caption(caption: str) -> str:
     prefix_substrings = [
@@ -56,9 +58,16 @@ async def create_captions_rich(image: Image.Image) -> str:
     modified_caption = modify_caption(decoded)
     return modified_caption
 
+async def translate_to_korean(text: str) -> str:
+    loop = asyncio.get_event_loop()
+    translated = await loop.run_in_executor(None, lambda: translator.translate(text, dest='ko').text)
+    return translated
+
 # Gradio 인터페이스 설정
 def create_captions_rich_sync(image):
-    return asyncio.run(create_captions_rich(image))
+    caption = asyncio.run(create_captions_rich(image))
+    translated_caption = asyncio.run(translate_to_korean(caption))
+    return translated_caption
 
 css = """
   #mkd {
@@ -125,7 +134,8 @@ class MyClient(discord.Client):
 async def process_image(image_url, message):
     image = await download_image(image_url)
     caption = await create_captions_rich(image)
-    return f"{message.author.mention}, 인식된 이미지 설명: {caption}"
+    translated_caption = await translate_to_korean(caption)
+    return f"{message.author.mention}, 인식된 이미지 설명: {translated_caption}"
 
 async def download_image(url):
     response = requests.get(url)