Update app.py

app.py

@@ -20,10 +20,7 @@ if not REPL_TOKEN:
 os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN
 
 # ── 모델 ────────────────────────────────────────────────────
-MODEL = (
-    "bytedance/sdxl-lightning-4step:"
-    "6f7a773af6fc3e8de9d5a3c00be77c17308914bf67772726aff83496ba1e3bbe"
-)
+MODEL = "luma/ray-flash-2-540p"  # 텍스트-투-비디오 모델 (540p)
 
 # ── 번역 파이프라인 (CPU) ───────────────────────────────────
 translator_kwargs = {"device": -1}
@@ -56,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 병렬 실행
@@ -80,28 +77,33 @@ class ImageBot(discord.Client):
 
         # ── Replicate 호출 ──────────────────────────────────
         def run_replicate():
-            return list(replicate.run(MODEL, input={"prompt": prompt_en}))
+            # 필요한 경우 input 딕셔너리에 duration, seed 등 추가 가능
+            return replicate.run(MODEL, input={"prompt": prompt_en})
 
         try:
-            images = await asyncio.get_running_loop().run_in_executor(None, run_replicate)
+            output = await asyncio.get_running_loop().run_in_executor(None, run_replicate)
         except Exception as e:
             logging.error(f"Replicate error: {e}")
-            await message.reply("⚠️ 이미지 생성 실패!")
+            await message.reply("⚠️ 비디오 생성 실패!")
             return
 
-        # ── 이미지 Discord 전송 ─────────────────────────────
-        files = []
-        for idx, item in enumerate(images):
-            try:
-                data = item.read() if hasattr(item, "read") else requests.get(item).content
-                files.append(discord.File(io.BytesIO(data), filename=f"img_{idx}.png"))
-            except Exception as e:
-                logging.warning(f"[IMG {idx}] 처리 실패: {e}")
-
-        await message.reply(
-            files=files if files else None,
-            content=None if files else "⚠️ 이미지를 전송할 수 없습니다."
-        )
+        # ── 비디오 Discord 전송 ─────────────────────────────
+        try:
+            # replicate.run 결과가 list/tuple 이면 첫 번째 요소 사용
+            if isinstance(output, (list, tuple)):
+                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:
+            logging.warning(f"비디오 처리 실패: {e}")
+            await message.reply("⚠️ 비디오를 전송할 수 없습니다.")
 
 # ── 실행 ────────────────────────────────────────────────────
 if __name__ == "__main__":