Update app.py

app.py

@@ -1,36 +1,33 @@
 # img_bot.py
 import discord, os, io, re, asyncio, logging, requests, replicate, subprocess
-from transformers import pipeline as transformers_pipeline   # 번역 파이프라인
+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
+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 토큰
+# message_content 인텐트 사용 여부(1 = ON, 0 = OFF)
+USE_MSG_INTENT = os.getenv("MESSAGE_CONTENT_INTENT", "1") != "0"
 
 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 값을 넣어주세요."
-    )
+    raise RuntimeError("OPENAI_API_KEY 에 Replicate Personal Access Token 값을 넣어주세요.")
 
 # Replicate 라이브러리가 참조하도록 토큰 주입
 os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN
 
 # ── 모델 ────────────────────────────────────────────────────
 MODEL = (
-    "nvidia/sana-sprint-1.6b:"
-    "6ed1ce77cdc8db65550e76d5ab82556d0cb31ac8ab3c4947b168a0bda7b962e4"
-
+    "bytedance/sdxl-lightning-4step:"
+    "6f7a773af6fc3e8de9d5a3c00be77c17308914bf67772726aff83496ba1e3bbe"
 )
 
 # ── 번역 파이프라인 (CPU) ───────────────────────────────────
 translator_kwargs = {"device": -1}
 if HF_TOKEN:
-    translator_kwargs["token"] = HF_TOKEN        # 인증 토큰 전달
-
+    translator_kwargs["token"] = HF_TOKEN
 translator = transformers_pipeline(
     "translation",
     model="Helsinki-NLP/opus-mt-ko-en",
@@ -38,7 +35,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()
@@ -55,7 +52,12 @@ logging.basicConfig(
 
 # ── Discord 설정 ────────────────────────────────────────────
 intents = discord.Intents.default()
-intents.message_content = True  # 메시지 콘텐츠 읽기
+if USE_MSG_INTENT:
+    intents.message_content = True
+    logging.info("Message Content Intent = ON  "
+                 "(Discord Developer Portal → Bot → Privileged Intents 에서 활성화 필요)")
+else:
+    logging.info("Message Content Intent = OFF  (MESSAGE_CONTENT_INTENT=0)")
 
 class ImageBot(discord.Client):
     async def on_ready(self):
@@ -68,15 +70,18 @@ class ImageBot(discord.Client):
             logging.warning(f"web.py 실행 실패: {e}")
 
     async def on_message(self, message: discord.Message):
-        # 봇 자신의 메시지 혹은 대상 아닌 채널 → 무시
+        # 인텐트 OFF 시 프롬프트를 읽을 수 없으므로 즉시 반환
+        if not USE_MSG_INTENT:
+            return
+        # 봇 자신의 메시지 또는 다른 채널 메시지 → 무시
         if message.author.id == self.user.id or message.channel.id != CHANNEL_ID:
             return
 
-        prompt = message.content.strip()
-        if not prompt:
+        prompt_raw = message.content.strip()
+        if not prompt_raw:
             return
+        prompt_en = ko2en(prompt_raw)
 
-        prompt_en = ko2en(prompt)  # 한글이면 영어로 변환
         await message.channel.typing()
 
         # ── Replicate 호출 ──────────────────────────────────
@@ -90,7 +95,7 @@ class ImageBot(discord.Client):
             await message.reply("⚠️ 이미지 생성 실패!")
             return
 
-        # ── 이미지 Discord 전송 ─────────────────────────────
+        # ── Discord 파일 업로드 ─────────────────────────────
         files = []
         for idx, item in enumerate(images):
             try: