Update app.py

app.py

@@ -1,29 +1,31 @@
 # img_bot.py
 import discord, os, io, re, asyncio, logging, requests, replicate, subprocess
-import tempfile, pathlib, contextlib
-from PIL import Image                                  # ← NEW: 이미지 변환용
-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 Personal Access Token
+REPL_TOKEN  = (os.getenv("OPENAI_API_KEY") or "").strip()    # Replicate 토큰(동일 변수 사용)
 HF_TOKEN    = (os.getenv("HF_TOKEN") or "").strip()          # Hugging Face Personal Access Token
 
 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 값을 넣어주세요."
+    )
 
-os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN   # Replicate 라이브러리에 토큰 주입
+# Replicate 라이브러리가 참조하도록 토큰 주입
+os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN
 
 # ── 모델 ────────────────────────────────────────────────────
-MODEL = "luma/ray-flash-2-540p"                  # 최신 Luma Ray Flash 2 (540p)
+MODEL = "luma/ray-flash-2-540p"  # 텍스트-투-비디오 모델 (540p)
 
 # ── 번역 파이프라인 (CPU) ───────────────────────────────────
 translator_kwargs = {"device": -1}
 if HF_TOKEN:
-    translator_kwargs["token"] = HF_TOKEN
+    translator_kwargs["token"] = HF_TOKEN        # 인증 토큰 전달
 
 translator = transformers_pipeline(
     "translation",
@@ -32,7 +34,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()
@@ -40,27 +42,6 @@ def ko2en(text: str) -> str:
             logging.warning(f"번역 실패, 원문 사용: {e}")
     return text
 
-# ── 보조: 이미지를 PNG 로 변환 후 Replicate 에 업로드 ───────────
-def upload_image_as_png(url: str) -> str:
-    """
-    Discord CDN URL → PNG 파일로 변환 → replicate.files.upload 로 업로드
-    Returns: HTTPS URL (replicate.delivery/*)
-    """
-    try:
-        resp = requests.get(url, timeout=30)
-        resp.raise_for_status()
-        img = Image.open(io.BytesIO(resp.content)).convert("RGB")     # webp 등 → rgb
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            img.save(tmp, format="PNG")
-            tmp_path = tmp.name
-        uploaded_url = replicate.files.upload(tmp_path)
-        return uploaded_url
-    finally:
-        # 임시 파일 정리
-        with contextlib.suppress(Exception):
-            if 'tmp_path' in locals():
-                pathlib.Path(tmp_path).unlink(missing_ok=True)
-
 # ── 로깅 ────────────────────────────────────────────────────
 logging.basicConfig(
     level=logging.INFO,
@@ -72,7 +53,7 @@ logging.basicConfig(
 intents = discord.Intents.default()
 intents.message_content = True  # 메시지 콘텐츠 읽기
 
-class ImageBot(discord.Client):
+class ImageBot(discord.Client):  # 이름 유지
     async def on_ready(self):
         logging.info(f"Logged in as {self.user} (id={self.user.id})")
         # web.py 병렬 실행
@@ -83,46 +64,21 @@ 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
 
-        prompt_raw: str = (message.content or "").strip()
-        # 프롬프트·이미지 모두 없으면 무시
-        if not prompt_raw and not message.attachments:
+        prompt = message.content.strip()
+        if not prompt:
             return
 
-        prompt_en = ko2en(prompt_raw)
+        prompt_en = ko2en(prompt)  # 한글이면 영어로 변환
         await message.channel.typing()
 
-        # ── 첨부 이미지 → start/end URL 변환 ─────────────────────
-        start_url, end_url = None, None
-        if message.attachments:
-            img_atts = [
-                att for att in message.attachments
-                if (att.content_type or "").startswith("image")
-            ]
-            if img_atts:
-                try:
-                    start_url = upload_image_as_png(img_atts[0].url)
-                    if len(img_atts) > 1:
-                        end_url = upload_image_as_png(img_atts[1].url)
-                except Exception as e:
-                    logging.warning(f"첨부 이미지 처리 실패: {e}")
-
         # ── Replicate 호출 ──────────────────────────────────
         def run_replicate():
-            input_dict = {"prompt": prompt_en}          # 필수
-            if start_url:
-                input_dict["start_image_url"] = start_url
-            if end_url:
-                input_dict["end_image_url"] = end_url
-            # FileOutput → URL 문자열만 받도록 설정
-            return replicate.run(
-                MODEL,
-                input=input_dict,
-                use_file_output=False
-            )
+            # 필요한 경우 input 딕셔너리에 duration, seed 등 추가 가능
+            return replicate.run(MODEL, input={"prompt": prompt_en})
 
         try:
             output = await asyncio.get_running_loop().run_in_executor(None, run_replicate)
@@ -133,10 +89,16 @@ class ImageBot(discord.Client):
 
         # ── 비디오 Discord 전송 ─────────────────────────────
         try:
+            # replicate.run 결과가 list/tuple 이면 첫 번째 요소 사용
             if isinstance(output, (list, tuple)):
-                output = output[0]          # URL 문자열
-            # URL → bytes 다운로드
-            data = requests.get(output, timeout=60).content
+                output = output[0]
+
+            # URL(str) → 다운로드, bytes/file-like → 그대로
+            if isinstance(output, str):
+                data = requests.get(output).content
+            else:
+                data = output.read() if hasattr(output, "read") else output
+
             video_file = discord.File(io.BytesIO(data), filename="output.mp4")
             await message.reply(files=[video_file])
         except Exception as e: