Update app.py

app.py

@@ -1,30 +1,43 @@
 # img_bot.py
 import discord, os, io, asyncio, logging, requests, replicate, subprocess
 
-# ── 필수 환경 변수 ──────────────────────────────────────────────
-TOKEN       = os.getenv("DISCORD_TOKEN")                  # 디스코드 봇 토큰
-CHANNEL_ID  = int(os.getenv("DISCORD_CHANNEL_ID"))        # 감시·응답할 채널 ID
-REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip() # Replicate 토큰(같은 변수 재사용)
-# ───────────────────────────────────────────────────────────────
+# ── 환경 변수 ────────────────────────────────────────────────
+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 토큰(동일 변수 사용)
 
+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 값을 넣어주세요."
+    )
+
+# Replicate 라이브러리가 참조하도록 토큰 주입
+os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN
+
+# ── 모델 ────────────────────────────────────────────────────
 MODEL = (
     "bytedance/sdxl-lightning-4step:"
     "6f7a773af6fc3e8de9d5a3c00be77c17308914bf67772726aff83496ba1e3bbe"
 )
 
+# ── 로깅 ────────────────────────────────────────────────────
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s [%(levelname)s] %(message)s",
     handlers=[logging.StreamHandler()]
 )
 
+# ── Discord 설정 ────────────────────────────────────────────
 intents = discord.Intents.default()
-intents.message_content = True  # 텍스트 읽기 허용
+intents.message_content = True  # 메시지 콘텐츠 읽기
 
 class ImageBot(discord.Client):
     async def on_ready(self):
         logging.info(f"Logged in as {self.user} (id={self.user.id})")
-        # web.py 서버 기동
+        # web.py 병렬 실행
         try:
             subprocess.Popen(["python", "web.py"])
             logging.info("web.py server has been started.")
@@ -32,7 +45,7 @@ class ImageBot(discord.Client):
             logging.warning(f"web.py 실행 실패: {e}")
 
     async def on_message(self, message: discord.Message):
-        # 봇이 보낸 메시지이거나 다른 채널이면 무시
+        # 봇 자신의 메시지 혹은 대상 아닌 채널 → 무시
         if message.author.id == self.user.id or message.channel.id != CHANNEL_ID:
             return
 
@@ -42,18 +55,20 @@ class ImageBot(discord.Client):
 
         await message.channel.typing()
 
-        # Replicate 호출(옵션은 기본값: prompt만 전달)
+        # ── Replicate 호출 ──────────────────────────────────
         def run_replicate():
             return list(replicate.run(MODEL, input={"prompt": prompt}))
 
         try:
-            loop   = asyncio.get_running_loop()
-            images = await loop.run_in_executor(None, run_replicate)
+            images = await asyncio.get_running_loop().run_in_executor(
+                None, run_replicate
+            )
         except Exception as e:
             logging.error(f"Replicate error: {e}")
             await message.reply("⚠️ 이미지 생성 실패!")
             return
 
+        # ── 이미지 Discord 전송 ─────────────────────────────
         files = []
         for idx, item in enumerate(images):
             try:
@@ -62,10 +77,12 @@ class ImageBot(discord.Client):
             except Exception as e:
                 logging.warning(f"[IMG {idx}] 처리 실패: {e}")
 
-        await message.reply(files=files if files else None,
-                            content=None if files else "⚠️ 이미지를 전송할 수 없습니다.")
+        await message.reply(
+            files=files if files else None,
+            content=None if files else "⚠️ 이미지를 전송할 수 없습니다."
+        )
 
+# ── 실행 ────────────────────────────────────────────────────
 if __name__ == "__main__":
-    # Replicate 클라이언트 초기화
-    replicate.Client(api_token=REPL_TOKEN)
+    replicate.Client(api_token=REPL_TOKEN)  # 인증
     ImageBot(intents=intents).run(TOKEN)