From 2f38db1af50cb98ea81ffab24ca8552cbc9ba6b9 Mon Sep 17 00:00:00 2001 From: "kyj@bowong.ai" Date: Fri, 25 Jul 2025 16:43:46 +0800 Subject: [PATCH] FIX --- __init__.py | 6 +++--- nodes/image_modal_nodes.py | 7 ++++--- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/__init__.py b/__init__.py index 8c65a6e..dd65599 100644 --- a/__init__.py +++ b/__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":"顺序合并视频" } diff --git a/nodes/image_modal_nodes.py b/nodes/image_modal_nodes.py index 7aa34ad..ade2efb 100644 --- a/nodes/image_modal_nodes.py +++ b/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,)