FIX

__init__.py

@@ -85,9 +85,9 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Jinja2RenderTemplate": "Jinja2格式Prompt模板渲染",
     "JMGestureCorrect": "人物侧身图片转为正面图-即梦",
     "ModalClothesMask": "模特指定衣服替换为指定颜色",
-    "ModalEditCustom": "Prompt生/修图",
-    "ModalMidJourneyGenerateImage": "Prompt生/修图",
-    "ModalMidJourneyDescribeImage": "反推生图提示词",
+    "ModalEditCustom": "Prompt生/修图-Gemini",
+    "ModalMidJourneyGenerateImage": "Prompt生/修图-MJ",
+    "ModalMidJourneyDescribeImage": "反推生图提示词-MJ",
     "JMCustom": "Prompt生视频",
     "VideoMerge":"顺序合并视频"
 }

nodes/image_modal_nodes.py

@@ -1,5 +1,6 @@
 import io
 import json
+import traceback
 from time import sleep
 
 import folder_paths
@@ -241,8 +242,8 @@ class ModalMidJourneyGenerateImage:
                                     timeout=150)
             job_resp.raise_for_status()
             job_resp = job_resp.json()
-            if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]:
-                raise Exception("生成失败")
+            if not job_resp["status"]:
+                raise Exception("生成失败, 可能因为风控")
             job_id = job_resp["data"]
             for _ in range(0, timeout // interval, interval):
                 logger.info("查询结果")
@@ -296,7 +297,7 @@ class ModalMidJourneyDescribeImage:
                                     timeout=150)
             job_resp.raise_for_status()
             job_resp = job_resp.json()
-            if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]:
+            if not job_resp["status"]:
                 raise Exception("描述失败")
             result = job_resp["data"]
             return (result,)