Update app.py

app.py

@@ -1,7 +1,8 @@
 # img_bot.py
-import discord, os, io, re, random, asyncio, logging, requests, replicate, subprocess
+import discord, os, io, re, random, asyncio, logging, requests, replicate, subprocess, base64
+from urllib.parse import urljoin
 from transformers import pipeline as transformers_pipeline
-from gradio_client import Client, handle_file
+from gradio_client import Client
 
 # ── 환경 변수 ────────────────────────────────────────────────
 TOKEN       = os.getenv("DISCORD_TOKEN")
@@ -19,7 +20,6 @@ os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN            # 구조 유지 (Repli
 # ── Gradio 서버 ─────────────────────────────────────────────
 GRADIO_URL  = "http://211.233.58.201:7971"
 GRADIO_API  = "/process_and_save_image"
-DUMMY_IMG   = "https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png"
 
 # ── 번역 파이프라인 (CPU) ───────────────────────────────────
 translator = transformers_pipeline(
@@ -30,7 +30,6 @@ translator = transformers_pipeline(
 )
 
 async def ko2en_async(text: str) -> str:
-    """한글 포함 시 별도 쓰레드에서 영어 번역."""
     if not re.search(r"[가-힣]", text):
         return text
     loop = asyncio.get_running_loop()
@@ -50,7 +49,7 @@ logging.basicConfig(level=logging.INFO,
 
 # ── Discord 인텐트 ──────────────────────────────────────────
 intents = discord.Intents.default()
-intents.message_content = True   # Portal에서도 Message-Content Intent ON
+intents.message_content = True
 
 class ImageBot(discord.Client):
     async def on_ready(self):
@@ -78,15 +77,16 @@ class ImageBot(discord.Client):
             return client.predict(
                 height=768,
                 width=768,
-                steps=30,                       # inference_steps → steps
-                scales=3.5,                     # guidance → scales
+                steps=30,
+                scales=3.5,
                 prompt=prompt_en,
                 seed=random.randint(0, 2**32 - 1),
                 api_name=GRADIO_API
-            )[0]                               # dict(path|url|…)
+            )[0]  # list → dict
 
         try:
             img_info = await asyncio.get_running_loop().run_in_executor(None, generate_image)
+            logging.info(f"Gradio result: {img_info}")
         except Exception as e:
             logging.error(f"Gradio API error: {e}")
             await message.reply("⚠️ 이미지 생성 실패!")
@@ -94,21 +94,37 @@ class ImageBot(discord.Client):
 
         # ── Discord 전송 ──────────────────────────────────
         files = []
-        try:
-            if isinstance(img_info, str):      # 경우 1: 문자열 경로/URL
-                data = requests.get(img_info).content if img_info.startswith("http") else open(img_info, "rb").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"):
-                    data = open(img_info["path"], "rb").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}")
+        data = None
+
+        # 1️⃣ url 필드 우선
+        url = img_info.get("url")
+        if url:
+            if url.startswith("data:"):  # base64 data URI
+                b64 = re.sub(r"^data:image/[^;]+;base64,", "", url)
+                try:
+                    data = base64.b64decode(b64)
+                except Exception as e:
+                    logging.warning(f"base64 디코딩 실패: {e}")
+            else:
+                if url.startswith("/"):  # 상대 URL
+                    url = urljoin(GRADIO_URL, url)
+                try:
+                    data = requests.get(url, timeout=10).content
+                except Exception as e:
+                    logging.warning(f"URL 다운로드 실패: {e}")
+
+        # 2️⃣ path 필드 (파일 존재할 때만)
+        if data is None:
+            path = img_info.get("path")
+            if path and os.path.isfile(path):
+                try:
+                    with open(path, "rb") as f:
+                        data = f.read()
+                except Exception as e:
+                    logging.warning(f"파일 열기 실패: {e}")
+
+        if data:
+            files.append(discord.File(io.BytesIO(data), filename="generated.webp"))
 
         await message.reply(
             files=files if files else None,
@@ -117,5 +133,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)