ComfyUI-FairLab/__init__.py at main · yanhuifair/ComfyUI-FairLab · 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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
from .nodes.translate_utility import StringTranslateNode

from .nodes.text_utility import SaveStringToDirectoryNode
from .nodes.text_utility import LoadStringFromDirectoryNode
from .nodes.text_utility import FixUTF8StringNode
from .nodes.text_utility import StringAppendNode
from .nodes.text_utility import StringNode
from .nodes.text_utility import IntNode
from .nodes.text_utility import FloatNode
from .nodes.text_utility import RangeStringNode
from .nodes.text_utility import PrependTagsNode
from .nodes.text_utility import AppendTagsNode
from .nodes.text_utility import ExcludeTagsNode
from .nodes.text_utility import ShowStringNode
from .nodes.text_utility import LoadStringNode
from .nodes.text_utility import UniqueTagsNode
from .nodes.text_utility import ASCIICharNode

from .nodes.image_utility import DownloadImageNode
from .nodes.image_utility import SaveImageToDirectoryNode
from .nodes.image_utility import ResizeImageNode
from .nodes.image_utility import VideoToImageNode
from .nodes.image_utility import ImageToVideoNode
from .nodes.image_utility import LoadImageFromURLNode
from .nodes.image_utility import LoadImageFromDirectoryNode
from .nodes.image_utility import LoadImageBatchFromDirectoryNode
from .nodes.image_utility import FillAlphaNode
from .nodes.image_utility import ImageToBase64Node
from .nodes.image_utility import Base64ToImageNode
from .nodes.image_utility import OutpaintingPadNode
from .nodes.image_utility import ImageSizeNode
from .nodes.image_utility import ImagesRangeNode
from .nodes.image_utility import ImagesIndexNode
from .nodes.image_utility import ImagesCatNode
from .nodes.image_utility import ImageShapeNode
from .nodes.image_utility import ModulationNode
from .nodes.image_utility import ModulationDirectionNode
from .nodes.image_utility import ImageRemoveAlphaNode
from .nodes.image_utility import MaskMapNode
from .nodes.image_utility import DetailMapNode
from .nodes.image_utility import RoughnessToSmoothnessNode
from .nodes.image_utility import SaveImageToFolderNode
from .nodes.image_utility import PureColorImageNode
from .nodes.image_utility import PerfectPixelNode


from .nodes.utility import PrintAnyNode
from .nodes.utility import PrintImageNode
from .nodes.utility import LoraLoaderDualNode
from .nodes.utility import AspectRatiosNode

from .nodes.script import PythonScriptNode

from .nodes.logic import MaxNode
from .nodes.logic import MinNode
from .nodes.logic import AddNode
from .nodes.logic import SubtractNode
from .nodes.logic import MultiplyNode
from .nodes.logic import MultiplyIntNode
from .nodes.logic import DivideNode
from .nodes.logic import NumberNode
from .nodes.logic import IfNode
from .nodes.logic import FloatToIntNode
from .nodes.logic import IntToFloatNode


NODE_CLASS_MAPPINGS = {
    "StringTranslateNode": StringTranslateNode,
    "LoadImageFromDirectoryNode": LoadImageFromDirectoryNode,
    "LoadImageBatchFromDirectoryNode": LoadImageBatchFromDirectoryNode,
    "FillAlphaNode": FillAlphaNode,
    "SaveStringToDirectoryNode": SaveStringToDirectoryNode,
    "LoadStringFromDirectoryNode": LoadStringFromDirectoryNode,
    "FixUTF8StringNode": FixUTF8StringNode,
    "StringAppendNode": StringAppendNode,
    "StringNode": StringNode,
    "IntNode": IntNode,
    "FloatNode": FloatNode,
    "RangeStringNode": RangeStringNode,
    "PrependTagsNode": PrependTagsNode,
    "AppendTagsNode": AppendTagsNode,
    "ExcludeTagsNode": ExcludeTagsNode,
    "LoadStringNode": LoadStringNode,
    "UniqueTagsNode": UniqueTagsNode,
    "ShowStringNode": ShowStringNode,
    "DownloadImageNode": DownloadImageNode,
    "SaveImageToDirectoryNode": SaveImageToDirectoryNode,
    "ResizeImageNode": ResizeImageNode,
    "VideoToImageNode": VideoToImageNode,
    "ImageToVideoNode": ImageToVideoNode,
    "LoadImageFromURLNode": LoadImageFromURLNode,
    "PrintAnyNode": PrintAnyNode,
    "PrintImageNode": PrintImageNode,
    "LoraLoaderDualNode": LoraLoaderDualNode,
    "AspectRatiosNode": AspectRatiosNode,
    "PythonScriptNode": PythonScriptNode,
    "MaxNode": MaxNode,
    "MinNode": MinNode,
    "AddNode": AddNode,
    "SubtractNode": SubtractNode,
    "MultiplyNode": MultiplyNode,
    "MultiplyIntNode": MultiplyIntNode,
    "DivideNode": DivideNode,
    "NumberNode": NumberNode,
    "IfNode": IfNode,
    "FloatToIntNode": FloatToIntNode,
    "IntToFloatNode": IntToFloatNode,
    "ImageToBase64Node": ImageToBase64Node,
    "OutpaintingPadNode": OutpaintingPadNode,
    "ImageSizeNode": ImageSizeNode,
    "ImagesRangeNode": ImagesRangeNode,
    "ImagesIndexNode": ImagesIndexNode,
    "ImagesCatNode": ImagesCatNode,
    "ImageShapeNode": ImageShapeNode,
    "Base64ToImageNode": Base64ToImageNode,
    "ASCIICharNode": ASCIICharNode,
    "ModulationNode": ModulationNode,
    "ModulationDirectionNode": ModulationDirectionNode,
    "ImageRemoveAlphaNode": ImageRemoveAlphaNode,
    "MaskMapNode": MaskMapNode,
    "DetailMapNode": DetailMapNode,
    "RoughnessToSmoothnessNode": RoughnessToSmoothnessNode,
    "SaveImageToFolderNode": SaveImageToFolderNode,
    "PureColorImageNode": PureColorImageNode,
    "PerfectPixelNode": PerfectPixelNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "StringTranslateNode": "String Translate",
    "LoadImageFromDirectoryNode": "Load Image From Directory",
    "LoadImageBatchFromDirectoryNode": "Load Image Batch From Directory",
    "FillAlphaNode": "Fill Alpha",
    "SaveStringToDirectoryNode": "Save String To Directory",
    "LoadStringFromDirectoryNode": "Load String From Directory",
    "FixUTF8StringNode": "Fix UTF-8 String",
    "StringAppendNode": "String Append",
    "StringNode": "String",
    "IntNode": "Int",
    "FloatNode": "Float",
    "RangeStringListNode": "Range String",
    "PrependTagsNode": "Prepend Tags",
    "AppendTagsNode": "Append Tags",
    "ExcludeTagsNode": "Exclude Tags",
    "LoadStringNode": "Load String",
    "UniqueTagsNode": "Unique Tags",
    "ShowStringNode": "Show String",
    "DownloadImageNode": "Download Image",
    "SaveImageToDirectoryNode": "Save Image To Directory",
    "ResizeImageNode": "Resize Image",
    "VideoToImageNode": "Video To Image",
    "ImageToVideoNode": "Image To Video",
    "LoadImageFromURLNode": "Load Image From URL",
    "PrintAnyNode": "Print Any",
    "PrintImageNode": "Print Image",
    "LoraLoaderDualNode": "Load LoRA Dual",
    "AspectRatiosNode": "Aspect Ratios",
    "PythonScriptNode": "Python Script",
    "MaxNode": "Max",
    "MinNode": "Min",
    "AddNode": "Add",
    "SubtractNode": "Subtract",
    "MultiplyNode": "Multiply",
    "MultiplyIntNode": "Multiply Int",
    "DivideNode": "Divide",
    "NumberNode": "Number",
    "IfNode": "If",
    "FloatToIntNode": "Float To Int",
    "IntToFloatNode": "Int To Float",
    "ImageToBase64Node": "Image To Base64",
    "Base64ToImageNode": "Base64 To Image",
    "OutpaintingPadNode": "Outpainting Pad",
    "ImageSizeNode": "Image Size",
    "ImagesRangeNode": "Images Range",
    "ImagesIndexNode": "Images Index",
    "ImagesCatNode": "Images Cat",
    "ImageShapeNode": "Image Shape",
    "ASCIICharNode": "ASCII Art Text",
    "ModulationNode": "Modulation",
    "ModulationDirectionNode": "Modulation Direction",
    "ImageRemoveAlphaNode": "Image Remove Alpha",
    "MaskMapNode": "Mask Map",
    "DetailMapNode": "Detail Map",
    "RoughnessToSmoothnessNode": "Roughness To Smoothness",
    "SaveImageToFolderNode": "Save Image To Folder",
    "PureColorImageNode": "Pure Color Image",
    "PerfectPixelNode": "Perfect Pixel",
}


WEB_DIRECTORY = "./js"

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