70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
# 从 nodes 模块导入节点类
|
|
from .nodes.multi_save_image import MultiSaveImage
|
|
from .nodes.simple_llm_node import SimpleLLMNode
|
|
from .nodes.advanced_image_crop import AdvancedImageCrop
|
|
from .nodes.image_loader import ImageLoader
|
|
from .nodes.mask_loader import MaskLoader
|
|
from .nodes.text_area_input import TextAreaInput
|
|
from .nodes.smart_empty_latent import SmartEmptyLatent
|
|
from .nodes.size_preset_safe import SizePresetSafe
|
|
from .nodes.smart_exact_resize import SmartExactResize
|
|
from .nodes.outpaint_size_preset_safe import OutpaintSizePresetSafe
|
|
from .nodes.rmbg_birefnet import MeuxRMBG
|
|
from .nodes.realesrgan_upscale import MeuxRealESRGANUpscale
|
|
from .nodes.mask_fill_holes import MeuxMaskFillHoles
|
|
from .nodes.mask_blur_plus import MeuxMaskBlurPlus
|
|
from .nodes.seed_node import MeuxSeed
|
|
from .nodes.artistic_text_preview import MeuxArtisticTextPreview
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MeuxImageLoader": ImageLoader,
|
|
"MeuxMultiSaveImage": MultiSaveImage,
|
|
"MeuxSimpleLLMNode": SimpleLLMNode,
|
|
"MeuxMaskLoader": MaskLoader,
|
|
"MeuxAdvancedImageCrop": AdvancedImageCrop,
|
|
"MeuxTextAreaInput": TextAreaInput,
|
|
"MeuxSmartEmptyLatent": SmartEmptyLatent,
|
|
"MeuxSizePresetSafe": SizePresetSafe,
|
|
"MeuxSmartExactResize": SmartExactResize,
|
|
"MeuxOutpaintSizePresetSafe": OutpaintSizePresetSafe,
|
|
"MeuxRMBG": MeuxRMBG,
|
|
"MeuxRealESRGANUpscale": MeuxRealESRGANUpscale,
|
|
"MeuxMaskFillHoles": MeuxMaskFillHoles,
|
|
"MeuxMaskBlurPlus": MeuxMaskBlurPlus,
|
|
"MeuxSeed": MeuxSeed,
|
|
"MeuxArtisticTextPreview": MeuxArtisticTextPreview,
|
|
"Mask Fill Holes": MeuxMaskFillHoles,
|
|
"MaskBlur+": MeuxMaskBlurPlus,
|
|
"Seed": MeuxSeed,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"MeuxImageLoader": "Meux Image Loader",
|
|
"MeuxMultiSaveImage": "Meux Multi Save Image",
|
|
"MeuxSimpleLLMNode": "Meux LLM API Call",
|
|
"MeuxMaskLoader": "Meux Mask Loader",
|
|
"MeuxAdvancedImageCrop": "Meux Advanced Image Crop",
|
|
"MeuxTextAreaInput": "Meux Text Area",
|
|
"MeuxSmartEmptyLatent": "Meux Smart Empty Latent",
|
|
"MeuxSizePresetSafe": "Meux Size Preset Node",
|
|
"MeuxSmartExactResize": "Meux Smart Exact Resize",
|
|
"MeuxOutpaintSizePresetSafe": "Meux Outpaint Size Preset Node",
|
|
"MeuxRMBG": "Meux RMBG (BiRefNet)",
|
|
"MeuxRealESRGANUpscale": "Meux ESRGAN Upscale",
|
|
"MeuxMaskFillHoles": "Meux Mask Fill Holes",
|
|
"MeuxMaskBlurPlus": "Meux Mask Blur+",
|
|
"MeuxSeed": "Meux Seed",
|
|
"MeuxArtisticTextPreview": "Meux Artistic Text Preview",
|
|
"Mask Fill Holes": "Mask Fill Holes",
|
|
"MaskBlur+": "MaskBlur+",
|
|
"Seed": "Seed",
|
|
|
|
}
|
|
|
|
# 可选：添加版本和作者信息
|
|
__version__ = "1.7.1"
|
|
__author__ = "fangchangjun"
|
|
|
|
# 调试信息 - 可以帮助确认导入是否成功
|
|
print(f"[INFO] 成功加载自定义节点: {list(NODE_CLASS_MAPPINGS.keys())}")
|