Update app.py

app.py

@@ -16,16 +16,13 @@ if not REPL_TOKEN:
 # Replicate 라이브러리가 참조하도록 토큰 주입
 os.environ["REPLICATE_API_TOKEN"] = REPL_TOKEN
 
-# ── 이미지 생성 모델 (변경된 부분) ────────────────────────────
-MODEL = (
-    "nvidia/sana-sprint-1.6b:"
-    "6ed1ce77cdc8db65550e76d5ab82556d0cb31ac8ab3c4947b168a0bda7b962e4"
-)
+# ── 이미지 생성 모델 (Flux Schnell) ─────────────────────────
+MODEL = "black-forest-labs/flux-schnell"
 
 # ── 번역 파이프라인 (CPU) ───────────────────────────────────
 translator_kwargs = {"device": -1}
 if HF_TOKEN:
-    translator_kwargs["token"] = HF_TOKEN        # 인증 토큰 전달
+    translator_kwargs["token"] = HF_TOKEN
 
 translator = transformers_pipeline(
     "translation",
@@ -34,7 +31,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()
@@ -77,16 +74,12 @@ class ImageBot(discord.Client):
 
         # ── Replicate 호출 ──────────────────────────────────
         def run_replicate():
-            # 모델에 따라 필요한 입력 파라미터 이름이 다를 수 있습니다.
-            # 일반적으로 "prompt" 키를 사용하지만 필요하면 수정하세요.
-            try:
-                result = replicate.run(MODEL, input={"prompt": prompt_en})
-            except replicate.exceptions.ReplicateError:
-                # 파라미터 이름이 다를 경우 빈 dict로 재시도
-                result = replicate.run(MODEL, input={})
-
-            # 결과가 리스트가 아닐 수 있으므로 리스트로 감싸 반환
-            return result if isinstance(result, list) else [result]
+            return list(
+                replicate.run(
+                    MODEL,
+                    input={"prompt": prompt_en}
+                )
+            )
 
         try:
             images = await asyncio.get_running_loop().run_in_executor(None, run_replicate)
@@ -100,7 +93,7 @@ class ImageBot(discord.Client):
         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}.jpg"))
+                files.append(discord.File(io.BytesIO(data), filename=f"img_{idx}.webp"))
             except Exception as e:
                 logging.warning(f"[IMG {idx}] 처리 실패: {e}")