21 lines
685 B
Python
21 lines
685 B
Python
from .api_request import APIRequestNode
|
|
from .image_post_node import PostImageToAPI
|
|
from .text_prompt_combiner_node import TextPromptCombinerNode
|
|
|
|
# Define the mappings for your custom nodes
|
|
NODE_CLASS_MAPPINGS = {
|
|
"APIRequestNode": APIRequestNode,
|
|
"PostImageToAPI": PostImageToAPI,
|
|
"TextPromptCombinerNode": TextPromptCombinerNode,
|
|
}
|
|
|
|
# Define human-readable names for your nodes (optional)
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"APIRequestNode": "API Request Node",
|
|
"PostImageToAPI": "Image Post Node",
|
|
"TextPromptCombinerNode": "Text Prompt Combiner Node",
|
|
}
|
|
|
|
# Export the mappings for use by ComfyUI
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|