Amatsukast
4cebd98a82
Add three new nodes for flexible prompt generation optimized for FLUX.2: - Prompt Variable: Define variables with sequential/random modes * Tag-based template replacement with [TAG_NAME] syntax * prefix/suffix support for text formatting * Empty value support ([NONE] or empty line) * Randomize toggle for mode switching - Variable Combiner: Merge multiple variable lists * Support for 6 variable lists per node * Chainable for unlimited expansion * Unified var_list naming - Prompt Compiler: Generate prompts from variables * Automatic template tag replacement * Simple join mode for empty templates * Sequential + Random combination support * Debug output with all combinations * Auto-append unused variables * Warning system for unmatched tags Architecture improvements: - Variables output as VARIABLE_LIST for direct connection - Optimized for natural language prompt generation
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
"""ComfyUI-SBTools - Custom Node Collection"""
|
|
|
|
__version__ = "1.2.0"
|
|
|
|
from .nodes.birefnet_node import NODE_CLASS_MAPPINGS as BIREFNET_MAPPINGS
|
|
from .nodes.birefnet_node import NODE_DISPLAY_NAME_MAPPINGS as BIREFNET_DISPLAY
|
|
from .nodes.alpha_to_chroma_key import NODE_CLASS_MAPPINGS as CHROMAKEY_MAPPINGS
|
|
from .nodes.alpha_to_chroma_key import NODE_DISPLAY_NAME_MAPPINGS as CHROMAKEY_DISPLAY
|
|
from .nodes.prompt_variable import NODE_CLASS_MAPPINGS as PROMPTVAR_MAPPINGS
|
|
from .nodes.prompt_variable import NODE_DISPLAY_NAME_MAPPINGS as PROMPTVAR_DISPLAY
|
|
from .nodes.variable_combiner import NODE_CLASS_MAPPINGS as VARCOMB_MAPPINGS
|
|
from .nodes.variable_combiner import NODE_DISPLAY_NAME_MAPPINGS as VARCOMB_DISPLAY
|
|
from .nodes.prompt_compiler import NODE_CLASS_MAPPINGS as PROMPTCOMP_MAPPINGS
|
|
from .nodes.prompt_compiler import NODE_DISPLAY_NAME_MAPPINGS as PROMPTCOMP_DISPLAY
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
**BIREFNET_MAPPINGS,
|
|
**CHROMAKEY_MAPPINGS,
|
|
**PROMPTVAR_MAPPINGS,
|
|
**VARCOMB_MAPPINGS,
|
|
**PROMPTCOMP_MAPPINGS,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**BIREFNET_DISPLAY,
|
|
**CHROMAKEY_DISPLAY,
|
|
**PROMPTVAR_DISPLAY,
|
|
**VARCOMB_DISPLAY,
|
|
**PROMPTCOMP_DISPLAY,
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|