Files
Zuellni-ComfyUI-ExLlama-Nodes/text.py
T

56 lines
1.3 KiB
Python

_CATEGORY = "Zuellni/Text"
_MAPPING = "ZuellniText"
class Previewer:
@classmethod
def INPUT_TYPES(cls):
return {"required": {"text": ("STRING", {"forceInput": True})}}
CATEGORY = _CATEGORY
FUNCTION = "preview"
OUTPUT_NODE = True
RETURN_TYPES = ()
def preview(self, text):
return {"ui": {"text": [text]}}
class Replacer:
@classmethod
def INPUT_TYPES(cls):
return {
"required": {"text": ("STRING", {"multiline": True})},
"optional": {
"a": ("STRING", {"forceInput": True, "multiline": True}),
"b": ("STRING", {"forceInput": True, "multiline": True}),
"c": ("STRING", {"forceInput": True, "multiline": True}),
"d": ("STRING", {"forceInput": True, "multiline": True}),
},
}
CATEGORY = _CATEGORY
FUNCTION = "replace"
RETURN_NAMES = ("TEXT",)
RETURN_TYPES = ("STRING",)
def replace(self, text, **inputs):
for key, value in inputs.items():
if value:
text = text.replace(f"[{key}]", value)
return (text,)
NODE_CLASS_MAPPINGS = {
f"{_MAPPING}Previewer": Previewer,
f"{_MAPPING}Replacer": Replacer,
}
NODE_DISPLAY_NAME_MAPPINGS = {
f"{_MAPPING}Previewer": "Preview",
f"{_MAPPING}Replacer": "Replace",
}
WEB_DIRECTORY = "."