mirror of
https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git
synced 2025-12-08 20:34:35 +08:00
213 lines
9.1 KiB
Python
213 lines
9.1 KiB
Python
from .nodes.nodes import *
|
|
from .nodes.curve_nodes import *
|
|
from .nodes.batchcrop_nodes import *
|
|
from .nodes.audioscheduler_nodes import *
|
|
NODE_CLASS_MAPPINGS = {
|
|
#constants
|
|
"INTConstant": INTConstant,
|
|
"FloatConstant": FloatConstant,
|
|
"StringConstant": StringConstant,
|
|
"StringConstantMultiline": StringConstantMultiline,
|
|
#conditioning
|
|
"ConditioningMultiCombine": ConditioningMultiCombine,
|
|
"ConditioningSetMaskAndCombine": ConditioningSetMaskAndCombine,
|
|
"ConditioningSetMaskAndCombine3": ConditioningSetMaskAndCombine3,
|
|
"ConditioningSetMaskAndCombine4": ConditioningSetMaskAndCombine4,
|
|
"ConditioningSetMaskAndCombine5": ConditioningSetMaskAndCombine5,
|
|
"CondPassThrough": CondPassThrough,
|
|
#masking
|
|
"BatchCLIPSeg": BatchCLIPSeg,
|
|
"RoundMask": RoundMask,
|
|
"ResizeMask": ResizeMask,
|
|
"OffsetMask": OffsetMask,
|
|
"MaskBatchMulti": MaskBatchMulti,
|
|
"GrowMaskWithBlur": GrowMaskWithBlur,
|
|
"ColorToMask": ColorToMask,
|
|
"CreateGradientMask": CreateGradientMask,
|
|
"CreateTextMask": CreateTextMask,
|
|
"CreateAudioMask": CreateAudioMask,
|
|
"CreateFadeMask": CreateFadeMask,
|
|
"CreateFadeMaskAdvanced": CreateFadeMaskAdvanced,
|
|
"CreateFluidMask" :CreateFluidMask,
|
|
"CreateShapeMask": CreateShapeMask,
|
|
"CreateVoronoiMask": CreateVoronoiMask,
|
|
"CreateMagicMask": CreateMagicMask,
|
|
"RemapMaskRange": RemapMaskRange,
|
|
#images
|
|
"ImageBatchMulti": ImageBatchMulti,
|
|
"ColorMatch": ColorMatch,
|
|
"CrossFadeImages": CrossFadeImages,
|
|
"GetImageRangeFromBatch": GetImageRangeFromBatch,
|
|
"SaveImageWithAlpha": SaveImageWithAlpha,
|
|
"ReverseImageBatch": ReverseImageBatch,
|
|
"ImageGridComposite2x2": ImageGridComposite2x2,
|
|
"ImageGridComposite3x3": ImageGridComposite3x3,
|
|
"ImageConcanate": ImageConcanate,
|
|
"ImageBatchTestPattern": ImageBatchTestPattern,
|
|
"ReplaceImagesInBatch": ReplaceImagesInBatch,
|
|
"ImageGrabPIL": ImageGrabPIL,
|
|
"AddLabel": AddLabel,
|
|
"ImageUpscaleWithModelBatched": ImageUpscaleWithModelBatched,
|
|
"GetImagesFromBatchIndexed": GetImagesFromBatchIndexed,
|
|
"InsertImagesToBatchIndexed": InsertImagesToBatchIndexed,
|
|
"ImageBatchRepeatInterleaving": ImageBatchRepeatInterleaving,
|
|
"ImageNormalize_Neg1_To_1": ImageNormalize_Neg1_To_1,
|
|
"RemapImageRange": RemapImageRange,
|
|
"ImagePass": ImagePass,
|
|
"ImagePadForOutpaintMasked": ImagePadForOutpaintMasked,
|
|
"ImageAndMaskPreview": ImageAndMaskPreview,
|
|
#batch cropping
|
|
"BatchCropFromMask": BatchCropFromMask,
|
|
"BatchCropFromMaskAdvanced": BatchCropFromMaskAdvanced,
|
|
"FilterZeroMasksAndCorrespondingImages": FilterZeroMasksAndCorrespondingImages,
|
|
"InsertImageBatchByIndexes": InsertImageBatchByIndexes,
|
|
"BatchUncrop": BatchUncrop,
|
|
"BatchUncropAdvanced": BatchUncropAdvanced,
|
|
"SplitBboxes": SplitBboxes,
|
|
"BboxToInt": BboxToInt,
|
|
"BboxVisualize": BboxVisualize,
|
|
#noise
|
|
"GenerateNoise": GenerateNoise,
|
|
"FlipSigmasAdjusted": FlipSigmasAdjusted,
|
|
"InjectNoiseToLatent": InjectNoiseToLatent,
|
|
"CustomSigmas": CustomSigmas,
|
|
#utility
|
|
"WidgetToString": WidgetToString,
|
|
"DummyLatentOut": DummyLatentOut,
|
|
"GetLatentsFromBatchIndexed": GetLatentsFromBatchIndexed,
|
|
"ScaleBatchPromptSchedule": ScaleBatchPromptSchedule,
|
|
"CameraPoseVisualizer": CameraPoseVisualizer,
|
|
"JoinStrings": JoinStrings,
|
|
"Sleep": Sleep,
|
|
"VRAM_Debug" : VRAM_Debug,
|
|
"SomethingToString" : SomethingToString,
|
|
"EmptyLatentImagePresets": EmptyLatentImagePresets,
|
|
#audioscheduler stuff
|
|
"NormalizedAmplitudeToMask": NormalizedAmplitudeToMask,
|
|
"OffsetMaskByNormalizedAmplitude": OffsetMaskByNormalizedAmplitude,
|
|
"ImageTransformByNormalizedAmplitude": ImageTransformByNormalizedAmplitude,
|
|
#curve nodes
|
|
"SplineEditor": SplineEditor,
|
|
"CreateShapeMaskOnPath": CreateShapeMaskOnPath,
|
|
"WeightScheduleExtend": WeightScheduleExtend,
|
|
"MaskOrImageToWeight": MaskOrImageToWeight,
|
|
"WeightScheduleConvert": WeightScheduleConvert,
|
|
"FloatToMask": FloatToMask,
|
|
#experimental
|
|
"StabilityAPI_SD3": StabilityAPI_SD3,
|
|
"SoundReactive": SoundReactive,
|
|
"StableZero123_BatchSchedule": StableZero123_BatchSchedule,
|
|
"SV3D_BatchSchedule": SV3D_BatchSchedule,
|
|
"LoadResAdapterNormalization": LoadResAdapterNormalization,
|
|
"Superprompt": Superprompt,
|
|
"GLIGENTextBoxApplyBatch": GLIGENTextBoxApplyBatch,
|
|
"Intrinsic_lora_sampling": Intrinsic_lora_sampling,
|
|
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"INTConstant": "INT Constant",
|
|
"FloatConstant": "Float Constant",
|
|
"ImageBatchMulti": "Image Batch Multi",
|
|
"MaskBatchMulti": "Mask Batch Multi",
|
|
"ConditioningMultiCombine": "Conditioning Multi Combine",
|
|
"ConditioningSetMaskAndCombine": "ConditioningSetMaskAndCombine",
|
|
"ConditioningSetMaskAndCombine3": "ConditioningSetMaskAndCombine3",
|
|
"ConditioningSetMaskAndCombine4": "ConditioningSetMaskAndCombine4",
|
|
"ConditioningSetMaskAndCombine5": "ConditioningSetMaskAndCombine5",
|
|
"GrowMaskWithBlur": "GrowMaskWithBlur",
|
|
"ColorToMask": "ColorToMask",
|
|
"CreateGradientMask": "CreateGradientMask",
|
|
"CreateTextMask" : "CreateTextMask",
|
|
"CreateFadeMask" : "CreateFadeMask (Deprecated)",
|
|
"CreateFadeMaskAdvanced" : "CreateFadeMaskAdvanced",
|
|
"CreateFluidMask" : "CreateFluidMask",
|
|
"CreateAudioMask" : "CreateAudioMask (Deprecated)",
|
|
"VRAM_Debug" : "VRAM Debug",
|
|
"CrossFadeImages": "CrossFadeImages",
|
|
"SomethingToString": "SomethingToString",
|
|
"EmptyLatentImagePresets": "EmptyLatentImagePresets",
|
|
"ColorMatch": "ColorMatch",
|
|
"GetImageRangeFromBatch": "GetImageRangeFromBatch",
|
|
"InsertImagesToBatchIndexed": "InsertImagesToBatchIndexed",
|
|
"SaveImageWithAlpha": "SaveImageWithAlpha",
|
|
"ReverseImageBatch": "ReverseImageBatch",
|
|
"ImageGridComposite2x2": "ImageGridComposite2x2",
|
|
"ImageGridComposite3x3": "ImageGridComposite3x3",
|
|
"ImageConcanate": "ImageConcatenate",
|
|
"ImageBatchTestPattern": "ImageBatchTestPattern",
|
|
"ReplaceImagesInBatch": "ReplaceImagesInBatch",
|
|
"BatchCropFromMask": "BatchCropFromMask",
|
|
"BatchCropFromMaskAdvanced": "BatchCropFromMaskAdvanced",
|
|
"FilterZeroMasksAndCorrespondingImages": "FilterZeroMasksAndCorrespondingImages",
|
|
"InsertImageBatchByIndexes": "InsertImageBatchByIndexes",
|
|
"BatchUncrop": "BatchUncrop",
|
|
"BatchUncropAdvanced": "BatchUncropAdvanced",
|
|
"BatchCLIPSeg": "BatchCLIPSeg",
|
|
"RoundMask": "RoundMask",
|
|
"ResizeMask": "ResizeMask",
|
|
"OffsetMask": "OffsetMask",
|
|
"WidgetToString": "WidgetToString",
|
|
"CreateShapeMask": "CreateShapeMask",
|
|
"CreateVoronoiMask": "CreateVoronoiMask",
|
|
"CreateMagicMask": "CreateMagicMask",
|
|
"BboxToInt": "BboxToInt",
|
|
"SplitBboxes": "SplitBboxes",
|
|
"ImageGrabPIL": "ImageGrabPIL",
|
|
"DummyLatentOut": "DummyLatentOut",
|
|
"FlipSigmasAdjusted": "FlipSigmasAdjusted",
|
|
"InjectNoiseToLatent": "InjectNoiseToLatent",
|
|
"AddLabel": "AddLabel",
|
|
"SoundReactive": "SoundReactive",
|
|
"GenerateNoise": "GenerateNoise",
|
|
"StableZero123_BatchSchedule": "StableZero123_BatchSchedule",
|
|
"SV3D_BatchSchedule": "SV3D_BatchSchedule",
|
|
"GetImagesFromBatchIndexed": "GetImagesFromBatchIndexed",
|
|
"ImageBatchRepeatInterleaving": "ImageBatchRepeatInterleaving",
|
|
"NormalizedAmplitudeToMask": "NormalizedAmplitudeToMask",
|
|
"OffsetMaskByNormalizedAmplitude": "OffsetMaskByNormalizedAmplitude",
|
|
"ImageTransformByNormalizedAmplitude": "ImageTransformByNormalizedAmplitude",
|
|
"GetLatentsFromBatchIndexed": "GetLatentsFromBatchIndexed",
|
|
"StringConstant": "StringConstant",
|
|
"GLIGENTextBoxApplyBatch": "GLIGENTextBoxApplyBatch",
|
|
"CondPassThrough": "CondPassThrough",
|
|
"ImageUpscaleWithModelBatched": "ImageUpscaleWithModelBatched",
|
|
"ScaleBatchPromptSchedule": "ScaleBatchPromptSchedule",
|
|
"ImageNormalize_Neg1_To_1": "ImageNormalize_Neg1_To_1",
|
|
"Intrinsic_lora_sampling": "Intrinsic_lora_sampling",
|
|
"RemapMaskRange": "RemapMaskRange",
|
|
"LoadResAdapterNormalization": "LoadResAdapterNormalization",
|
|
"Superprompt": "Superprompt",
|
|
"RemapImageRange": "RemapImageRange",
|
|
"CameraPoseVisualizer": "CameraPoseVisualizer",
|
|
"BboxVisualize": "BboxVisualize",
|
|
"StringConstantMultiline": "StringConstantMultiline",
|
|
"JoinStrings": "JoinStrings",
|
|
"Sleep": "🛌 Sleep 🛌",
|
|
"ImagePadForOutpaintMasked": "Pad Image For Outpaint Masked",
|
|
"ImageAndMaskPreview": "Image & Mask Preview",
|
|
"StabilityAPI_SD3": "Stability API SD3",
|
|
"MaskOrImageToWeight": "Mask Or Image To Weight",
|
|
"WeightScheduleConvert": "Weight Schedule Convert",
|
|
"FloatToMask": "Float To Mask",
|
|
"CustomSigmas": "Custom Sigmas",
|
|
"ImagePass": "ImagePass",
|
|
#curve nodes
|
|
"SplineEditor": "Spline Editor",
|
|
"CreateShapeMaskOnPath": "Create Shape Mask On Path",
|
|
"WeightScheduleExtend": "Weight Schedule Extend"
|
|
}
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
from aiohttp import web
|
|
from server import PromptServer
|
|
from pathlib import Path
|
|
|
|
if hasattr(PromptServer, "instance"):
|
|
|
|
# NOTE: we add an extra static path to avoid comfy mechanism
|
|
# that loads every script in web.
|
|
PromptServer.instance.app.add_routes(
|
|
[web.static("/kjweb_async", (Path(__file__).parent.absolute() / "kjweb_async").as_posix())]
|
|
) |