bowong
/
ComfyUI-CustomNode
6
0
Fork 0
Code Pull Requests Releases Activity

FIX midjourney节点改为修图节点

This commit is contained in:
kyj@bowong.ai 2025-07-14 11:43:41 +08:00
parent c7051da39f
commit 3192a9b4af
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
__init__.py
View File

@ -83,6 +83,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"JMGestureCorrect": "人物侧身图片转为正面图-即梦", "JMGestureCorrect": "人物侧身图片转为正面图-即梦",
"ModalClothesMask": "模特指定衣服替换为指定颜色", "ModalClothesMask": "模特指定衣服替换为指定颜色",
"ModalEditCustom": "自定义Prompt修改图片", "ModalEditCustom": "自定义Prompt修改图片",
"ModalMidJourneyGenerateImage": "Prompt生图", "ModalMidJourneyGenerateImage": "Prompt修图",
"ModalMidJourneyDescribeImage": "描述图片内容" "ModalMidJourneyDescribeImage": "描述图片内容"
} }

24
nodes/image_modal_nodes.py
View File

@ -202,6 +202,7 @@ class ModalMidJourneyGenerateImage:
return { return {
"required": { "required": {
"prompt": ("STRING", {"default": "一幅宏大壮美的山川画卷", "multiline": True}), "prompt": ("STRING", {"default": "一幅宏大壮美的山川画卷", "multiline": True}),
"image": ("IMAGE",),
"endpoint": ("STRING", {"default": "bowongai-dev--bowong-ai-video-gemini-fastapi-webapp.modal.run"}), "endpoint": ("STRING", {"default": "bowongai-dev--bowong-ai-video-gemini-fastapi-webapp.modal.run"}),
}, },
} }
@ -212,15 +213,19 @@ class ModalMidJourneyGenerateImage:
OUTPUT_NODE = False OUTPUT_NODE = False
CATEGORY = "不忘科技-自定义节点🚩/图片/Midjourney" CATEGORY = "不忘科技-自定义节点🚩/图片/Midjourney"
def process(self, prompt: str, endpoint: str): def process(self, prompt: str, image:torch.Tensor, endpoint: str):
try: try:
logger.info("请求同步接口") logger.info("请求同步接口")
format = "PNG"
job_resp = send_request("post", f"https://{endpoint}/mj_router/sync/generate/image", job_resp = send_request("post", f"https://{endpoint}/mj_router/sync/generate/image",
headers={'Authorization': 'Bearer bowong7777'}, headers={'Authorization': 'Bearer bowong7777'},
data={ data={
"prompt": prompt, "prompt": prompt,
}, },
timeout=60) files={"img_file": (
'image.' + format.lower(), tensor_to_image_bytes(image, format),
f'image/{format.lower()}')},
timeout=150)
job_resp.raise_for_status() job_resp.raise_for_status()
job_resp = job_resp.json() job_resp = job_resp.json()
if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]: if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]:
@ -236,7 +241,7 @@ class ModalMidJourneyDescribeImage:
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
return { return {
"required": { "required": {
"img_url": ("STRING", {"default": "https://vcg03.cfp.cn/creative/vcg/800/new/VCG41N948031096.jpg", "multiline": True}), "image": ("IMAGE",),
"endpoint": ("STRING", {"default": "bowongai-dev--bowong-ai-video-gemini-fastapi-webapp.modal.run"}), "endpoint": ("STRING", {"default": "bowongai-dev--bowong-ai-video-gemini-fastapi-webapp.modal.run"}),
}, },
} }
@ -247,15 +252,16 @@ class ModalMidJourneyDescribeImage:
OUTPUT_NODE = False OUTPUT_NODE = False
CATEGORY = "不忘科技-自定义节点🚩/图片/Midjourney" CATEGORY = "不忘科技-自定义节点🚩/图片/Midjourney"
def process(self, img_url: str, endpoint: str): def process(self, image: torch.Tensor, endpoint: str):
try: try:
logger.info("请求同步接口") logger.info("请求同步接口")
job_resp = send_request("post", f"https://{endpoint}/mj_router/sync/describe/image", format = "PNG"
job_resp = send_request("post", f"https://{endpoint}/mj_router/sync/describe/image/file",
headers={'Authorization': 'Bearer bowong7777'}, headers={'Authorization': 'Bearer bowong7777'},
data={ files={"img_file": (
"image_url": img_url, 'image.' + format.lower(), tensor_to_image_bytes(image, format),
}, f'image/{format.lower()}')},
timeout=60) timeout=150)
job_resp.raise_for_status() job_resp.raise_for_status()
job_resp = job_resp.json() job_resp = job_resp.json()
if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]: if "失败" in job_resp["msg"] or "fail" in job_resp["msg"] or "error" in job_resp["msg"]: