Behzad
cec7f9f9cd
The two template sets shared one GRID_SUBJECTS list even though they load different styles, so the "fully dressed"/"dressed" phrasing was leaking into the FLUX set. That phrasing exists for Z-Image alone: the Renaissance style compiles an empty negative and Z-Image has no negative channel to hold one. Split it into ZIMAGE_GRID_SUBJECTS and FLUX_GRID_SUBJECTS, each with its own demo subject. Z-Image's three are unchanged. FLUX gets three built around props, spill and cast shadow. The FLUX set also moves from Noir Low-Key Portrait to Raw Theatrical Cinematic (azch4ptx-2ca7d3a07f37) — same reasoning as before (light over brushwork, subject-agnostic, carries inspiration images), a style whose staged-clutter identity suits the new subjects. Known gap: the five FLUX template thumbnails still show the old style and subjects. They need re-rendering in a follow-up.
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
"""
|
|
StyleRef nodes for ComfyUI — https://styleref.io
|
|
|
|
Define a creative style once on StyleRef, then load it as a node and reuse it
|
|
across your workflows.
|
|
|
|
MIT licensed. Issues: https://github.com/StyleRef/styleref-comfyui
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
# The node modules import `styleref_api` etc. as top-level names. ComfyUI adds
|
|
# custom_nodes/<pack>/ to sys.path only in some layouts, so make it explicit
|
|
# rather than depending on how this pack was installed.
|
|
_PACK_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
if _PACK_DIR not in sys.path:
|
|
sys.path.insert(0, _PACK_DIR)
|
|
|
|
from styleref_nodes.apply import StyleRefApply # noqa: E402
|
|
from styleref_nodes.facets import StyleRefFacets # noqa: E402
|
|
from styleref_nodes.load import StyleRefLoad # noqa: E402
|
|
from styleref_nodes.login import StyleRefLogin # noqa: E402
|
|
from styleref_nodes.reference_images import StyleRefReferenceImages # noqa: E402
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"StyleRefLoad": StyleRefLoad,
|
|
"StyleRefApply": StyleRefApply,
|
|
"StyleRefFacets": StyleRefFacets,
|
|
"StyleRefReferenceImages": StyleRefReferenceImages,
|
|
"StyleRefLogin": StyleRefLogin,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"StyleRefLoad": "StyleRef Load",
|
|
"StyleRefApply": "StyleRef Apply",
|
|
"StyleRefFacets": "StyleRef Facets",
|
|
"StyleRefReferenceImages": "StyleRef Reference Images",
|
|
"StyleRefLogin": "StyleRef Login",
|
|
}
|
|
|
|
# Served by ComfyUI at /extensions/styleref-comfyui/ — the search widget's
|
|
# progressive enhancement. Nothing here is required for the nodes to work.
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
_FALLBACK_VERSION = "1.0.5"
|
|
|
|
|
|
def _pack_version() -> str:
|
|
"""Single-source the version from pyproject.toml, with a constant fallback."""
|
|
try:
|
|
with open(os.path.join(_PACK_DIR, "pyproject.toml"), encoding="utf-8") as fh:
|
|
match = re.search(r'^version\s*=\s*"([^"]+)"', fh.read(), re.MULTILINE)
|
|
return match.group(1) if match else _FALLBACK_VERSION
|
|
except OSError:
|
|
return _FALLBACK_VERSION
|
|
|
|
|
|
# The install docs tell users to look for a `[StyleRef]` line at startup, so a
|
|
# healthy import must print one — silence would read as a failed install.
|
|
print(f"[StyleRef] v{_pack_version()} — {len(NODE_CLASS_MAPPINGS)} nodes registered")
|
|
|
|
# Registering the search route is optional: without it the enhanced widget
|
|
# simply doesn't populate, and the vanilla `search` field still works.
|
|
try:
|
|
import server_routes # noqa: F401
|
|
except Exception as err: # noqa: BLE001 - never block node registration
|
|
print(f"[StyleRef] Search route unavailable ({err}); using the plain search field.")
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|