CoCoandClaude Opus 4.6
5315094b83
Use relative path for WEB_DIRECTORY. Remove commented-out registration code from modules. Replace bare except with specific exception types. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
|
# Import all available modules
|
|
from .modules.image_loader import ImageLoader
|
|
from .modules.load_exr import LoadExr
|
|
from .modules.load_exr_sequence import LoadExrSequence
|
|
from .modules.saver import SaverNode
|
|
from .modules.load_exr_layer_by_name import LoadExrLayerByName, CryptomatteLayer
|
|
from .modules.colorspace import ColorspaceNode
|
|
from .modules.znormalize import ZNormalizeNode
|
|
|
|
# Initialize node mappings
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
# Add all available node classes
|
|
NODE_CLASS_MAPPINGS.update({
|
|
"ImageLoader": ImageLoader,
|
|
"LoadExr": LoadExr,
|
|
"LoadExrSequence": LoadExrSequence,
|
|
"SaverNode": SaverNode,
|
|
"LoadExrLayerByName": LoadExrLayerByName,
|
|
"CryptomatteLayer": CryptomatteLayer,
|
|
"ColorspaceNode": ColorspaceNode,
|
|
"ZNormalizeNode": ZNormalizeNode
|
|
})
|
|
|
|
# Add display names for better UI presentation
|
|
NODE_DISPLAY_NAME_MAPPINGS.update({
|
|
"ImageLoader": "CoCo Loader",
|
|
"LoadExr": "CoCo Load EXR",
|
|
"LoadExrSequence": "CoCo Load EXR Sequence",
|
|
"SaverNode": "CoCo Saver",
|
|
"LoadExrLayerByName": "CoCo Load EXR Layer by Name",
|
|
"CryptomatteLayer": "CoCo Cryptomatte Layer",
|
|
"ColorspaceNode": "CoCo Colorspace",
|
|
"ZNormalizeNode": "CoCo Z Normalize"
|
|
})
|
|
|
|
# Expose what ComfyUI needs
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
]
|