scofano
4c32fde481
Add Line Context Counter and UTF-8 Cleaner nodes alongside the existing Text Stats functionality, including line counts and text cleanup utilities for ComfyUI workflows. Update the README to reflect the new node pack scope, installation steps, and usage details so users can discover and use the added text traversal and normalization features.feat: expand text stats into multi-node text utility pack Add Line Context Counter and UTF-8 Cleaner nodes alongside the existing Text Stats functionality, including line counts and text cleanup utilities for ComfyUI workflows. Update the README to reflect the new node pack scope, installation steps, and usage details so users can discover and use the added text traversal and normalization features.
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
# text_stats.py
|
|
import re
|
|
from typing import Tuple
|
|
|
|
# NOTE:
|
|
# - Input: STRING (multiline enabled)
|
|
# - Output: INT (character count), INT (word count), INT (line count)
|
|
# - Category: Text / Utils
|
|
# - Deterministic: yes
|
|
|
|
_WORD_PATTERN = re.compile(r"\b\w+\b", flags=re.UNICODE)
|
|
|
|
class TextStats:
|
|
"""
|
|
A simple ComfyUI node that counts characters, words, and lines in a string.
|
|
- Character count is the length of the string (including spaces and newlines).
|
|
- Word count uses a Unicode-aware word regex: r"\\b\\w+\\b".
|
|
- Line count uses splitlines(), so an empty string has 0 lines.
|
|
"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"text": ("STRING", {"multiline": True, "default": ""}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT", "INT", "INT")
|
|
RETURN_NAMES = ("char_count", "word_count", "line_count")
|
|
FUNCTION = "compute"
|
|
CATEGORY = "Text/Utils"
|
|
# Mark as deterministic: same input -> same output, unaffected by seeds.
|
|
OUTPUT_NODE = False
|
|
|
|
def compute(self, text: str) -> Tuple[int, int, int]:
|
|
if text is None:
|
|
text = ""
|
|
if not isinstance(text, str):
|
|
text = str(text)
|
|
|
|
char_count = len(text)
|
|
# Unicode-aware word counting:
|
|
# Counts sequences of letters/digits/underscore bounded by word boundaries.
|
|
word_count = len(_WORD_PATTERN.findall(text))
|
|
line_count = len(text.splitlines())
|
|
|
|
# ComfyUI expects a tuple aligned with RETURN_TYPES
|
|
return (char_count, word_count, line_count)
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, **kwargs) -> float:
|
|
"""
|
|
Optional optimization hint to ComfyUI: change hash for caching.
|
|
Return a value that changes when inputs change.
|
|
Here we derive from the text content length + hash to be safe.
|
|
"""
|
|
text = kwargs.get("text", "") or ""
|
|
# lightweight, stable-ish indicator
|
|
return float(len(text))
|