feat(ui): group nodes by category in add-node dialog

invokeai/app/invocations/batch.py

@@ -56,7 +56,7 @@ def __init__(self):
     "image_batch",
     title="Image Batch",
     tags=["primitives", "image", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -87,7 +87,7 @@ class ImageGeneratorField(BaseModel):
     "image_generator",
     title="Image Generator",
     tags=["primitives", "board", "image", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -111,7 +111,7 @@ def invoke(self, context: InvocationContext) -> ImageGeneratorOutput:
     "string_batch",
     title="String Batch",
     tags=["primitives", "string", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -142,7 +142,7 @@ class StringGeneratorField(BaseModel):
     "string_generator",
     title="String Generator",
     tags=["primitives", "string", "number", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -166,7 +166,7 @@ def invoke(self, context: InvocationContext) -> StringGeneratorOutput:
     "integer_batch",
     title="Integer Batch",
     tags=["primitives", "integer", "number", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -195,7 +195,7 @@ class IntegerGeneratorField(BaseModel):
     "integer_generator",
     title="Integer Generator",
     tags=["primitives", "int", "number", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -219,7 +219,7 @@ def invoke(self, context: InvocationContext) -> IntegerGeneratorOutput:
     "float_batch",
     title="Float Batch",
     tags=["primitives", "float", "number", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )
@@ -250,7 +250,7 @@ class FloatGeneratorField(BaseModel):
     "float_generator",
     title="Float Generator",
     tags=["primitives", "float", "number", "batch", "special"],
-    category="primitives",
+    category="batch",
     version="1.0.0",
     classification=Classification.Special,
 )

invokeai/app/invocations/canny.py

@@ -11,7 +11,7 @@
     "canny_edge_detection",
     title="Canny Edge Detection",
     tags=["controlnet", "canny"],
-    category="controlnet",
+    category="controlnet_preprocessors",
     version="1.0.0",
 )
 class CannyEdgeDetectionInvocation(BaseInvocation, WithMetadata, WithBoard):

invokeai/app/invocations/cogview4_denoise.py

@@ -33,7 +33,7 @@
     "cogview4_denoise",
     title="Denoise - CogView4",
     tags=["image", "cogview4"],
-    category="image",
+    category="latents",
     version="1.0.0",
     classification=Classification.Prototype,
 )

invokeai/app/invocations/cogview4_image_to_latents.py

@@ -27,7 +27,7 @@
     "cogview4_i2l",
     title="Image to Latents - CogView4",
     tags=["image", "latents", "vae", "i2l", "cogview4"],
-    category="image",
+    category="latents",
     version="1.0.0",
     classification=Classification.Prototype,
 )

invokeai/app/invocations/cogview4_text_encoder.py

@@ -20,7 +20,7 @@
     "cogview4_text_encoder",
     title="Prompt - CogView4",
     tags=["prompt", "conditioning", "cogview4"],
-    category="conditioning",
+    category="prompt",
     version="1.0.0",
     classification=Classification.Prototype,
 )

invokeai/app/invocations/collections.py

@@ -11,9 +11,7 @@
 from invokeai.app.util.misc import SEED_MAX
 
 
-@invocation(
-    "range", title="Integer Range", tags=["collection", "integer", "range"], category="collections", version="1.0.0"
-)
+@invocation("range", title="Integer Range", tags=["collection", "integer", "range"], category="batch", version="1.0.0")
 class RangeInvocation(BaseInvocation):
     """Creates a range of numbers from start to stop with step"""
 
@@ -35,7 +33,7 @@ def invoke(self, context: InvocationContext) -> IntegerCollectionOutput:
     "range_of_size",
     title="Integer Range of Size",
     tags=["collection", "integer", "size", "range"],
-    category="collections",
+    category="batch",
     version="1.0.0",
 )
 class RangeOfSizeInvocation(BaseInvocation):
@@ -55,7 +53,7 @@ def invoke(self, context: InvocationContext) -> IntegerCollectionOutput:
     "random_range",
     title="Random Range",
     tags=["range", "integer", "random", "collection"],
-    category="collections",
+    category="batch",
     version="1.0.1",
     use_cache=False,
 )

invokeai/app/invocations/color_map.py

@@ -11,7 +11,7 @@
     "color_map",
     title="Color Map",
     tags=["controlnet"],
-    category="controlnet",
+    category="controlnet_preprocessors",
     version="1.0.0",
 )
 class ColorMapInvocation(BaseInvocation, WithMetadata, WithBoard):

invokeai/app/invocations/compel.py

@@ -43,7 +43,7 @@
     "compel",
     title="Prompt - SD1.5",
     tags=["prompt", "compel"],
-    category="conditioning",
+    category="prompt",
     version="1.2.1",
 )
 class CompelInvocation(BaseInvocation):
@@ -248,7 +248,7 @@ def _lora_loader() -> Iterator[Tuple[ModelPatchRaw, float]]:
     "sdxl_compel_prompt",
     title="Prompt - SDXL",
     tags=["sdxl", "compel", "prompt"],
-    category="conditioning",
+    category="prompt",
     version="1.2.1",
 )
 class SDXLCompelPromptInvocation(BaseInvocation, SDXLPromptInvocationBase):
@@ -342,7 +342,7 @@ def invoke(self, context: InvocationContext) -> ConditioningOutput:
     "sdxl_refiner_compel_prompt",
     title="Prompt - SDXL Refiner",
     tags=["sdxl", "compel", "prompt"],
-    category="conditioning",
+    category="prompt",
     version="1.1.2",
 )
 class SDXLRefinerCompelPromptInvocation(BaseInvocation, SDXLPromptInvocationBase):
@@ -391,7 +391,7 @@ class CLIPSkipInvocationOutput(BaseInvocationOutput):
     "clip_skip",
     title="Apply CLIP Skip - SD1.5, SDXL",
     tags=["clipskip", "clip", "skip"],
-    category="conditioning",
+    category="prompt",
     version="1.1.1",
 )
 class CLIPSkipInvocation(BaseInvocation):

invokeai/app/invocations/content_shuffle.py

@@ -9,7 +9,7 @@
     "content_shuffle",
     title="Content Shuffle",
     tags=["controlnet", "normal"],
-    category="controlnet",
+    category="controlnet_preprocessors",
     version="1.0.0",
 )
 class ContentShuffleInvocation(BaseInvocation, WithMetadata, WithBoard):

invokeai/app/invocations/controlnet.py

@@ -64,7 +64,7 @@ class ControlOutput(BaseInvocationOutput):
 
 
 @invocation(
-    "controlnet", title="ControlNet - SD1.5, SD2, SDXL", tags=["controlnet"], category="controlnet", version="1.1.3"
+    "controlnet", title="ControlNet - SD1.5, SD2, SDXL", tags=["controlnet"], category="conditioning", version="1.1.3"
 )
 class ControlNetInvocation(BaseInvocation):
     """Collects ControlNet info to pass to other nodes"""
@@ -116,7 +116,7 @@ def invoke(self, context: InvocationContext) -> ControlOutput:
     "heuristic_resize",
     title="Heuristic Resize",
     tags=["image, controlnet"],
-    category="image",
+    category="controlnet_preprocessors",
     version="1.1.1",
     classification=Classification.Prototype,
 )

invokeai/app/invocations/create_denoise_mask.py

@@ -18,7 +18,7 @@
     "create_denoise_mask",
     title="Create Denoise Mask",
     tags=["mask", "denoise"],
-    category="latents",
+    category="mask",
     version="1.0.2",
 )
 class CreateDenoiseMaskInvocation(BaseInvocation):

invokeai/app/invocations/create_gradient_mask.py

@@ -41,7 +41,7 @@ class GradientMaskOutput(BaseInvocationOutput):
     "create_gradient_mask",
     title="Create Gradient Mask",
     tags=["mask", "denoise"],
-    category="latents",
+    category="mask",
     version="1.3.0",
 )
 class CreateGradientMaskInvocation(BaseInvocation):

invokeai/app/invocations/depth_anything.py

@@ -20,7 +20,7 @@
     "depth_anything_depth_estimation",
     title="Depth Anything Depth Estimation",
     tags=["controlnet", "depth", "depth anything"],
-    category="controlnet",
+    category="controlnet_preprocessors",
     version="1.0.0",
 )
 class DepthAnythingDepthEstimationInvocation(BaseInvocation, WithMetadata, WithBoard):

invokeai/app/invocations/dw_openpose.py

@@ -11,7 +11,7 @@
     "dw_openpose_detection",
     title="DW Openpose Detection",
     tags=["controlnet", "dwpose", "openpose"],
-    category="controlnet",
+    category="controlnet_preprocessors",
     version="1.1.1",
 )
 class DWOpenposeDetectionInvocation(BaseInvocation, WithMetadata, WithBoard):

invokeai/app/invocations/facetools.py

@@ -435,7 +435,9 @@ def get_faces_list(
     return all_faces
 
 
-@invocation("face_off", title="FaceOff", tags=["image", "faceoff", "face", "mask"], category="image", version="1.2.2")
+@invocation(
+    "face_off", title="FaceOff", tags=["image", "faceoff", "face", "mask"], category="segmentation", version="1.2.2"
+)
 class FaceOffInvocation(BaseInvocation, WithMetadata):
     """Bound, extract, and mask a face from an image using MediaPipe detection"""
 
@@ -514,7 +516,9 @@ def invoke(self, context: InvocationContext) -> FaceOffOutput:
         return output
 
 
-@invocation("face_mask_detection", title="FaceMask", tags=["image", "face", "mask"], category="image", version="1.2.2")
+@invocation(
+    "face_mask_detection", title="FaceMask", tags=["image", "face", "mask"], category="segmentation", version="1.2.2"
+)
 class FaceMaskInvocation(BaseInvocation, WithMetadata):
     """Face mask creation using mediapipe face detection"""
 
@@ -617,7 +621,11 @@ def invoke(self, context: InvocationContext) -> FaceMaskOutput:
 
 
 @invocation(
-    "face_identifier", title="FaceIdentifier", tags=["image", "face", "identifier"], category="image", version="1.2.2"
+    "face_identifier",
+    title="FaceIdentifier",
+    tags=["image", "face", "identifier"],
+    category="segmentation",
+    version="1.2.2",
 )
 class FaceIdentifierInvocation(BaseInvocation, WithMetadata, WithBoard):
     """Outputs an image with detected face IDs printed on each face. For use with other FaceTools."""

invokeai/app/invocations/flux2_denoise.py

@@ -53,7 +53,7 @@
     "flux2_denoise",
     title="FLUX2 Denoise",
     tags=["image", "flux", "flux2", "klein", "denoise"],
-    category="image",
+    category="latents",
     version="1.4.0",
     classification=Classification.Prototype,
 )

invokeai/app/invocations/flux2_klein_text_encoder.py

@@ -45,7 +45,7 @@
     "flux2_klein_text_encoder",
     title="Prompt - Flux2 Klein",
     tags=["prompt", "conditioning", "flux", "klein", "qwen3"],
-    category="conditioning",
+    category="prompt",
     version="1.1.1",
     classification=Classification.Prototype,
 )

invokeai/app/invocations/flux_controlnet.py

@@ -50,7 +50,7 @@ class FluxControlNetOutput(BaseInvocationOutput):
     "flux_controlnet",
     title="FLUX ControlNet",
     tags=["controlnet", "flux"],
-    category="controlnet",
+    category="conditioning",
     version="1.0.0",
 )
 class FluxControlNetInvocation(BaseInvocation):

invokeai/app/invocations/flux_denoise.py

@@ -70,7 +70,7 @@
     "flux_denoise",
     title="FLUX Denoise",
     tags=["image", "flux"],
-    category="image",
+    category="latents",
     version="4.5.1",
 )
 class FluxDenoiseInvocation(BaseInvocation):

invokeai/app/invocations/flux_fill.py

@@ -29,7 +29,7 @@ class FluxFillOutput(BaseInvocationOutput):
     "flux_fill",
     title="FLUX Fill Conditioning",
     tags=["inpaint"],
-    category="inpaint",
+    category="conditioning",
     version="1.0.0",
     classification=Classification.Beta,
 )

invokeai/app/invocations/flux_ip_adapter.py

@@ -24,7 +24,7 @@
     "flux_ip_adapter",
     title="FLUX IP-Adapter",
     tags=["ip_adapter", "control"],
-    category="ip_adapter",
+    category="conditioning",
     version="1.0.0",
 )
 class FluxIPAdapterInvocation(BaseInvocation):

invokeai/app/invocations/flux_redux.py

@@ -47,7 +47,7 @@ class FluxReduxOutput(BaseInvocationOutput):
     "flux_redux",
     title="FLUX Redux",
     tags=["ip_adapter", "control"],
-    category="ip_adapter",
+    category="conditioning",
     version="2.1.0",
     classification=Classification.Beta,
 )

invokeai/app/invocations/flux_text_encoder.py

@@ -28,7 +28,7 @@
     "flux_text_encoder",
     title="Prompt - FLUX",
     tags=["prompt", "conditioning", "flux"],
-    category="conditioning",
+    category="prompt",
     version="1.1.2",
 )
 class FluxTextEncoderInvocation(BaseInvocation):

invokeai/app/invocations/grounding_dino.py

@@ -24,7 +24,7 @@
     "grounding_dino",
     title="Grounding DINO (Text Prompt Object Detection)",
     tags=["prompt", "object detection"],
-    category="image",
+    category="segmentation",
     version="1.0.0",
 )
 class GroundingDinoInvocation(BaseInvocation):

invokeai/app/invocations/hed.py

@@ -11,7 +11,7 @@
     "hed_edge_detection",
     title="HED Edge Detection",
     tags=["controlnet", "hed", "softedge"],
-    category="controlnet",
+    category="controlnet_preprocessors",
     version="1.0.0",
 )
 class HEDEdgeDetectionInvocation(BaseInvocation, WithMetadata, WithBoard):

invokeai/app/invocations/ideal_size.py

@@ -21,6 +21,7 @@ class IdealSizeOutput(BaseInvocationOutput):
     "ideal_size",
     title="Ideal Size - SD1.5, SDXL",
     tags=["latents", "math", "ideal_size"],
+    category="latents",
     version="1.0.6",
 )
 class IdealSizeInvocation(BaseInvocation):