Update app.py

app.py

@@ -3,8 +3,10 @@ import logging
 import os
 import uuid
 import torch
+import subprocess
 from huggingface_hub import snapshot_download
 from diffusers import StableDiffusion3Pipeline, StableDiffusion3Img2ImgPipeline
+from transformers import pipeline
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
@@ -35,6 +37,9 @@ def load_pipeline(pipeline_type):
 # 디바이스 설정
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
+# 번역 파이프라인 설정
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
 # 디스코드 봇 클래스
 class MyClient(discord.Client):
     def __init__(self, *args, **kwargs):
@@ -45,6 +50,8 @@ class MyClient(discord.Client):
 
     async def on_ready(self):
         logging.info(f'{self.user}로 로그인되었습니다!')
+        subprocess.Popen(["python", "web.py"])
+        logging.info("web.py 서버가 시작되었습니다.")
 
     async def on_message(self, message):
         if message.author == self.user:
@@ -53,7 +60,9 @@ class MyClient(discord.Client):
             self.is_processing = True
             try:
                 prompt = message.content[len('!image '):]
-                image_path = await self.generate_image(prompt)
+                prompt_en = translate_prompt(prompt)
+                logging.debug(f'Translated prompt: {prompt_en}')
+                image_path = await self.generate_image(prompt_en)
                 await message.channel.send(file=discord.File(image_path, 'generated_image.png'))
             finally:
                 self.is_processing = False
@@ -65,6 +74,14 @@ class MyClient(discord.Client):
         images[0].save(image_path)
         return image_path
 
+# 프롬프트 번역 함수
+def translate_prompt(prompt):
+    logging.debug(f'Translating prompt: {prompt}')
+    translation = translator(prompt, max_length=512)
+    translated_text = translation[0]['translation_text']
+    logging.debug(f'Translated text: {translated_text}')
+    return translated_text
+
 # 디스코드 토큰 및 봇 실행
 if __name__ == "__main__":
     discord_token = os.getenv('DISCORD_TOKEN')