-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy path__init__.py
66 lines (63 loc) · 3.04 KB
/
__init__.py
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
"""
@author: Manny Gonzalez
@title: 🐯 YFG Comical Nodes
@nickname: 🐯 YFG Comical Nodes
@description: Utility custom nodes for special effects, image manipulation and quality of life tools.
"""
from .image2histograms import ImageHistogramsNode
from .image2histogramscompact import ImageHistogramsNodeCompact
from .image2halftone import ImageHalftoneNode
from .images2sidebyside import ImagesSideBySideNode
from .switchers.imageSwitchers import *
from .loaders.image2imgbb import Image2ImgBB
from .loaders.imgbbLoader import *
from .loaders.smartcheckpointloader import SmartCheckpointLoader
from .image2contrastmask import ImageToContrastMask
from .VAEDecodePreview import VAEDecodePreview
from .MonoClip import MonoClip
from .pixelart import PixelArt
from .textMaskOverlay import TextMaskOverlay
NODE_CLASS_MAPPINGS = {
"image_histograms_node": ImageHistogramsNode,
"image_histograms_node_compact": ImageHistogramsNodeCompact,
"image_halftone": ImageHalftoneNode,
"images_side_by_side": ImagesSideBySideNode,
"Image3Switcher_node": Image3SwitcherNode,
"Image5Switcher_node": Image5SwitcherNode,
"Image10Switcher_node": Image10SwitcherNode,
"Image15Switcher_node": Image15SwitcherNode,
"Image20Switcher_node": Image20SwitcherNode,
"image2imbgg_node": Image2ImgBB,
"imgbbLoader_node": ImgbbLoader,
"storeURL_node": storeURL,
"smartCheckpointLoader_node": SmartCheckpointLoader,
"image2contrastMask_node": ImageToContrastMask,
"VAEDecodePreview_node": VAEDecodePreview,
"MonoClip_node": MonoClip,
"PixelArt_node": PixelArt,
"textMaskOverlay_node": TextMaskOverlay,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"image_histograms_node": "\ud83d\udc2f YFG Histograms Generator",
"image_histograms_node_compact": "\ud83d\udc2f YFG Histograms Generator (compact)",
"image_halftone": "\ud83d\udc2f YFG Image Halftone Generator",
"images_side_by_side": "\ud83d\udc2f YFG Image Side by Side",
"Image3Switcher_node": "\ud83d\udc2f YFG 3 Image Switcher",
"Image5Switcher_node": "\ud83d\udc2f YFG 5 Image Switcher",
"Image10Switcher_node": "\ud83d\udc2f YFG 10 Image Switcher",
"Image15Switcher_node": "\ud83d\udc2f YFG 15 Image Switcher",
"Image20Switcher_node": "\ud83d\udc2f YFG 20 Image Switcher",
"image2imbgg_node": "\ud83d\udc2f YFG Image to imgBB",
"imgbbLoader_node": "\ud83d\udc2f YFG imgBB to Image",
"storeURL_node": "\ud83d\udc2f YFG Store URL",
"smartCheckpointLoader_node": "\ud83d\udc2f YFG Smart Checkpoint Loader",
"image2contrastMask_node": "\ud83d\udc2f YFG Image To Contrast Mask",
"VAEDecodePreview_node": "\ud83d\udc2f YFG VAE Decode with Preview",
"MonoClip_node": "\ud83d\udc2f YFG Monochrome Image Clip",
"PixelArt_node": "\ud83d\udc2f YFG PixelArt",
"textMaskOverlay_node": "\ud83d\udc2f YFG Text Mask Overlay",
}
WEB_DIRECTORY = "./web"
print("------------------+-------------------")
print(f"\033[1;32m[YFG Comical]: 🐯 \033[93m\033[3m Loaded {len(NODE_CLASS_MAPPINGS)} nodes.\033[0m 🐯")
print("------------------+-------------------")