Update app.py

app.py

@@ -1,7 +1,7 @@
 # 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
+from gradio_client import Client, handle_file            # ⬅️ handle_file 추가
 
 # ── 환경 변수 ────────────────────────────────────────────────
 TOKEN       = os.getenv("DISCORD_TOKEN")
@@ -14,14 +14,14 @@ if not TOKEN or not 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",
@@ -35,11 +35,10 @@ async def ko2en_async(text: str) -> str:
         return text
     loop = asyncio.get_running_loop()
     try:
-        result = await loop.run_in_executor(
+        return 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
@@ -69,19 +68,21 @@ class ImageBot(discord.Client):
         if not prompt_raw:
             return
 
-        prompt_en = await ko2en_async(prompt_raw)   # ✔️ 비동기 번역
+        prompt_en = await ko2en_async(prompt_raw)
         await message.channel.typing()
 
-        # ── Gradio 호출 (블로킹 → executor) ─────────────────
+        # ── Gradio 호출 ────────────────────────────────────
         def generate_image():
             client = Client(GRADIO_URL)
             return client.predict(
                 prompt=prompt_en,
-                width=768, height=768,
-                guidance=3.5, inference_steps=30,
+                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},
+                init_image=handle_file(DUMMY_IMG),   # ⬅️ path 포함 dict
                 image2image_strength=0.8,
                 resize_img=True,
                 api_name=GRADIO_API
@@ -106,8 +107,10 @@ class ImageBot(discord.Client):
         except Exception as e:
             logging.warning(f"이미지 처리 실패: {e}")
 
-        await message.reply(files=files if files else None,
-                            content=None if files else "⚠️ 이미지를 전송할 수 없습니다.")
+        await message.reply(
+            files=files if files else None,
+            content=None if files else "⚠️ 이미지를 전송할 수 없습니다."
+        )
 
 # ── 실행 ────────────────────────────────────────────────────
 if __name__ == "__main__":