86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
import os
|
|
import folder_paths
|
|
|
|
|
|
def strip_comments(text):
|
|
"""
|
|
Strip lines that start with # but preserve lines starting with \\#.
|
|
|
|
Args:
|
|
text: Input text that may contain comment lines
|
|
|
|
Returns:
|
|
Text with comment lines removed
|
|
"""
|
|
lines = text.split('\n')
|
|
filtered_lines = []
|
|
|
|
for line in lines:
|
|
stripped = line.lstrip()
|
|
# Keep line if it starts with \# or doesn't start with #
|
|
if stripped.startswith('\\#'):
|
|
# Remove the escape character
|
|
unescaped = line.replace('\\#', '#', 1)
|
|
filtered_lines.append(unescaped)
|
|
elif not stripped.startswith('#'):
|
|
filtered_lines.append(line)
|
|
|
|
return '\n'.join(filtered_lines)
|
|
|
|
|
|
UPSCALE_DIR = os.path.join(folder_paths.models_dir, "upscale_models")
|
|
|
|
|
|
def get_upscale_model_list():
|
|
exts = (".pth", ".pt", ".safetensors")
|
|
models = []
|
|
|
|
for root, _, files in os.walk(UPSCALE_DIR):
|
|
for name in files:
|
|
if name.lower().endswith(exts):
|
|
full_path = os.path.join(root, name)
|
|
rel_path = os.path.relpath(full_path, UPSCALE_DIR)
|
|
models.append(rel_path)
|
|
|
|
return ["none"] + sorted(models)
|
|
|
|
|
|
class Node:
|
|
"""Wrapper for ComfyUI nodes to simplify function calls."""
|
|
|
|
def __init__(self, node_name):
|
|
"""
|
|
Initialize the wrapper with a node name.
|
|
|
|
Args:
|
|
node_name: Name of the node in NODE_CLASS_MAPPINGS
|
|
"""
|
|
self.node_name = node_name
|
|
self._node_instance = None
|
|
self._function = None
|
|
|
|
@property
|
|
def node(self):
|
|
"""Lazy load and cache the node instance."""
|
|
if self._node_instance is None:
|
|
from nodes import NODE_CLASS_MAPPINGS
|
|
|
|
node_class = NODE_CLASS_MAPPINGS.get(self.node_name)
|
|
if node_class is None:
|
|
raise ValueError(
|
|
f"Node '{self.node_name}' not found in mappings"
|
|
)
|
|
self._node_instance = node_class()
|
|
return self._node_instance
|
|
|
|
@property
|
|
def function(self):
|
|
"""Get the node's main function."""
|
|
if self._function is None:
|
|
self._function = getattr(self.node, self.node.FUNCTION)
|
|
return self._function
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
"""Allow calling the wrapper directly."""
|
|
return self.function(*args, **kwargs)
|