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, handle_file            # ⬅️ handle_file 추가
+from gradio_client import Client, handle_file
 
 # ── 환경 변수 ────────────────────────────────────────────────
 TOKEN       = os.getenv("DISCORD_TOKEN")
@@ -14,7 +14,7 @@ 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            # 구조 유지 (Replicate 미사용)
 
 # ── Gradio 서버 ─────────────────────────────────────────────
 GRADIO_URL  = "http://211.233.58.201:7896"
@@ -45,7 +45,8 @@ async def ko2en_async(text: str) -> str:
 
 # ── 로깅 ────────────────────────────────────────────────────
 logging.basicConfig(level=logging.INFO,
-                    format="%(asctime)s [%(levelname)s] %(message)s")
+                    format="%(asctime)s [%(levelname)s] %(message)s",
+                    handlers=[logging.StreamHandler()])
 
 # ── Discord 인텐트 ──────────────────────────────────────────
 intents = discord.Intents.default()
@@ -82,11 +83,11 @@ class ImageBot(discord.Client):
                 inference_steps=30,
                 seed=random.randint(0, 2**32 - 1),
                 do_img2img=False,
-                init_image=handle_file(DUMMY_IMG),   # ⬅️ path 포함 dict
+                init_image=handle_file(DUMMY_IMG),  # path·mime 포함 dict
                 image2image_strength=0.8,
                 resize_img=True,
                 api_name=GRADIO_API
-            )[0]                                    # dict(path|url)
+            )[0]                                   # str or dict
 
         try:
             img_info = await asyncio.get_running_loop().run_in_executor(None, generate_image)
@@ -98,12 +99,23 @@ class ImageBot(discord.Client):
         # ── Discord 전송 ──────────────────────────────────
         files = []
         try:
-            if img_info.get("path"):
-                with open(img_info["path"], "rb") as f:
-                    files.append(discord.File(io.BytesIO(f.read()), filename="generated.webp"))
-            elif img_info.get("url"):
-                data = requests.get(img_info["url"]).content
+            if isinstance(img_info, str):          # 경우 1: 문자열 경로/URL
+                if img_info.startswith("http"):
+                    data = requests.get(img_info).content
+                else:
+                    with open(img_info, "rb") as f:
+                        data = f.read()
                 files.append(discord.File(io.BytesIO(data), filename="generated.webp"))
+            elif isinstance(img_info, dict):       # 경우 2: dict(path|url)
+                if img_info.get("path"):
+                    with open(img_info["path"], "rb") as f:
+                        data = f.read()
+                elif img_info.get("url"):
+                    data = requests.get(img_info["url"]).content
+                else:
+                    data = None
+                if data:
+                    files.append(discord.File(io.BytesIO(data), filename="generated.webp"))
         except Exception as e:
             logging.warning(f"이미지 처리 실패: {e}")
 
@@ -114,5 +126,5 @@ class ImageBot(discord.Client):
 
 # ── 실행 ────────────────────────────────────────────────────
 if __name__ == "__main__":
-    replicate.Client(api_token=REPL_TOKEN)  # 구조 유지(실제 사용 안 함)
+    replicate.Client(api_token=REPL_TOKEN)  # 구조 유지(사용 안 함)
     ImageBot(intents=intents).run(TOKEN)