Eugene
c88a64135e
use BaseTextRandomReplacerNode, added a Log to Workspace to view the generated prompts
50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
import os
|
|
|
|
# Import all independent node files safely
|
|
from .console_log import ConsoleLogNode, LogToWorkspace, LogToWorkspaceAdv
|
|
from .execution_delay import ExecutionDelayNode
|
|
from .file_loader import WildcardFileLoaderNode
|
|
from .randomizer import MultilineRandomChanceNode
|
|
from .text_concatenator import TextConcatenator4Node, TextConcatenator8Node
|
|
from .text_tag_replacer import TextRandomReplacerNode, TextMultiLineRandomReplacerNode, TextChainReplacerNode
|
|
|
|
# Get the exact absolute directory path of this __init__.py file
|
|
NODE_ROOT = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# Point ComfyUI safely to your web folder
|
|
WEB_DIRECTORY = os.path.join(NODE_ROOT, "web")
|
|
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MultilineRandomChanceNode": MultilineRandomChanceNode,
|
|
"ConsoleLogNode": ConsoleLogNode,
|
|
"LogToWorkspace": LogToWorkspace,
|
|
"LogToWorkspaceAdv": LogToWorkspaceAdv,
|
|
"TextRandomReplacerNode": TextRandomReplacerNode,
|
|
"TextMultiLineRandomReplacerNode": TextMultiLineRandomReplacerNode,
|
|
"TextChainReplacerNode": TextChainReplacerNode,
|
|
"WildcardFileLoaderNode": WildcardFileLoaderNode,
|
|
"ExecutionDelayNode": ExecutionDelayNode,
|
|
"TextConcatenator4Node": TextConcatenator4Node,
|
|
"TextConcatenator8Node": TextConcatenator8Node, # New mapping
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"MultilineRandomChanceNode": "🎲 Weighted Multiline Randomizer",
|
|
"TextRandomReplacerNode": "🏷️ Text Random Replacer",
|
|
"TextMultiLineRandomReplacerNode": "🏷️ Text Random Replacer (MultiLine)",
|
|
"TextChainReplacerNode": "🏷️ Text Random Replacer Chain",
|
|
"WildcardFileLoaderNode": "📂 Wildcard File Loader",
|
|
|
|
"TextConcatenator4Node": "🧩 Text Concatenator (4-Way)",
|
|
"TextConcatenator8Node": "🧱 Text Concatenator (8-Way)",
|
|
"ConsoleLogNode": "🖥️ Log to Console",
|
|
"LogToWorkspace": "🖥️ Log to Workspace",
|
|
"LogToWorkspaceAdv": "🖥️ Log to Workspace Advanced",
|
|
|
|
"ExecutionDelayNode": "⏳ Execution Delay",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|