scofano
3fdefe036b
Introduce a new Text/Utils node that applies editable per-character replacement rules from a multiline mapping list, including support for character removal and ordered replacements. Update the README with usage details, default replacement rules, and examples so users can normalize punctuation and clean text directly in ComfyUI workflows.feat: add character search/replace utility node Introduce a new Text/Utils node that applies editable per-character replacement rules from a multiline mapping list, including support for character removal and ordered replacements. Update the README with usage details, default replacement rules, and examples so users can normalize punctuation and clean text directly in ComfyUI workflows.
100 lines
2.4 KiB
Python
100 lines
2.4 KiB
Python
# character_search_replace.py
|
||
from typing import List, Tuple
|
||
|
||
|
||
DEFAULT_REPLACEMENTS = (
|
||
"“\"\n"
|
||
"”\"\n"
|
||
"„\"\n"
|
||
"«\"\n"
|
||
"»\"\n"
|
||
"‘'\n"
|
||
"’'\n"
|
||
"‚'\n"
|
||
"´'\n"
|
||
"`'\n"
|
||
"–-\n"
|
||
"—, \n"
|
||
"−-\n"
|
||
"‐-\n"
|
||
"‑-\n"
|
||
"•-\n"
|
||
"·-\n"
|
||
"*\n"
|
||
" _\n"
|
||
"…..."
|
||
)
|
||
|
||
|
||
def _parse_replacement_rules(replacements: str) -> List[Tuple[str, str]]:
|
||
rules: List[Tuple[str, str]] = []
|
||
|
||
for line in replacements.splitlines():
|
||
if line == "":
|
||
continue
|
||
|
||
search = line[0]
|
||
replace = line[1:]
|
||
rules.append((search, replace))
|
||
|
||
return rules
|
||
|
||
|
||
def _replace_characters(text: str, replacements: str) -> str:
|
||
result = text
|
||
|
||
for search, replace in _parse_replacement_rules(replacements):
|
||
result = result.replace(search, replace)
|
||
|
||
return result
|
||
|
||
|
||
class CharacterSearchReplace:
|
||
"""
|
||
Replaces characters in a string using a multiline mapping list.
|
||
|
||
Each non-empty line in the replacement list defines one rule:
|
||
- first character: character to search for
|
||
- remaining characters on the same line: replacement text
|
||
- if the line contains only one character, that character is removed
|
||
|
||
Spaces are treated as valid characters, so rules can search for or replace
|
||
spaces as needed.
|
||
"""
|
||
|
||
@classmethod
|
||
def INPUT_TYPES(cls):
|
||
return {
|
||
"required": {
|
||
"text": ("STRING", {"multiline": True, "default": ""}),
|
||
"replacements": (
|
||
"STRING",
|
||
{"multiline": True, "default": DEFAULT_REPLACEMENTS},
|
||
),
|
||
}
|
||
}
|
||
|
||
RETURN_TYPES = ("STRING",)
|
||
RETURN_NAMES = ("processed_text",)
|
||
FUNCTION = "process"
|
||
CATEGORY = "Text/Utils"
|
||
OUTPUT_NODE = False
|
||
|
||
def process(self, text: str, replacements: str) -> Tuple[str]:
|
||
if text is None:
|
||
text = ""
|
||
if not isinstance(text, str):
|
||
text = str(text)
|
||
|
||
if replacements is None:
|
||
replacements = ""
|
||
if not isinstance(replacements, str):
|
||
replacements = str(replacements)
|
||
|
||
return (_replace_characters(text, replacements),)
|
||
|
||
@classmethod
|
||
def IS_CHANGED(cls, **kwargs) -> float:
|
||
text = kwargs.get("text", "") or ""
|
||
replacements = kwargs.get("replacements", "") or ""
|
||
return float(len(text) + len(replacements)) |