Node name/category modifications (#140)

This commit is contained in:
Jedrzej Kosinski 2025-05-05 11:17:44 -05:00 committed by GitHub
parent 0e34f00fee
commit 961651f373
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 52 additions and 52 deletions

View File

@ -200,7 +200,7 @@ class FluxProUltraImageNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/bfl"
CATEGORY = "api node/image/BFL"
def api_call(
self,
@ -326,7 +326,7 @@ class FluxProImageNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/bfl"
CATEGORY = "api node/image/BFL"
def api_call(
self,
@ -468,7 +468,7 @@ class FluxProExpandNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/bfl"
CATEGORY = "api node/image/BFL"
def api_call(
self,
@ -579,7 +579,7 @@ class FluxProFillNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/bfl"
CATEGORY = "api node/image/BFL"
def api_call(
self,
@ -713,7 +713,7 @@ class FluxProCannyNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/bfl"
CATEGORY = "api node/image/BFL"
def api_call(
self,
@ -841,7 +841,7 @@ class FluxProDepthNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/bfl"
CATEGORY = "api node/image/BFL"
def api_call(
self,

View File

@ -307,7 +307,7 @@ class IdeogramV1(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node/image/ideogram/v1"
CATEGORY = "api node/image/Ideogram/v1"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -462,7 +462,7 @@ class IdeogramV2(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node/image/ideogram/v2"
CATEGORY = "api node/image/Ideogram/v2"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -625,7 +625,7 @@ class IdeogramV3(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node/image/ideogram/v3"
CATEGORY = "api node/image/Ideogram/v3"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True

View File

@ -27,7 +27,7 @@ import logging
class MinimaxTextToVideoNode:
"""
Generates videos synchronously based on a prompt, and optional parameters using Minimax's API.
Generates videos synchronously based on a prompt, and optional parameters using MiniMax's API.
"""
@classmethod
@ -71,9 +71,9 @@ class MinimaxTextToVideoNode:
}
RETURN_TYPES = ("VIDEO",)
DESCRIPTION = "Generates videos from prompts using Minimax's API"
DESCRIPTION = "Generates videos from prompts using MiniMax's API"
FUNCTION = "generate_video"
CATEGORY = "api node/video/Minimax"
CATEGORY = "api node/video/MiniMax"
API_NODE = True
OUTPUT_NODE = True
@ -87,7 +87,7 @@ class MinimaxTextToVideoNode:
auth_token=None,
):
'''
Function used between Minimax nodes - supports T2V, I2V, and S2V, based on provided arguments.
Function used between MiniMax nodes - supports T2V, I2V, and S2V, based on provided arguments.
'''
if image is None:
validate_string(prompt_text, field_name="prompt_text")
@ -124,7 +124,7 @@ class MinimaxTextToVideoNode:
task_id = response.task_id
if not task_id:
raise Exception(f"Minimax generation failed: {response.base_resp}")
raise Exception(f"MiniMax generation failed: {response.base_resp}")
video_generate_operation = PollingOperation(
poll_endpoint=ApiEndpoint(
@ -174,7 +174,7 @@ class MinimaxTextToVideoNode:
class MinimaxImageToVideoNode(MinimaxTextToVideoNode):
"""
Generates videos synchronously based on an image and prompt, and optional parameters using Minimax's API.
Generates videos synchronously based on an image and prompt, and optional parameters using MiniMax's API.
"""
@classmethod
@ -225,16 +225,16 @@ class MinimaxImageToVideoNode(MinimaxTextToVideoNode):
}
RETURN_TYPES = ("VIDEO",)
DESCRIPTION = "Generates videos from an image and prompts using Minimax's API"
DESCRIPTION = "Generates videos from an image and prompts using MiniMax's API"
FUNCTION = "generate_video"
CATEGORY = "api node/video/Minimax"
CATEGORY = "api node/video/MiniMax"
API_NODE = True
OUTPUT_NODE = True
class MinimaxSubjectToVideoNode(MinimaxTextToVideoNode):
"""
Generates videos synchronously based on an image and prompt, and optional parameters using Minimax's API.
Generates videos synchronously based on an image and prompt, and optional parameters using MiniMax's API.
"""
@classmethod
@ -283,9 +283,9 @@ class MinimaxSubjectToVideoNode(MinimaxTextToVideoNode):
}
RETURN_TYPES = ("VIDEO",)
DESCRIPTION = "Generates videos from an image and prompts using Minimax's API"
DESCRIPTION = "Generates videos from an image and prompts using MiniMax's API"
FUNCTION = "generate_video"
CATEGORY = "api node/video/Minimax"
CATEGORY = "api node/video/MiniMax"
API_NODE = True
OUTPUT_NODE = True
@ -300,7 +300,7 @@ NODE_CLASS_MAPPINGS = {
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"MinimaxTextToVideoNode": "Minimax Text to Video",
"MinimaxImageToVideoNode": "Minimax Image to Video",
"MinimaxSubjectToVideoNode": "Minimax Subject to Video",
"MinimaxTextToVideoNode": "MiniMax Text to Video",
"MinimaxImageToVideoNode": "MiniMax Image to Video",
"MinimaxSubjectToVideoNode": "MiniMax Subject to Video",
}

View File

@ -101,7 +101,7 @@ class OpenAIDalle2(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node/image/openai"
CATEGORY = "api node/image/OpenAI"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -247,7 +247,7 @@ class OpenAIDalle3(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node/image/openai"
CATEGORY = "api node/image/OpenAI"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -380,7 +380,7 @@ class OpenAIGPTImage1(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node/image/openai"
CATEGORY = "api node/image/OpenAI"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True

View File

@ -54,20 +54,20 @@ def upload_image_to_pixverse(image: torch.Tensor, auth_token=None):
response_upload: PixverseImageUploadResponse = operation.execute()
if response_upload.Resp is None:
raise Exception(f"Pixverse image upload request failed: '{response_upload.ErrMsg}'")
raise Exception(f"PixVerse image upload request failed: '{response_upload.ErrMsg}'")
return response_upload.Resp.img_id
class PixverseTemplateNode:
"""
Select template for Pixverse Video generation.
Select template for PixVerse Video generation.
"""
RETURN_TYPES = (PixverseIO.TEMPLATE,)
RETURN_NAMES = ("pixverse_template",)
FUNCTION = "create_template"
CATEGORY = "api node/video/Pixverse"
CATEGORY = "api node/video/PixVerse"
@classmethod
def INPUT_TYPES(s):
@ -94,7 +94,7 @@ class PixverseTextToVideoNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/video/Pixverse"
CATEGORY = "api node/video/PixVerse"
@classmethod
def INPUT_TYPES(s):
@ -142,7 +142,7 @@ class PixverseTextToVideoNode(ComfyNodeABC):
"pixverse_template": (
PixverseIO.TEMPLATE,
{
"tooltip": "An optional template to influence style of generation, created by the Pixverse Template node."
"tooltip": "An optional template to influence style of generation, created by the PixVerse Template node."
}
)
},
@ -195,7 +195,7 @@ class PixverseTextToVideoNode(ComfyNodeABC):
response_api = operation.execute()
if response_api.Resp is None:
raise Exception(f"Pixverse request failed: '{response_api.ErrMsg}'")
raise Exception(f"PixVerse request failed: '{response_api.ErrMsg}'")
operation = PollingOperation(
poll_endpoint=ApiEndpoint(
@ -224,7 +224,7 @@ class PixverseImageToVideoNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/video/Pixverse"
CATEGORY = "api node/video/PixVerse"
@classmethod
def INPUT_TYPES(s):
@ -272,7 +272,7 @@ class PixverseImageToVideoNode(ComfyNodeABC):
"pixverse_template": (
PixverseIO.TEMPLATE,
{
"tooltip": "An optional template to influence style of generation, created by the Pixverse Template node."
"tooltip": "An optional template to influence style of generation, created by the PixVerse Template node."
}
)
},
@ -327,7 +327,7 @@ class PixverseImageToVideoNode(ComfyNodeABC):
response_api = operation.execute()
if response_api.Resp is None:
raise Exception(f"Pixverse request failed: '{response_api.ErrMsg}'")
raise Exception(f"PixVerse request failed: '{response_api.ErrMsg}'")
operation = PollingOperation(
poll_endpoint=ApiEndpoint(
@ -356,7 +356,7 @@ class PixverseTransitionVideoNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/video/Pixverse"
CATEGORY = "api node/video/PixVerse"
@classmethod
def INPUT_TYPES(s):
@ -407,7 +407,7 @@ class PixverseTransitionVideoNode(ComfyNodeABC):
"pixverse_template": (
PixverseIO.TEMPLATE,
{
"tooltip": "An optional template to influence style of generation, created by the Pixverse Template node."
"tooltip": "An optional template to influence style of generation, created by the PixVerse Template node."
}
)
},
@ -465,7 +465,7 @@ class PixverseTransitionVideoNode(ComfyNodeABC):
response_api = operation.execute()
if response_api.Resp is None:
raise Exception(f"Pixverse request failed: '{response_api.ErrMsg}'")
raise Exception(f"PixVerse request failed: '{response_api.ErrMsg}'")
operation = PollingOperation(
poll_endpoint=ApiEndpoint(
@ -493,8 +493,8 @@ NODE_CLASS_MAPPINGS = {
}
NODE_DISPLAY_NAME_MAPPINGS = {
"PixverseTextToVideoNode": "Pixverse Text to Video",
"PixverseImageToVideoNode": "Pixverse Image to Video",
"PixverseTransitionVideoNode": "Pixverse Transition Video",
"PixverseTemplateNode": "Pixverse Template",
"PixverseTextToVideoNode": "PixVerse Text to Video",
"PixverseImageToVideoNode": "PixVerse Image to Video",
"PixverseTransitionVideoNode": "PixVerse Transition Video",
"PixverseTemplateNode": "PixVerse Template",
}

View File

@ -55,7 +55,7 @@ class StabilityStableImageUltraNode:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/stability"
CATEGORY = "api node/image/Stability AI"
@classmethod
def INPUT_TYPES(s):
@ -182,7 +182,7 @@ class StabilityStableImageSD_3_5Node:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/stability"
CATEGORY = "api node/image/Stability AI"
@classmethod
def INPUT_TYPES(s):
@ -320,7 +320,7 @@ class StabilityUpscaleConservativeNode:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/stability"
CATEGORY = "api node/image/Stability AI"
@classmethod
def INPUT_TYPES(s):
@ -420,7 +420,7 @@ class StabilityUpscaleCreativeNode:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/stability"
CATEGORY = "api node/image/Stability AI"
@classmethod
def INPUT_TYPES(s):
@ -543,7 +543,7 @@ class StabilityUpscaleFastNode:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node/image/stability"
CATEGORY = "api node/image/Stability AI"
@classmethod
def INPUT_TYPES(s):
@ -601,9 +601,9 @@ NODE_CLASS_MAPPINGS = {
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"StabilityStableImageUltraNode": "Stability Stable Image Ultra",
"StabilityStableImageSD_3_5Node": "Stability Stable Diffusion 3.5 Image",
"StabilityUpscaleConservativeNode": "Stability Upscale Conservative",
"StabilityUpscaleCreativeNode": "Stability Upscale Creative",
"StabilityUpscaleFastNode": "Stability Upscale Fast",
"StabilityStableImageUltraNode": "Stability AI Stable Image Ultra",
"StabilityStableImageSD_3_5Node": "Stability AI Stable Diffusion 3.5 Image",
"StabilityUpscaleConservativeNode": "Stability AI Upscale Conservative",
"StabilityUpscaleCreativeNode": "Stability AI Upscale Creative",
"StabilityUpscaleFastNode": "Stability AI Upscale Fast",
}