xiaden
315666c7b6
- Split flat node modules into categorized subpackages (tagging, detect, edge, pose, segment, restore, compare, judge, transform, utility) - Extract shared internals into _shared/ (bases, tensor, formatting) - Centralize node registration via register_nodes() helper - Clean up: remove dead is_scored return, unnecessary hasattr guard, inline imports
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
"""Case-insensitive substring membership check on label strings."""
|
|
|
|
from comfy_api.latest import io
|
|
|
|
|
|
class ImgUtilsLabelContains(io.ComfyNode):
|
|
"""Check whether a comma-separated label string contains a search term (case-insensitive)."""
|
|
|
|
@classmethod
|
|
def define_schema(cls) -> io.Schema:
|
|
return io.Schema(
|
|
node_id="ImgUtilsLabelContains",
|
|
display_name="Imgutils Label Contains",
|
|
category="imgutils/utility",
|
|
description=(
|
|
"Check whether a comma-separated label string contains a search term. "
|
|
"Case-insensitive. Useful for branching on detection/tagging results."
|
|
),
|
|
search_aliases=["contains", "label", "search", "filter", "match", "boolean"],
|
|
inputs=[
|
|
io.String.Input(
|
|
"labels",
|
|
tooltip="Comma-separated label string (e.g. from Detect or Tagging output).",
|
|
),
|
|
io.String.Input(
|
|
"search",
|
|
tooltip="Term to search for in the labels. Case-insensitive.",
|
|
),
|
|
],
|
|
outputs=[
|
|
io.Boolean.Output(display_name="boolean"),
|
|
],
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, labels, search) -> io.NodeOutput:
|
|
return io.NodeOutput(search.lower() in labels.lower())
|