23 lines
742 B
Python
23 lines
742 B
Python
from .nodes.image_watermark_node import ImageWatermarkNode
|
|
from .nodes.text_extractor_node import TextExtractorNode
|
|
from .nodes.flux_kontext_diff_merge import FluxKontextDiffMerge
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
# Unique keys matching the class names in your .py files
|
|
"ImageWatermarkNode": ImageWatermarkNode,
|
|
"TextExtractorNode": TextExtractorNode,
|
|
"FluxKontextDiffMerge": FluxKontextDiffMerge,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
# Public-facing names in the ComfyUI “Add Node” menu
|
|
"ImageWatermarkNode": "Add Image Watermark",
|
|
"TextExtractorNode": "Text Extractor Node",
|
|
"FluxKontextDiffMerge": "Flux Kontext Diff Merge (Optimized)",
|
|
}
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS"
|
|
]
|