43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""
|
|
Kaizen Package - ComfyUI Custom Nodes
|
|
Entry point for ComfyUI node registration
|
|
"""
|
|
|
|
# Import the actual node implementations from the new structure
|
|
try:
|
|
# Try relative import first (when used as part of a package)
|
|
from .modules.kaizen.image_ops import (
|
|
ImageComposite,
|
|
NODE_CLASS_MAPPINGS as _NODE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as _NODE_DISPLAY_NAME_MAPPINGS
|
|
)
|
|
except ImportError:
|
|
# Fallback to absolute import (when used directly)
|
|
from modules.kaizen.image_ops import (
|
|
ImageComposite,
|
|
NODE_CLASS_MAPPINGS as _NODE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as _NODE_DISPLAY_NAME_MAPPINGS
|
|
)
|
|
|
|
# Export the mappings that ComfyUI expects to find
|
|
NODE_CLASS_MAPPINGS = _NODE_CLASS_MAPPINGS
|
|
NODE_DISPLAY_NAME_MAPPINGS = _NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
# Optional: Add version info for debugging
|
|
WEB_DIRECTORY = "./web"
|
|
__version__ = "0.0.1"
|
|
|
|
__author__ = """Alejandro Olivares"""
|
|
__email__ = "codekaizendev@gmail.com"
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"ImageComposite",
|
|
]
|
|
|
|
# Make sure the mappings are available at the package level
|
|
__all__.extend(["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"])
|
|
|
|
|