Update app.py

app.py

@@ -14,6 +14,7 @@ import numpy as np
 import torch
 from PIL import Image, ImageDraw, ImageFont
 from diffusers import FluxPipeline
+from transformers import pipeline
 
 # 메모리 정리 함수
 def clear_memory():
@@ -55,6 +56,21 @@ pipe = FluxPipeline.from_pretrained(
 )
 pipe.enable_attention_slicing(slice_size="auto")
 
+
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=-1)  # CPU에서 실행
+
+def translate_to_english(text: str) -> str:
+    """한글 텍스트를 영어로 번역"""
+    try:
+        if any(ord('가') <= ord(char) <= ord('힣') for char in text):
+            translated = translator(text, max_length=128)[0]['translation_text']
+            print(f"Translated '{text}' to '{translated}'")
+            return translated
+        return text
+    except Exception as e:
+        print(f"Translation error: {str(e)}")
+        return text
+        
 # Eric cat LoRA 가중치 로드
 try:
     lora_path = hf_hub_download(
@@ -91,7 +107,7 @@ def save_generated_image(image, prompt):
     image.save(filepath)
     return filepath
 
-@spaces.GPU(duration=60)  # gr.GPU 대신 spaces.GPU 사용
+@spaces.GPU(duration=60)
 def generate_image(
     prompt: str,
     seed: int,
@@ -103,6 +119,9 @@ def generate_image(
     progress: gr.Progress = gr.Progress()
 ):
     try:
+        # 프롬프트 번역 추가
+        prompt = translate_to_english(prompt)
+        
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)