feat: add basic main category mapping

Author: Yourz

server.py

@@ -51,6 +51,262 @@
     import comfyui_manager
 
 
+MAIN_CATEGORY_OVERRIDES = {name: "Basics" for name in [
+    "CLIPTextEncodeSDXL+",
+    "ConditioningCombineMultiple+",
+    "SD3NegativeConditioning+",
+    "ImageEnhanceDifference+",
+    "ImageBatchMultiple+",
+    "ImageExpandBatch+",
+    "ImageFromBatch+",
+    "ImageListToBatch+",
+    "ImageComposite+",
+    "ImageCompositeFromMaskBatch+",
+    "ImageCrop+",
+    "ImageFlip+",
+    "ImageRandomTransform+",
+    "ImageRemoveBackground+",
+    "ImageResize+",
+    "ImageSeamCarving+",
+    "ImageTile+",
+    "ImageUntile+",
+    "RemBGSession+",
+    "TransparentBGSession+",
+    "ImageApplyLUT+",
+    "ImageCASharpening+",
+    "ImageColorMatch+",
+    "ImageColorMatchAdobe+",
+    "ImageDesaturate+",
+    "ImageHistogramMatch+",
+    "ImagePosterize+",
+    "PixelOEPixelize+",
+    "GetImageSize+",
+    "ImagePreviewFromLatent+",
+    "ImageRemoveAlpha+",
+    "ImageToDevice+",
+    "NoiseFromImage+",
+    "Mask Bounding Box Aspect Ratio",
+    "MaskBlur+",
+    "MaskBoundingBox+",
+    "MaskFix+",
+    "MaskFlip+",
+    "MaskFromColor+",
+    "MaskFromList+",
+    "MaskFromRGBCMYBW+",
+    "MaskFromSegmentation+",
+    "MaskPreview+",
+    "MaskSmooth+",
+    "TransitionMask+",
+    "MaskBatch+",
+    "MaskExpandBatch+",
+    "MaskFromBatch+",
+    "FluxSamplerParams+",
+    "InjectLatentNoise+",
+    "KSamplerVariationsStochastic+",
+    "KSamplerVariationsWithNoise+",
+    "PlotParameters+",
+    "ApplyCLIPSeg+",
+    "LoadCLIPSegModels+",
+    "DrawText+",
+    "BatchCount+",
+    "ConsoleDebug+",
+    "DebugTensorShape+",
+    "ModelCompile+",
+    "RemoveLatentMask+",
+    "SDXLEmptyLatentSizePicker+",
+    "SimpleMath+",
+    "Intrinsic_lora_sampling",
+    "AudioConcatenate",
+    "ImageTransformByNormalizedAmplitude",
+    "NormalizedAmplitudeToFloatList",
+    "NormalizedAmplitudeToMask",
+    "OffsetMaskByNormalizedAmplitude",
+    "SoundReactive",
+    "BOOLConstant",
+    "FloatConstant",
+    "INTConstant",
+    "StringConstant",
+    "StringConstantMultiline",
+    "CustomControlNetWeightsFluxFromList",
+    "CreateAudioMask",
+    "CreateFadeMask",
+    "ApplyRifleXRoPE_HunuyanVideo",
+    "ApplyRifleXRoPE_WanVideo",
+    "CFGZeroStarAndInit",
+    "CheckpointLoaderKJ",
+    "CheckpointPerturbWeights",
+    "DiffusionModelLoaderKJ",
+    "DiffusionModelSelector",
+    "DiTBlockLoraLoader",
+    "FastPreview",
+    "FluxBlockLoraSelect",
+    "GGUFLoaderKJ",
+    "GLIGENTextBoxApplyBatchCoords",
+    "HunyuanVideoBlockLoraSelect",
+    "InterpolateCoords",
+    "LatentInpaintTTM",
+    "LeapfusionHunyuanI2VPatcher",
+    "LoadResAdapterNormalization",
+    "ModelPatchTorchSettings",
+    "NABLA_AttentionKJ",
+    "PatchModelPatcherOrder",
+    "PathchSageAttentionKJ",
+    "PlotCoordinates",
+    "PointsEditor",
+    "ScheduledCFGGuidance",
+    "StableZero123_BatchSchedule",
+    "StyleModelApplyAdvanced",
+    "SV3D_BatchSchedule",
+    "Wan21BlockLoraSelect",
+    "WanVideoEnhanceAVideoKJ",
+    "WanVideoNAG",
+    "WebcamCaptureCV2",
+    "ColorMatch",
+    "CreateGradientFromCoords",
+    "CreateShapeImageOnPath",
+    "CrossFadeImages",
+    "CrossFadeImagesMulti",
+    "CutAndDragOnPath",
+    "GetImageRangeFromBatch",
+    "GetImagesFromBatchIndexed",
+    "GetImageSizeAndCount",
+    "GetLatentSizeAndCount",
+    "GradientToFloat",
+    "ImageAddMulti",
+    "ImageBatchExtendWithOverlap",
+    "ImageBatchFilter",
+    "ImageBatchJoinWithTransition",
+    "ImageBatchMulti",
+    "ImageBatchRepeatInterleaving",
+    "ImageConcanate",
+    "ImageConcatFromBatch",
+    "ImageConcatMulti",
+    "ImageCropByMask",
+    "ImageCropByMaskAndResize",
+    "ImageCropByMaskBatch",
+    "ImageGrabPIL",
+    "ImageGridComposite2x2",
+    "ImageGridComposite3x3",
+    "ImageGridtoBatch",
+    "ImageNoiseAugmentation",
+    "ImageNormalize_Neg1_To_1",
+    "ImagePadKJ",
+    "ImagePass",
+    "ImageResizeKJ",
+    "ImageResizeKJv2",
+    "ImageTensorList",
+    "ImageUncropByMask",
+    "ImageUpscaleWithModelBatched",
+    "InsertImageBatchByIndexes",
+    "InsertImagesToBatchIndexed",
+    "LoadAndResizeImage",
+    "MergeImageChannels",
+    "PadImageBatchInterleaved",
+    "PreviewAnimation",
+    "RemapImageRange",
+    "ReplaceImagesInBatch",
+    "ReverseImageBatch",
+    "SaveImageWithAlpha",
+    "Screencap_mss",
+    "ShuffleImageBatch",
+    "SplitImageChannels",
+    "TransitionImagesInBatch",
+    "TransitionImagesMulti",
+    "AppendInstanceDiffusionTracking",
+    "CreateInstanceDiffusionTracking",
+    "DrawInstanceDiffusionTracking",
+    "EmptyLatentImageCustomPresets",
+    "EmptyLatentImagePresets",
+    "GetLatentRangeFromBatch",
+    "GetLatentsFromBatchIndexed",
+    "InsertLatentToIndexed",
+    "LoraExtractKJ",
+    "LoraReduceRankKJ",
+    "BatchCLIPSeg",
+    "BatchCropFromMask",
+    "BatchCropFromMaskAdvanced",
+    "BatchUncrop",
+    "BatchUncropAdvanced",
+    "BboxToInt",
+    "BboxVisualize",
+    "BlockifyMask",
+    "ColorToMask",
+    "ConsolidateMasksKJ",
+    "DownloadAndLoadCLIPSeg",
+    "DrawGaussianNoiseOnImage",
+    "DrawMaskOnImage",
+    "FilterZeroMasksAndCorrespondingImages",
+    "GetMaskSizeAndCount",
+    "GrowMaskWithBlur",
+    "ImageAndMaskPreview",
+    "MaskBatchMulti",
+    "OffsetMask",
+    "RemapMaskRange",
+    "ResizeMask",
+    "RoundMask",
+    "SeparateMasks",
+    "SplitBboxes",
+    "CameraPoseVisualizer",
+    "CondPassThrough",
+    "DummyOut",
+    "LazySwitchKJ",
+    "ModelPassThrough",
+    "ScaleBatchPromptSchedule",
+    "SimpleCalculatorKJ",
+    "Sleep",
+    "StringToFloatList",
+    "TimerNodeKJ",
+    "VRAM_Debug",
+    "CustomSigmas",
+    "FlipSigmasAdjusted",
+    "FloatToSigmas",
+    "GenerateNoise",
+    "InjectNoiseToLatent",
+    "SigmasToFloat",
+    "WanVideoTeaCacheKJ",
+    "AddLabel",
+    "AppendStringsToList",
+    "CreateTextMask",
+    "ImageBatchTestPattern",
+    "JoinStringMulti",
+    "JoinStrings",
+    "SomethingToString",
+    "Superprompt",
+    "WidgetToString",
+    "TorchCompileControlNet",
+    "TorchCompileCosmosModel",
+    "TorchCompileLTXModel",
+    "TorchCompileModelAdvanced",
+    "TorchCompileModelFluxAdvanced",
+    "TorchCompileModelFluxAdvancedV2",
+    "TorchCompileModelHyVideo",
+    "TorchCompileModelQwenImage",
+    "TorchCompileModelWanVideo",
+    "TorchCompileModelWanVideoV2",
+    "TorchCompileVAE",
+    "VAELoaderKJ",
+    "HunyuanVideoEncodeKeyframesToCond",
+    "MaskOrImageToWeight",
+    "SplineEditor",
+    "WeightScheduleConvert",
+    "WeightScheduleExtend",
+    "ConditioningMultiCombine",
+    "ConditioningSetMaskAndCombine",
+    "ConditioningSetMaskAndCombine3",
+    "ConditioningSetMaskAndCombine4",
+    "ConditioningSetMaskAndCombine5",
+    "CreateFadeMaskAdvanced",
+    "CreateFluidMask",
+    "CreateGradientMask",
+    "CreateMagicMask",
+    "CreateShapeMask",
+    "CreateShapeMaskOnPath",
+    "CreateTextOnPath",
+    "CreateVoronoiMask",
+    "FloatToMask",
+]}
+
+
 def _remove_sensitive_from_queue(queue: list) -> list:
     """Remove sensitive data (index 5) from queue item tuples."""
     return [item[:5] for item in queue]
@@ -652,7 +908,10 @@ async def get_prompt(request):
         def node_info(node_class):
             obj_class = nodes.NODE_CLASS_MAPPINGS[node_class]
             if issubclass(obj_class, _ComfyNodeInternal):
-                return obj_class.GET_NODE_INFO_V1()
+                info = obj_class.GET_NODE_INFO_V1()
+                if node_class in MAIN_CATEGORY_OVERRIDES:
+                    info['main_category'] = MAIN_CATEGORY_OVERRIDES[node_class]
+                return info
             info = {}
             info['input'] = obj_class.INPUT_TYPES()
             info['input_order'] = {key: list(value.keys()) for (key, value) in obj_class.INPUT_TYPES().items()}
@@ -694,6 +953,9 @@ def node_info(node_class):
             if hasattr(obj_class, 'MAIN_CATEGORY'):
                 info['main_category'] = obj_class.MAIN_CATEGORY
 
+            if node_class in MAIN_CATEGORY_OVERRIDES:
+                info['main_category'] = MAIN_CATEGORY_OVERRIDES[node_class]
+
             return info
 
         @routes.get("/object_info")