Update app.py

app.py

@@ -1,13 +1,11 @@
-# img_bot.py
+# img_bot.py  (변경된 부분 ★표시)
 import os, io, re, random, asyncio, logging, subprocess, base64
 from urllib.parse import urljoin, quote_plus
 
-import discord
-import requests
-import replicate
+import discord, requests, 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")
@@ -19,7 +17,7 @@ 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"
@@ -34,7 +32,6 @@ translator = transformers_pipeline(
 )
 
 async def ko2en_async(text: str) -> str:
-    """한글 포함 시 영어 번역(비동기)."""
     if not re.search(r"[가-힣]", text):
         return text
     loop = asyncio.get_running_loop()
@@ -98,40 +95,34 @@ class ImageBot(discord.Client):
             await message.reply("⚠️ 이미지 생성 실패!")
             return
 
-        # 결과 정규화(list → 첫 요소)
-        if isinstance(result, list):
+        if isinstance(result, list):        # 리스트면 첫 항목
             result = result[0]
 
         # ───────────── 이미지 데이터 확보 ─────────────
         data = None
         candidate_urls = []
 
-        # dict 결과 처리
+        def add_remote(path: str):
+            candidate_urls.append(urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(path)}"))
+
+        # 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 파일 경로
-                candidate_urls.append(urljoin(GRADIO_URL, f"/gradio_api/file={quote_plus(path)}"))
-                # ② 로컬 파일 경로
-                if os.path.isfile(path):
+            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"]):
                     try:
-                        with open(path, "rb") as f:
+                        with open(result["path"], "rb") as f:
                             data = f.read()
-                    except Exception as e:
-                        logging.warning(f"로컬 파일 열기 실패: {e}")
+                    except Exception:
+                        pass
 
-        # str 결과 처리
+        # 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:
@@ -140,22 +131,22 @@ class ImageBot(discord.Client):
                 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)}"))
+            else:
+                add_remote(s)
                 if os.path.isfile(s):
                     try:
                         with open(s, "rb") as f:
                             data = f.read()
-                    except Exception as e:
-                        logging.warning(f"로컬 파일 열기 실패: {e}")
+                    except Exception:
+                        pass
 
-        # URL 목록에서 다운로드
+        # URL 다운로드 (★ 헤더 검사 제거)
         for u in candidate_urls:
             if data:
                 break
             try:
                 r = requests.get(u, timeout=10)
-                if r.ok and r.headers.get("content-type", "").startswith("image/"):
+                if r.ok and r.content:
                     data = r.content
                     break
             except Exception as e:
@@ -173,7 +164,6 @@ class ImageBot(discord.Client):
             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")])
 
 # ────────────────── 실행 ────────────────────────