Update app.py

app.py

@@ -7,7 +7,7 @@ import requests
 import replicate
 from transformers import pipeline as transformers_pipeline
 from gradio_client import Client
-from PIL import Image     # WEBP → PNG 변환용
+from PIL import Image                         # WEBP → PNG 변환용
 
 # ────────────────── 환경 변수 ──────────────────
 TOKEN       = os.getenv("DISCORD_TOKEN")
@@ -98,57 +98,82 @@ class ImageBot(discord.Client):
             await message.reply("⚠️ 이미지 생성 실패!")
             return
 
-        # Gradio 결과 → list / dict / str 중 첫 요소로 정규화
+        # 결과 정규화(list → 첫 요소)
         if isinstance(result, list):
             result = result[0]
 
-        # ───────────── URL 있으면 링크로 전송 ─────────────
-        url = None
-        if isinstance(result, dict):
-            url = result.get("url")
-            if url and url.startswith("/"):
-                url = urljoin(GRADIO_URL, url)
-        elif isinstance(result, str) and result.startswith(("http", "/")):
-            url = result if result.startswith("http") else urljoin(GRADIO_URL, result)
-
-        if url:
-            await message.reply(content=url)   # Discord가 embed 미리보기
-            return
-
-        # ───────────── 첨부 파일 준비 ─────────────
+        # ───────────── 이미지 데이터 확보 ─────────────
         data = None
-        # dict → path
+        candidate_urls = []
+
+        # dict 결과 처리
         if isinstance(result, dict):
+            # url 필드
+            url = result.get("url")
+            if url:
+                if url.startswith("/"):
+                    url = urljoin(GRADIO_URL, url)
+                candidate_urls.append(url)
+            # path 필드
             path = result.get("path")
             if path:
-                # Gradio 파일 다운로드 우선
-                remote = urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(path)}")
-                try:
-                    data = requests.get(remote, timeout=10).content
-                except Exception:
-                    if os.path.isfile(path):
+                # ① 원격 Gradio 파일 경로
+                candidate_urls.append(urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(path)}"))
+                # ② 로컬 파일 경로
+                if os.path.isfile(path):
+                    try:
                         with open(path, "rb") as f:
                             data = f.read()
-        # str → data URI
-        elif isinstance(result, str) and result.startswith("data:"):
+                    except Exception as e:
+                        logging.warning(f"로컬 파일 열기 실패: {e}")
+
+        # str 결과 처리
+        elif isinstance(result, str):
+            s = result.strip()
+            if s.startswith("data:"):
+                # data URI
+                try:
+                    data = base64.b64decode(re.sub(r"^data:image/[^;]+;base64,", "", s))
+                except Exception as e:
+                    logging.warning(f"base64 디코딩 실패: {e}")
+            elif s.startswith("http"):
+                candidate_urls.append(s)
+            elif s.startswith("/"):
+                candidate_urls.append(urljoin(GRADIO_URL, s))
+            else:  # 로컬/원격 tmp 경로
+                candidate_urls.append(urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(s)}"))
+                if os.path.isfile(s):
+                    try:
+                        with open(s, "rb") as f:
+                            data = f.read()
+                    except Exception as e:
+                        logging.warning(f"로컬 파일 열기 실패: {e}")
+
+        # URL 목록에서 다운로드
+        for u in candidate_urls:
+            if data:
+                break
             try:
-                b64 = re.sub(r"^data:image/[^;]+;base64,", "", result)
-                data = base64.b64decode(b64)
+                r = requests.get(u, timeout=10)
+                if r.ok and r.headers.get("content-type", "").startswith("image/"):
+                    data = r.content
+                    break
             except Exception as e:
-                logging.warning(f"base64 디코딩 실패: {e}")
+                logging.warning(f"URL 다운로드 실패({u}): {e}")
 
         if not data:
             await message.reply("⚠️ 이미지를 전송할 수 없습니다.")
             return
 
-        # ───────────── WEBP → PNG 변환 (미리보기용) ─────────────
+        # ───────────── 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:
-            # 변환 실패 시 원본 그대로
+        except Exception as e:
+            logging.warning(f"PNG 변환 실패: {e}")
+            # 변환 실패 → 원본 확장자 유지
             await message.reply(files=[discord.File(io.BytesIO(data), filename="image.webp")])
 
 # ────────────────── 실행 ────────────────────────