Update app.py

app.py

@@ -1,11 +1,20 @@
-# img_bot.py  (변경된 부분 ★표시)
+# img_bot.py
+# “이미지 첨부 → 실패 시 URL 출력” 완결판
+
 import os, io, re, random, asyncio, logging, subprocess, base64
 from urllib.parse import urljoin, quote_plus
 
-import discord, requests, replicate
+import discord
+import requests
+import replicate
 from transformers import pipeline as transformers_pipeline
 from gradio_client import Client
-from PIL import Image                                     # WEBP → PNG 변환용
+
+try:
+    from PIL import Image                # WEBP → PNG 변환
+    PIL_OK = True
+except Exception:
+    PIL_OK = False                       # Pillow 미설치 시 변환 생략
 
 # ────────────────── 환경 변수 ──────────────────
 TOKEN       = os.getenv("DISCORD_TOKEN")
@@ -17,7 +26,8 @@ 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            # Replicate 미사용이지만 구조 유지
+
+os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN          # (Replicate 미사용)
 
 # ────────────────── Gradio 서버 ─────────────────
 GRADIO_URL = "http://211.233.58.201:7971"
@@ -32,6 +42,7 @@ translator = transformers_pipeline(
 )
 
 async def ko2en_async(text: str) -> str:
+    """한글 포함 시 영어 번역."""
     if not re.search(r"[가-힣]", text):
         return text
     loop = asyncio.get_running_loop()
@@ -95,26 +106,34 @@ class ImageBot(discord.Client):
             await message.reply("⚠️ 이미지 생성 실패!")
             return
 
-        if isinstance(result, list):        # 리스트면 첫 항목
+        # 리스트 결과 → 첫 요소
+        if isinstance(result, list):
             result = result[0]
 
         # ───────────── 이미지 데이터 확보 ─────────────
         data = None
-        candidate_urls = []
+        remote_path = None               # 실패 시 사용자에게 보여줄 URL
+        candidate_urls: list[str] = []
 
-        def add_remote(path: str):
-            candidate_urls.append(urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(path)}"))
+        def add_remote(p: str):
+            url = urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(p)}")
+            candidate_urls.append(url)
+            return url
 
         # dict 결과
         if isinstance(result, dict):
-            if result.get("url"):
-                u = result["url"]
-                candidate_urls.append(urljoin(GRADIO_URL, u) if u.startswith("/") else u)
-            if result.get("path"):
-                add_remote(result["path"])
-                if os.path.isfile(result["path"]):
+            u = result.get("url")
+            if u:
+                if u.startswith("/"):
+                    u = urljoin(GRADIO_URL, u)
+                candidate_urls.append(u)
+                remote_path = u
+            p = result.get("path")
+            if p:
+                remote_path = add_remote(p)
+                if os.path.isfile(p):
                     try:
-                        with open(result["path"], "rb") as f:
+                        with open(p, "rb") as f:
                             data = f.read()
                     except Exception:
                         pass
@@ -129,10 +148,11 @@ class ImageBot(discord.Client):
                     logging.warning(f"base64 디코딩 실패: {e}")
             elif s.startswith("http"):
                 candidate_urls.append(s)
-            elif s.startswith("/"):
-                candidate_urls.append(urljoin(GRADIO_URL, s))
+                remote_path = s
             else:
-                add_remote(s)
+                remote_path = add_remote(s)
+                if s.startswith("/"):
+                    candidate_urls.append(urljoin(GRADIO_URL, s))
                 if os.path.isfile(s):
                     try:
                         with open(s, "rb") as f:
@@ -140,7 +160,7 @@ class ImageBot(discord.Client):
                     except Exception:
                         pass
 
-        # URL 다운로드 (★ 헤더 검사 제거)
+        # URL 다운로드 (헤더 무시, 200 OK 만 확인)
         for u in candidate_urls:
             if data:
                 break
@@ -152,21 +172,28 @@ class ImageBot(discord.Client):
             except Exception as e:
                 logging.warning(f"URL 다운로드 실패({u}): {e}")
 
-        if not data:
-            await message.reply("⚠️ 이미지를 전송할 수 없습니다.")
+        # ───────────── Discord 전송 ─────────────
+        if data:
+            if PIL_OK:  # WEBP → PNG 변환
+                try:
+                    buf = io.BytesIO()
+                    Image.open(io.BytesIO(data)).convert("RGB").save(buf, format="PNG")
+                    buf.seek(0)
+                    await message.reply(files=[discord.File(buf, filename="image.png")])
+                    return
+                except Exception as e:
+                    logging.warning(f"PNG 변환 실패: {e}")
+            # 변환 실패 또는 Pillow 없음 → 원본 전송
+            await message.reply(files=[discord.File(io.BytesIO(data), filename="image.webp")])
             return
 
-        # ───────────── WEBP → PNG 변환 ─────────────
-        buf = io.BytesIO()
-        try:
-            Image.open(io.BytesIO(data)).convert("RGB").save(buf, format="PNG")
-            buf.seek(0)
-            await message.reply(files=[discord.File(buf, filename="image.png")])
-        except Exception as e:
-            logging.warning(f"PNG 변환 실패: {e}")
-            await message.reply(files=[discord.File(io.BytesIO(data), filename="image.webp")])
+        # ───────────── 실패 시 URL 출력 ─────────────
+        if remote_path:
+            await message.reply(content=remote_path)
+        else:
+            await message.reply("⚠️ 이미지를 전송할 수 없습니다.")
 
 # ────────────────── 실행 ────────────────────────
 if __name__ == "__main__":
-    replicate.Client(api_token=REPL_TOKEN)  # 구조 유지
+    replicate.Client(api_token=REPL_TOKEN)   # 구조 유지
     ImageBot(intents=intents).run(TOKEN)