ComfyUI-NO8D-controls/__init__.py at main · Sluggish20/ComfyUI-NO8D-controls · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
"""NO8D controls nodes for ComfyUI."""

from . import prompt_server  # noqa: F401
from .slider_lora_stack import (
    NODE_CLASS_MAPPINGS as _STACK_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _STACK_NODE_DISPLAY_NAME_MAPPINGS,
)
from .compare_slider_preview import (
    NODE_CLASS_MAPPINGS as _COMPARE_SLIDER_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _COMPARE_SLIDER_NODE_DISPLAY_NAME_MAPPINGS,
)
from .image_loader import (
    NODE_CLASS_MAPPINGS as _IMAGE_LOADER_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _IMAGE_LOADER_NODE_DISPLAY_NAME_MAPPINGS,
)
from .prompt_plus import (
    NODE_CLASS_MAPPINGS as _PROMPT_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _PROMPT_NODE_DISPLAY_NAME_MAPPINGS,
)
from .prompt_merge import (
    NODE_CLASS_MAPPINGS as _PROMPT_MERGE_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _PROMPT_MERGE_NODE_DISPLAY_NAME_MAPPINGS,
)
from .conditioning_filter import (
    NODE_CLASS_MAPPINGS as _CONDITIONING_FILTER_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _CONDITIONING_FILTER_NODE_DISPLAY_NAME_MAPPINGS,
)
from .krea2_reference_latent_match import (
    NODE_CLASS_MAPPINGS as _KREA2_REFERENCE_MATCH_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _KREA2_REFERENCE_MATCH_NODE_DISPLAY_NAME_MAPPINGS,
)
from .save_image_text_dataset import (
    NODE_CLASS_MAPPINGS as _SAVE_DATASET_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _SAVE_DATASET_NODE_DISPLAY_NAME_MAPPINGS,
)
from .empty_latent import (
    NODE_CLASS_MAPPINGS as _EMPTY_LATENT_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _EMPTY_LATENT_NODE_DISPLAY_NAME_MAPPINGS,
)
from .generate import (
    NODE_CLASS_MAPPINGS as _GENERATE_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _GENERATE_NODE_DISPLAY_NAME_MAPPINGS,
)
from .krea2_reference import (
    NODE_CLASS_MAPPINGS as _KREA2_REFERENCE_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _KREA2_REFERENCE_NODE_DISPLAY_NAME_MAPPINGS,
)
from .image_grid import (
    NODE_CLASS_MAPPINGS as _IMAGE_GRID_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _IMAGE_GRID_NODE_DISPLAY_NAME_MAPPINGS,
)
from .krea_style_selector import (
    NODE_CLASS_MAPPINGS as _KREA_STYLE_NODE_CLASS_MAPPINGS,
    NODE_DISPLAY_NAME_MAPPINGS as _KREA_STYLE_NODE_DISPLAY_NAME_MAPPINGS,
)


NODE_CLASS_MAPPINGS = {}
NODE_CLASS_MAPPINGS.update(_STACK_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_COMPARE_SLIDER_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_IMAGE_LOADER_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_PROMPT_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_PROMPT_MERGE_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_CONDITIONING_FILTER_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_KREA2_REFERENCE_MATCH_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_SAVE_DATASET_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_EMPTY_LATENT_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_GENERATE_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_KREA2_REFERENCE_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_IMAGE_GRID_NODE_CLASS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(_KREA_STYLE_NODE_CLASS_MAPPINGS)

NODE_DISPLAY_NAME_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS.update(_STACK_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_COMPARE_SLIDER_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_IMAGE_LOADER_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_PROMPT_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_PROMPT_MERGE_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_CONDITIONING_FILTER_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_KREA2_REFERENCE_MATCH_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_SAVE_DATASET_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_EMPTY_LATENT_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_GENERATE_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_KREA2_REFERENCE_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_IMAGE_GRID_NODE_DISPLAY_NAME_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(_KREA_STYLE_NODE_DISPLAY_NAME_MAPPINGS)

WEB_DIRECTORY = "./web"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]