Antonio Cordero Balcazar
39d6008f6f
* Refactor PromptPostProcessor class to use the same default values for options * Add new nodes for the different option groups
40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
"""
|
|
@author: ACB
|
|
@title: Prompt Post Processor
|
|
@nickname: ACB PPP
|
|
@description: Node for processing prompts. Includes the following options: send to negative prompt, set variables, if/elif/else command for conditional content, wildcards and choices.
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from .ppp_comfyui import (
|
|
PromptPostProcessorComfyUINode,
|
|
PromptPostProcessorWildcardOptionsComfyUINode,
|
|
PromptPostProcessorENMappingOptionsComfyUINode,
|
|
PromptPostProcessorSTNOptionsComfyUINode,
|
|
PromptPostProcessorCleanupOptionsComfyUINode,
|
|
PromptPostProcessorSelectVariableComfyUINode,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ACBPromptPostProcessor": PromptPostProcessorComfyUINode,
|
|
"ACBPPPWildcardOptions": PromptPostProcessorWildcardOptionsComfyUINode,
|
|
"ACBPPPENMappingOptions": PromptPostProcessorENMappingOptionsComfyUINode,
|
|
"ACBPPPSendToNegativeOptions": PromptPostProcessorSTNOptionsComfyUINode,
|
|
"ACBPPPCleanupOptions": PromptPostProcessorCleanupOptionsComfyUINode,
|
|
"ACBPPPSelectVariable": PromptPostProcessorSelectVariableComfyUINode,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ACBPromptPostProcessor": "ACB Prompt Post Processor",
|
|
"ACBPPPWildcardOptions": "ACB PPP Wildcard Options",
|
|
"ACBPPPENMappingOptions": "ACB PPP ExtraNetwork Mapping Options",
|
|
"ACBPPPSendToNegativeOptions": "ACB PPP Send-To-Negative Options",
|
|
"ACBPPPCleanupOptions": "ACB PPP Cleanup Options",
|
|
"ACBPPPSelectVariable": "ACB PPP Select Variable",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|