800 lines
29 KiB
Python
800 lines
29 KiB
Python
import os
|
|
|
|
# pylint: disable=import-error
|
|
import folder_paths # type: ignore
|
|
import nodes # type: ignore
|
|
|
|
from .ppp import PromptPostProcessor
|
|
from .ppp_hosts import SUPPORTED_APPS
|
|
from .ppp_logging import DEBUG_LEVEL, PromptPostProcessorLogFactory
|
|
from .ppp_wildcards import PPPWildcards
|
|
from .ppp_enmappings import PPPExtraNetworkMappings
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit("This script must be run from ComfyUI")
|
|
|
|
|
|
class PromptPostProcessorComfyUINode:
|
|
"""
|
|
Node for processing prompts.
|
|
"""
|
|
|
|
logger = None
|
|
|
|
def __init__(self):
|
|
lf = PromptPostProcessorLogFactory(SUPPORTED_APPS.comfyui)
|
|
self.logger = lf.log
|
|
grammar_filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), "grammar.lark")
|
|
with open(grammar_filename, "r", encoding="utf-8") as file:
|
|
self.grammar_content = file.read()
|
|
self.wildcards_obj = PPPWildcards(lf.log)
|
|
self.extranetwork_mappings_obj = PPPExtraNetworkMappings(lf.log)
|
|
self.logger.info(f"{PromptPostProcessor.NAME} {PromptPostProcessor.VERSION} initialized")
|
|
|
|
class SmartType(str):
|
|
def __ne__(self, other):
|
|
if self == "*" or other == "*":
|
|
return False
|
|
selfset = set(self.split(","))
|
|
otherset = set(other.split(","))
|
|
return not otherset.issubset(selfset)
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"pos_prompt": (
|
|
"STRING",
|
|
{
|
|
"multiline": True,
|
|
"default": "",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"neg_prompt": (
|
|
"STRING",
|
|
{
|
|
"multiline": True,
|
|
"default": "",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
},
|
|
"optional": {
|
|
"model": (
|
|
cls.SmartType("MODEL,STRING"),
|
|
{
|
|
"default": "",
|
|
"placeholder": "internal model class name",
|
|
},
|
|
),
|
|
"modelname": (
|
|
"STRING",
|
|
{
|
|
"default": "",
|
|
"placeholder": "full path of the model",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"seed": (
|
|
"INT",
|
|
{
|
|
"default": -1,
|
|
"min": -1,
|
|
"tooltip": "Seed value for the prompt processing (use -1 for random)",
|
|
},
|
|
),
|
|
"debug_level": (
|
|
[e.value for e in DEBUG_LEVEL],
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_DEBUG_LEVEL,
|
|
"tooltip": "Debug level",
|
|
},
|
|
),
|
|
"on_warnings": (
|
|
[e.value for e in PromptPostProcessor.ONWARNING_CHOICES],
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_ONWARNING,
|
|
"tooltip": "How to handle invalid content warnings",
|
|
},
|
|
),
|
|
"process_wildcards": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_WC_PROCESS,
|
|
"tooltip": "Process wildcards in the prompt",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"do_cleanup": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_DO_CLEANUP,
|
|
"tooltip": "Do a cleanup of the prompt",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"cleanup_variables": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CLEANUP_VARIABLES,
|
|
"tooltip": "Do a cleanup of the output variables",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"wc_options": (
|
|
"PPP_OPTIONS_WC",
|
|
{
|
|
"default": None,
|
|
"tooltip": "Wildcard processing options",
|
|
},
|
|
),
|
|
"stn_options": (
|
|
"PPP_OPTIONS_STN",
|
|
{
|
|
"default": None,
|
|
"tooltip": "Send-To-Negative options",
|
|
},
|
|
),
|
|
"cup_options": (
|
|
"PPP_OPTIONS_CUP",
|
|
{
|
|
"default": None,
|
|
"tooltip": "Cleanup options",
|
|
},
|
|
),
|
|
"en_options": (
|
|
"PPP_OPTIONS_EN",
|
|
{
|
|
"default": None,
|
|
"tooltip": "ExtraNetworks mapping options",
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, input_types: dict[str, str]):
|
|
it = cls.INPUT_TYPES()
|
|
expected = {
|
|
k: cls.SmartType("COMBO,STRING") if isinstance(v[0], list) else v[0] # we allow string for combos
|
|
for k, v in {**it["required"], **it["optional"]}.items()
|
|
}
|
|
for input_name, input_type in input_types.items():
|
|
t = expected[input_name]
|
|
if input_type != t:
|
|
return f"Invalid type for input '{input_name}': {input_type} (expected {t})"
|
|
return True
|
|
|
|
RETURN_TYPES = (
|
|
"STRING",
|
|
"STRING",
|
|
"PPP_DICT",
|
|
)
|
|
RETURN_NAMES = (
|
|
"pos_prompt",
|
|
"neg_prompt",
|
|
"variables",
|
|
)
|
|
|
|
FUNCTION = "process"
|
|
|
|
CATEGORY = "ACB"
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, **kwargs): # pylint: disable=unused-argument
|
|
return float("NaN") # always process because we don't control the content of wildcards and the config file
|
|
|
|
def process(
|
|
self,
|
|
model,
|
|
modelname,
|
|
pos_prompt,
|
|
neg_prompt,
|
|
seed,
|
|
debug_level,
|
|
on_warnings,
|
|
process_wildcards,
|
|
do_cleanup,
|
|
cleanup_variables,
|
|
wc_options=None,
|
|
stn_options=None,
|
|
cup_options=None,
|
|
en_options=None,
|
|
):
|
|
modelclass = (
|
|
model.model.model_config.__class__.__name__ if model is not None and not isinstance(model, str) else model
|
|
) or ""
|
|
if modelclass == "":
|
|
self.logger.warning("Model class is not provided. System variables might not be properly set.")
|
|
if modelname == "":
|
|
self.logger.warning("Modelname is not provided. System variables will not be properly set.")
|
|
# model class values in ComfyUI\comfy\supported_models.py
|
|
env_info = {
|
|
"app": SUPPORTED_APPS.comfyui.value,
|
|
"models_path": folder_paths.models_dir,
|
|
"model_filename": modelname or "", # path is relative to checkpoints folder
|
|
"model_class": modelclass,
|
|
}
|
|
env_info.update({"is_" + k: False for k in PromptPostProcessor.KNOWN_MODELS})
|
|
env_info.update(
|
|
{
|
|
"is_sd1": modelclass in ("SD15", "SD15_instructpix2pix"),
|
|
"is_sd2": modelclass in ("SD20", "SD21UnclipL", "SD21UnclipH", "LotusD"),
|
|
"is_sdxl": (
|
|
modelclass
|
|
in ("SDXL", "SDXLRefiner", "SDXL_instructpix2pix", "Segmind_Vega", "KOALA_700M", "KOALA_1B")
|
|
),
|
|
"is_ssd": modelclass in ("SSD1B",),
|
|
"is_sd3": modelclass in ("SD3",),
|
|
"is_flux": modelclass in ("Flux", "FluxInpaint", "FluxSchnell"),
|
|
"is_auraflow": modelclass in ("AuraFlow",),
|
|
"is_pixart": modelclass in ("PixArtAlpha", "PixArtSigma"),
|
|
"is_lumina2": modelclass in ("Lumina2", "ZImage"),
|
|
"is_ltxv": modelclass in ("LTXV", "LTXAV"),
|
|
"is_cosmos": modelclass in ("CosmosT2V", "CosmosI2V"),
|
|
"is_cosmospredict2": modelclass in ("CosmosT2IPredict2", "CosmosI2VPredict2"),
|
|
"is_genmomochi": modelclass in ("GenmoMochi",),
|
|
"is_hunyuan": modelclass in ("HunyuanDiT", "HunyuanDiT1"),
|
|
"is_hunyuanvideo": modelclass
|
|
in (
|
|
"HunyuanVideo",
|
|
"HunyuanVideoI2V",
|
|
"HunyuanVideoSkyreelsI2V",
|
|
"HunyuanImage21",
|
|
"HunyuanImage21Refiner",
|
|
"HunyuanVideo15",
|
|
"HunyuanVideo15_SR_Distilled",
|
|
),
|
|
"is_hunyuan3d": modelclass in ("Hunyuan3Dv2", "Hunyuan3Dv2_1", "Hunyuan3Dv2mini"),
|
|
"is_wan": modelclass
|
|
in (
|
|
"WAN21_T2V",
|
|
"WAN21_I2V",
|
|
"WAN21_FunControl2V",
|
|
"WAN21_Camera",
|
|
"WAN22_Camera",
|
|
"WAN21_Vace",
|
|
"WAN21_HuMo",
|
|
"WAN22_S2V",
|
|
"WAN22_Animate",
|
|
"WAN22_T2V",
|
|
),
|
|
"is_hidream": modelclass in ("HiDream",),
|
|
"is_chroma": modelclass in ("Chroma", "ChromaRadiance"),
|
|
"is_omnigen2": modelclass in ("Omnigen2",),
|
|
"is_kandinsky5": modelclass in ("Kandinsky5", "Kandinsky5Image"),
|
|
"is_qwenimage": modelclass in ("QwenImage",),
|
|
"is_flux2": modelclass in ("Flux2"),
|
|
"is_anima": modelclass in ("Anima",),
|
|
}
|
|
)
|
|
# Also supported: SVD_img2vid, SVD3D_u, SVD3_p, Stable_Zero123, SD_X4Upscaler, Stable_Cascade_C, Stable_Cascade_B, StableAudio
|
|
|
|
wc_wildcards_folders = wc_options["wc_wildcards_folders"] if wc_options else ""
|
|
if wc_wildcards_folders == "":
|
|
try:
|
|
fp1 = folder_paths.get_folder_paths("ppp_wildcards")
|
|
except Exception: # pylint: disable=W0718
|
|
fp1 = None
|
|
try:
|
|
fp2 = folder_paths.get_folder_paths("wildcards")
|
|
except Exception: # pylint: disable=W0718
|
|
fp2 = None
|
|
wc_wildcards_folders = ",".join(fp1 or fp2 or [])
|
|
if wc_wildcards_folders == "":
|
|
wc_wildcards_folders = os.getenv("WILDCARD_DIR", PPPWildcards.DEFAULT_WILDCARDS_FOLDER)
|
|
wildcards_folders = [
|
|
(f if os.path.isabs(f) else os.path.abspath(os.path.join(folder_paths.models_dir, f)))
|
|
for f in wc_wildcards_folders.split(",")
|
|
if f.strip() != ""
|
|
]
|
|
en_mappings_folders = en_options["en_mappings_folders"] if en_options else ""
|
|
if en_mappings_folders == "":
|
|
try:
|
|
fp3 = folder_paths.get_folder_paths("ppp_extranetworkmappings")
|
|
except Exception: # pylint: disable=W0718
|
|
fp3 = None
|
|
en_mappings_folders = ",".join(fp3 or [])
|
|
if en_mappings_folders == "":
|
|
en_mappings_folders = os.getenv(
|
|
"EXTRANETWORKMAPPINGS_DIR", PPPExtraNetworkMappings.DEFAULT_ENMAPPINGS_FOLDER
|
|
)
|
|
enmappings_folders = [
|
|
(f if os.path.isabs(f) else os.path.abspath(os.path.join(folder_paths.models_dir, f)))
|
|
for f in en_mappings_folders.split(",")
|
|
if f.strip() != ""
|
|
]
|
|
|
|
options = {
|
|
"debug_level": debug_level,
|
|
"on_warnings": on_warnings,
|
|
"process_wildcards": process_wildcards,
|
|
"if_wildcards": (
|
|
wc_options["wc_if_wildcards"] if wc_options else PromptPostProcessor.IFWILDCARDS_CHOICES.stop.value
|
|
),
|
|
"choice_separator": (
|
|
wc_options["wc_choice_separator"] if wc_options else PromptPostProcessor.DEFAULT_CHOICE_SEPARATOR
|
|
),
|
|
"keep_choices_order": (
|
|
wc_options["wc_keep_choices_order"] if wc_options else PromptPostProcessor.DEFAULT_KEEP_CHOICES_ORDER
|
|
),
|
|
"stn_separator": stn_options["stn_separator"] if stn_options else PromptPostProcessor.DEFAULT_STN_SEPARATOR,
|
|
"stn_ignore_repeats": (
|
|
stn_options["stn_ignore_repeats"] if stn_options else PromptPostProcessor.DEFAULT_STN_IGNORE_REPEATS
|
|
),
|
|
"do_cleanup": do_cleanup,
|
|
"cleanup_variables": cleanup_variables,
|
|
"cleanup_extra_spaces": (
|
|
cup_options["cup_extra_spaces"] if cup_options else PromptPostProcessor.DEFAULT_CUP_EXTRA_SPACES
|
|
),
|
|
"cleanup_empty_constructs": (
|
|
cup_options["cup_empty_constructs"] if cup_options else PromptPostProcessor.DEFAULT_CUP_EMPTY_CONSTRUCTS
|
|
),
|
|
"cleanup_extra_separators": (
|
|
cup_options["cup_extra_separators"] if cup_options else PromptPostProcessor.DEFAULT_CUP_EXTRA_SEPARATORS
|
|
),
|
|
"cleanup_extra_separators2": (
|
|
cup_options["cup_extra_separators2"]
|
|
if cup_options
|
|
else PromptPostProcessor.DEFAULT_CUP_EXTRA_SEPARATORS2
|
|
),
|
|
"cleanup_extra_separators_include_eol": (
|
|
cup_options["cup_extra_separators_include_eol"]
|
|
if cup_options
|
|
else PromptPostProcessor.DEFAULT_CUP_EXTRA_SEPARATORS_INCLUDE_EOL
|
|
),
|
|
"cleanup_breaks": cup_options["cup_breaks"] if cup_options else PromptPostProcessor.DEFAULT_CUP_BREAKS,
|
|
"cleanup_breaks_eol": (
|
|
cup_options["cup_breaks_eol"] if cup_options else PromptPostProcessor.DEFAULT_CUP_BREAKS_EOL
|
|
),
|
|
"cleanup_ands": cup_options["cup_ands"] if cup_options else PromptPostProcessor.DEFAULT_CUP_ANDS,
|
|
"cleanup_ands_eol": (
|
|
cup_options["cup_ands_eol"] if cup_options else PromptPostProcessor.DEFAULT_CUP_ANDS_EOL
|
|
),
|
|
"cleanup_extranetwork_tags": (
|
|
cup_options["cup_extranetwork_tags"]
|
|
if cup_options
|
|
else PromptPostProcessor.DEFAULT_CUP_EXTRANETWORK_TAGS
|
|
),
|
|
"cleanup_merge_attention": (
|
|
cup_options["cup_merge_attention"] if cup_options else PromptPostProcessor.DEFAULT_CUP_MERGE_ATTENTION
|
|
),
|
|
"remove_extranetwork_tags": (
|
|
cup_options["cup_remove_extranetwork_tags"]
|
|
if cup_options
|
|
else PromptPostProcessor.DEFAULT_CUP_REMOVE_EXTRANETWORK_TAGS
|
|
),
|
|
}
|
|
self.wildcards_obj.refresh_wildcards(
|
|
debug_level,
|
|
wildcards_folders if options["process_wildcards"] else None,
|
|
wc_options["wc_wildcards_input"] if wc_options else "",
|
|
)
|
|
self.extranetwork_mappings_obj.refresh_extranetwork_mappings(
|
|
debug_level,
|
|
enmappings_folders,
|
|
en_options["en_mappings_input"] if en_options else "",
|
|
)
|
|
ppp = PromptPostProcessor(
|
|
self.logger,
|
|
self.interrupt,
|
|
env_info,
|
|
options,
|
|
self.grammar_content,
|
|
self.wildcards_obj,
|
|
self.extranetwork_mappings_obj,
|
|
)
|
|
pos_prompt, neg_prompt, variables = ppp.process_prompt(pos_prompt, neg_prompt, seed if seed is not None else 1)
|
|
return (
|
|
pos_prompt,
|
|
neg_prompt,
|
|
variables,
|
|
)
|
|
|
|
def interrupt(self):
|
|
nodes.interrupt_processing(True)
|
|
|
|
|
|
class PromptPostProcessorWildcardOptionsComfyUINode:
|
|
"""
|
|
Node for wildcard options.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {
|
|
"folders": (
|
|
"STRING",
|
|
{
|
|
"default": "",
|
|
"tooltip": "Comma separated list of wildcards folders",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"definitions": (
|
|
"STRING",
|
|
{
|
|
"default": "",
|
|
"multiline": True,
|
|
"placeholder": "wildcards definitions",
|
|
"tooltip": "Wildcards definitions in yaml/json format",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"if_wildcards": (
|
|
[e.value for e in PromptPostProcessor.IFWILDCARDS_CHOICES],
|
|
{
|
|
"default": PromptPostProcessor.IFWILDCARDS_CHOICES.stop.value,
|
|
"tooltip": "How to handle invalid wildcards in the prompt",
|
|
},
|
|
),
|
|
"choice_separator": (
|
|
"STRING",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CHOICE_SEPARATOR,
|
|
"tooltip": "Default separator for selected choices",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"keep_choices_order": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_KEEP_CHOICES_ORDER,
|
|
"tooltip": "Keep the order of the choices in the prompt",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("PPP_OPTIONS_WC",)
|
|
RETURN_NAMES = ("options",)
|
|
|
|
FUNCTION = "process"
|
|
|
|
CATEGORY = "ACB"
|
|
|
|
def process(
|
|
self,
|
|
folders: str,
|
|
definitions: str,
|
|
if_wildcards,
|
|
choice_separator: str,
|
|
keep_choices_order: bool,
|
|
):
|
|
options = {
|
|
"wc_wildcards_folders": folders,
|
|
"wc_wildcards_input": definitions,
|
|
"wc_if_wildcards": if_wildcards,
|
|
"wc_choice_separator": choice_separator,
|
|
"wc_keep_choices_order": keep_choices_order,
|
|
}
|
|
return (options,)
|
|
|
|
|
|
class PromptPostProcessorSTNOptionsComfyUINode:
|
|
"""
|
|
Node for Send-To-Negative options.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {
|
|
"separator": (
|
|
"STRING",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_STN_SEPARATOR,
|
|
"tooltip": "Separator for the content added to the negative prompt",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"ignore_repeats": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_STN_IGNORE_REPEATS,
|
|
"tooltip": "Ignore repeated content added to the negative prompt",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("PPP_OPTIONS_STN",)
|
|
RETURN_NAMES = ("options",)
|
|
|
|
FUNCTION = "process"
|
|
|
|
CATEGORY = "ACB"
|
|
|
|
def process(
|
|
self,
|
|
separator: str,
|
|
ignore_repeats: bool,
|
|
):
|
|
options = {
|
|
"stn_separator": separator,
|
|
"stn_ignore_repeats": ignore_repeats,
|
|
}
|
|
return (options,)
|
|
|
|
|
|
class PromptPostProcessorCleanupOptionsComfyUINode:
|
|
"""
|
|
Node for Cleanup options.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {
|
|
"extra_spaces": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_EXTRA_SPACES,
|
|
"tooltip": "Remove extra spaces",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"empty_constructs": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_EMPTY_CONSTRUCTS,
|
|
"tooltip": "Remove empty constructs",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"extra_separators": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_EXTRA_SEPARATORS,
|
|
"tooltip": "Remove extra separators",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"extra_separators_additional": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_EXTRA_SEPARATORS2,
|
|
"tooltip": "Remove extra separators (additional cases)",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"extra_separators_include_eol": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_EXTRA_SEPARATORS_INCLUDE_EOL,
|
|
"tooltip": "Extra separators options also remove EOLs",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"around_breaks": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_BREAKS,
|
|
"tooltip": "Cleanup around BREAKs",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"breaks_with_eol": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_BREAKS_EOL,
|
|
"tooltip": "Set BREAKs in their own line",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"around_ands": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_ANDS,
|
|
"tooltip": "Cleanup around ANDs",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"ands_with_eol": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_ANDS_EOL,
|
|
"tooltip": "Set ANDs in their own line",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"around_extranetwork_tags": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_EXTRANETWORK_TAGS,
|
|
"tooltip": "Clean up around extra network tags",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"merge_attention": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_MERGE_ATTENTION,
|
|
"tooltip": "Merge nested attention constructs",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
"remove_extranetwork_tags": (
|
|
"BOOLEAN",
|
|
{
|
|
"default": PromptPostProcessor.DEFAULT_CUP_REMOVE_EXTRANETWORK_TAGS,
|
|
"tooltip": "Remove extra network tags",
|
|
"label_on": "Yes",
|
|
"label_off": "No",
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("PPP_OPTIONS_CUP",)
|
|
RETURN_NAMES = ("options",)
|
|
|
|
FUNCTION = "process"
|
|
|
|
CATEGORY = "ACB"
|
|
|
|
def process(
|
|
self,
|
|
extra_spaces: bool,
|
|
empty_constructs: bool,
|
|
extra_separators: bool,
|
|
extra_separators_additional: bool,
|
|
extra_separators_include_eol: bool,
|
|
around_breaks: bool,
|
|
breaks_with_eol: bool,
|
|
around_ands: bool,
|
|
ands_with_eol: bool,
|
|
around_extranetwork_tags: bool,
|
|
merge_attention: bool,
|
|
remove_extranetwork_tags: bool,
|
|
):
|
|
options = {
|
|
"cup_extra_spaces": extra_spaces,
|
|
"cup_empty_constructs": empty_constructs,
|
|
"cup_extra_separators": extra_separators,
|
|
"cup_extra_separators2": extra_separators_additional,
|
|
"cup_extra_separators_include_eol": extra_separators_include_eol,
|
|
"cup_breaks": around_breaks,
|
|
"cup_breaks_eol": breaks_with_eol,
|
|
"cup_ands": around_ands,
|
|
"cup_ands_eol": ands_with_eol,
|
|
"cup_extranetwork_tags": around_extranetwork_tags,
|
|
"cup_merge_attention": merge_attention,
|
|
"cup_remove_extranetwork_tags": remove_extranetwork_tags,
|
|
}
|
|
return (options,)
|
|
|
|
|
|
class PromptPostProcessorENMappingOptionsComfyUINode:
|
|
"""
|
|
Node for ExtraNetworks mapping options.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {
|
|
"folders": (
|
|
"STRING",
|
|
{
|
|
"default": "",
|
|
"tooltip": "Comma separated list of extranetwork mappings folders",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
"definitions": (
|
|
"STRING",
|
|
{
|
|
"default": "",
|
|
"multiline": True,
|
|
"placeholder": "extranetwork mappings definitions",
|
|
"tooltip": "Extranetwork mappings definitions in yaml format",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("PPP_OPTIONS_EN",)
|
|
RETURN_NAMES = ("options",)
|
|
|
|
FUNCTION = "process"
|
|
|
|
CATEGORY = "ACB"
|
|
|
|
def process(
|
|
self,
|
|
folders: str,
|
|
definitions: str,
|
|
):
|
|
options = {
|
|
"en_mappings_folders": folders,
|
|
"en_mappings_input": definitions,
|
|
}
|
|
return (options,)
|
|
|
|
|
|
class PromptPostProcessorSelectVariableComfyUINode:
|
|
"""
|
|
Node for selecting a variable from a dictionary.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"variables": (
|
|
"PPP_DICT",
|
|
{
|
|
"forceInput": True,
|
|
},
|
|
),
|
|
},
|
|
"optional": {
|
|
"name": (
|
|
"STRING",
|
|
{
|
|
"placeholder": "variable name",
|
|
"multiline": False,
|
|
"default": "",
|
|
"dynamicPrompts": False,
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
RETURN_NAMES = ("value",)
|
|
|
|
FUNCTION = "select"
|
|
|
|
CATEGORY = "ACB"
|
|
|
|
def select(
|
|
self,
|
|
variables: dict[str, str],
|
|
name: str,
|
|
):
|
|
value = ""
|
|
if variables:
|
|
if name == "":
|
|
value = "\n".join(f"{k}: {v}" for k, v in variables.items())
|
|
elif name in variables:
|
|
value = variables[name]
|
|
return (value,)
|