Update app.py

app.py

@@ -1,24 +1,25 @@
 # 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                           # Gradio API
+from gradio_client import Client                                        # Gradio API
 
 # ── 환경 변수 ────────────────────────────────────────────────
-TOKEN       = os.getenv("DISCORD_TOKEN")
-CHANNEL_ID  = int(os.getenv("DISCORD_CHANNEL_ID"))
-REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip()
-HF_TOKEN    = (os.getenv("HF_TOKEN") or "").strip()
+TOKEN       = os.getenv("DISCORD_TOKEN")                 # Discord 봇 토큰
+CHANNEL_ID  = int(os.getenv("DISCORD_CHANNEL_ID"))       # 감시할 채널 ID
+REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip()# Replicate 토큰(구조 유지)
+HF_TOKEN    = (os.getenv("HF_TOKEN") or "").strip()      # Hugging-Face 토큰(옵션)
 
 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   # 구조 유지 (실제 사용 X)
+os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN           # 구조 유지
 
 # ── Gradio 서버 ─────────────────────────────────────────────
 GRADIO_URL  = "http://211.233.58.201:7896"
 GRADIO_API  = "/generate_image"
+DUMMY_IMG   = "https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png"
 
 # ── 번역 파이프라인 (CPU) ───────────────────────────────────
 translator = transformers_pipeline(
@@ -29,6 +30,7 @@ translator = transformers_pipeline(
 )
 
 def ko2en(text: str) -> str:
+    """한글 포함 시 영어로 번역."""
     if re.search(r"[가-힣]", text):
         try:
             return translator(text, max_length=512)[0]["translation_text"].strip()
@@ -38,16 +40,17 @@ def ko2en(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 인텐트 (항상 ON → 포털에서도 Message Content ON) ─
+# ── Discord 인텐트 ──────────────────────────────────────────
 intents = discord.Intents.default()
-intents.message_content = True
-logging.info("Message Content Intent = ON  (Portal → Bot → Privileged Intents 에서도 ON 필요)")
+intents.message_content = True   # 항상 ON (포털에서도 Message-Content Intent ON)
 
 class ImageBot(discord.Client):
     async def on_ready(self):
         logging.info(f"Logged in as {self.user} (id={self.user.id})")
+        # web.py 병렬 실행
         try:
             subprocess.Popen(["python", "web.py"])
             logging.info("web.py server has been started.")
@@ -77,12 +80,12 @@ class ImageBot(discord.Client):
                 inference_steps=30,
                 seed=seed,
                 do_img2img=False,
-                init_image={},              # ← 빈 dict (규격 만족)
+                init_image={"url": DUMMY_IMG},   # 필수 placeholder
                 image2image_strength=0.8,
                 resize_img=True,
                 api_name=GRADIO_API
             )
-            return result[0]  # dict(path|url|...)
+            return result[0]                     # dict(path|url)
 
         try:
             img_info = await asyncio.get_running_loop().run_in_executor(None, generate_image)
@@ -100,8 +103,6 @@ class ImageBot(discord.Client):
             elif img_info.get("url"):
                 data = requests.get(img_info["url"]).content
                 files.append(discord.File(io.BytesIO(data), filename="generated.webp"))
-            else:
-                raise ValueError("No image data returned")
         except Exception as e:
             logging.warning(f"이미지 처리 실패: {e}")
 
@@ -110,5 +111,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)