Update app.py

app.py

@@ -5,16 +5,16 @@ from transformers import pipeline as transformers_pipeline   # 번역 파이프
 # ── 환경 변수 ────────────────────────────────────────────────
 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 Personal Access Token
+REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip()    # Replicate 토큰 재사용
+HF_TOKEN    = (os.getenv("HF_TOKEN") or "").strip()          # Hugging Face 토큰
+USE_MSG_INTENT = os.getenv("MESSAGE_CONTENT_INTENT", "1") != "0"  # 1=ON(기본), 0=OFF
 
 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
+os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN       # Replicate 가 참조할 토큰
 
 # ── 이미지 생성 모델 (Flux Schnell) ─────────────────────────
 MODEL = "black-forest-labs/flux-schnell"
@@ -23,15 +23,12 @@ MODEL = "black-forest-labs/flux-schnell"
 translator_kwargs = {"device": -1}
 if HF_TOKEN:
     translator_kwargs["token"] = HF_TOKEN
-
 translator = transformers_pipeline(
-    "translation",
-    model="Helsinki-NLP/opus-mt-ko-en",
-    **translator_kwargs
+    "translation", model="Helsinki-NLP/opus-mt-ko-en", **translator_kwargs
 )
 
 def ko2en(text: str) -> str:
-    """프롬프트에 한글이 있으면 영어로 번역하여 반환."""
+    """프롬프트에 한글 포함 시 영어 번역."""
     if re.search(r"[가-힣]", text):
         try:
             return translator(text, max_length=512)[0]["translation_text"].strip()
@@ -40,15 +37,17 @@ def ko2en(text: str) -> str:
     return text
 
 # ── 로깅 ────────────────────────────────────────────────────
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s [%(levelname)s] %(message)s",
-    handlers=[logging.StreamHandler()]
-)
+logging.basicConfig(level=logging.INFO,
+                    format="%(asctime)s [%(levelname)s] %(message)s")
 
-# ── Discord 설정 ────────────────────────────────────────────
+# ── Discord 인텐트 설정 ────────────────────────────────────
 intents = discord.Intents.default()
-intents.message_content = True  # 메시지 콘텐츠 읽기
+if USE_MSG_INTENT:
+    intents.message_content = True
+    logging.info("Message Content Intent = ON  "
+                 "(Developer Portal → Bot → Privileged Intents 에서 ON 필요)")
+else:
+    logging.info("Message Content Intent = OFF  (MESSAGE_CONTENT_INTENT=0)")
 
 class ImageBot(discord.Client):
     async def on_ready(self):
@@ -61,7 +60,9 @@ class ImageBot(discord.Client):
             logging.warning(f"web.py 실행 실패: {e}")
 
     async def on_message(self, message: discord.Message):
-        # 봇 자신의 메시지 혹은 대상 아닌 채널 → 무시
+        # 메시지 내용 읽을 권한이 없으면 바로 반환
+        if not USE_MSG_INTENT:
+            return
         if message.author.id == self.user.id or message.channel.id != CHANNEL_ID:
             return
 
@@ -69,17 +70,12 @@ class ImageBot(discord.Client):
         if not prompt_raw:
             return
 
-        prompt_en = ko2en(prompt_raw)  # 한글이면 영어로 변환
+        prompt_en = ko2en(prompt_raw)
         await message.channel.typing()
 
-        # ── Replicate 호출 ──────────────────────────────────
+        # ── Replicate 요청 ─────────────────────────────────
         def run_replicate():
-            return list(
-                replicate.run(
-                    MODEL,
-                    input={"prompt": prompt_en}
-                )
-            )
+            return list(replicate.run(MODEL, input={"prompt": prompt_en}))
 
         try:
             images = await asyncio.get_running_loop().run_in_executor(None, run_replicate)
@@ -88,7 +84,7 @@ class ImageBot(discord.Client):
             await message.reply("⚠️ 이미지 생성 실패!")
             return
 
-        # ── 이미지 Discord 전송 ─────────────────────────────
+        # ��─ Discord 전송 ──────────────────────────────────
         files = []
         for idx, item in enumerate(images):
             try:
@@ -97,12 +93,10 @@ 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.Client(api_token=REPL_TOKEN)  # Replicate 인증
+    replicate.Client(api_token=REPL_TOKEN)   # Replicate 인증
     ImageBot(intents=intents).run(TOKEN)