Update app.py

app.py

@@ -1,27 +1,27 @@
 # img_bot.py
 import discord, os, io, re, random, asyncio, logging, requests, replicate, subprocess
 from transformers import pipeline as transformers_pipeline
-from gradio_client import Client                                        # Gradio API
+from gradio_client import Client
 
 # ── 환경 변수 ────────────────────────────────────────────────
-TOKEN       = os.getenv("DISCORD_TOKEN")                 # Discord 봇 토큰
-CHANNEL_ID  = int(os.getenv("DISCORD_CHANNEL_ID"))       # 감시할 채널 ID
-REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip()# Replicate 토큰(구조 유지)
-HF_TOKEN    = (os.getenv("HF_TOKEN") or "").strip()      # Hugging-Face 토큰(옵션)
+TOKEN       = os.getenv("DISCORD_TOKEN")
+CHANNEL_ID  = int(os.getenv("DISCORD_CHANNEL_ID"))
+REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip()
+HF_TOKEN    = (os.getenv("HF_TOKEN") or "").strip()
 
 if not TOKEN or not CHANNEL_ID:
     raise RuntimeError("DISCORD_TOKEN 과 DISCORD_CHANNEL_ID 환경 변수를 모두 지정하세요.")
 if not REPL_TOKEN:
     raise RuntimeError("OPENAI_API_KEY 에 Replicate Personal Access Token 값을 넣어주세요.")
 
-os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN           # 구조 유지
+os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN    # 구조 유지
 
 # ── Gradio 서버 ─────────────────────────────────────────────
 GRADIO_URL  = "http://211.233.58.201:7896"
 GRADIO_API  = "/generate_image"
 DUMMY_IMG   = "https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png"
 
-# ── 번역 파이프라인 (CPU) ───────────────────────────────────
+# ── 번역 파이프라인(마리안, CPU) ────────────────────────────
 translator = transformers_pipeline(
     "translation",
     model="Helsinki-NLP/opus-mt-ko-en",
@@ -29,28 +29,32 @@ translator = transformers_pipeline(
     **({"token": HF_TOKEN} if HF_TOKEN else {})
 )
 
-def ko2en(text: str) -> str:
-    """한글 포함 시 영어로 번역."""
-    if re.search(r"[가-힣]", text):
-        try:
-            return translator(text, max_length=512)[0]["translation_text"].strip()
-        except Exception as e:
-            logging.warning(f"번역 실패, 원문 사용: {e}")
-    return text
+async def ko2en_async(text: str) -> str:
+    """한글 포함 시 별도 쓰레드에서 영어 번역."""
+    if not re.search(r"[가-힣]", text):
+        return text
+    loop = asyncio.get_running_loop()
+    try:
+        result = await loop.run_in_executor(
+            None,
+            lambda: translator(text, max_length=256, num_beams=1)[0]["translation_text"].strip()
+        )
+        return result
+    except Exception as e:
+        logging.warning(f"번역 실패, 원문 사용: {e}")
+        return text
 
 # ── 로깅 ────────────────────────────────────────────────────
 logging.basicConfig(level=logging.INFO,
-                    format="%(asctime)s [%(levelname)s] %(message)s",
-                    handlers=[logging.StreamHandler()])
+                    format="%(asctime)s [%(levelname)s] %(message)s")
 
 # ── Discord 인텐트 ──────────────────────────────────────────
 intents = discord.Intents.default()
-intents.message_content = True   # 항상 ON (포털에서도 Message-Content Intent ON)
+intents.message_content = True   # Portal에서도 Message-Content Intent ON
 
 class ImageBot(discord.Client):
     async def on_ready(self):
         logging.info(f"Logged in as {self.user} (id={self.user.id})")
-        # web.py 병렬 실행
         try:
             subprocess.Popen(["python", "web.py"])
             logging.info("web.py server has been started.")
@@ -65,27 +69,23 @@ class ImageBot(discord.Client):
         if not prompt_raw:
             return
 
-        prompt_en = ko2en(prompt_raw)
+        prompt_en = await ko2en_async(prompt_raw)   # ✔️ 비동기 번역
         await message.channel.typing()
 
-        # ── Gradio 호출 ────────────────────────────────────
+        # ── Gradio 호출 (블로킹 → executor) ─────────────────
         def generate_image():
             client = Client(GRADIO_URL)
-            seed = random.randint(0, 2**32 - 1)
-            result = client.predict(
+            return client.predict(
                 prompt=prompt_en,
-                width=768,
-                height=768,
-                guidance=3.5,
-                inference_steps=30,
-                seed=seed,
+                width=768, height=768,
+                guidance=3.5, inference_steps=30,
+                seed=random.randint(0, 2**32 - 1),
                 do_img2img=False,
-                init_image={"url": DUMMY_IMG},   # 필수 placeholder
+                init_image={"url": DUMMY_IMG},
                 image2image_strength=0.8,
                 resize_img=True,
                 api_name=GRADIO_API
-            )
-            return result[0]                     # dict(path|url)
+            )[0]                                    # dict(path|url)
 
         try:
             img_info = await asyncio.get_running_loop().run_in_executor(None, generate_image)