komojini-comfyui-nodes/__init__.py at main · komojini/komojini-comfyui-nodes · 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
from .nodes import *
import folder_paths
from .komojini_server import server

WEB_DIRECTORY = "js"

END_EMOJI = "🔥"

NODE_CLASS_MAPPINGS = {
    "YouTubeVideoLoader": YouTubeVideoLoader,
    "ImageMerger": ImageMerger,
    "UltimateVideoLoader": UltimateVideoLoader,
    "UltimateVideoLoader (simple)": UltimateVideoLoader,
    "KSamplerCacheable": KSamplerCacheable,
    "KSamplerAdvancedCacheable": KSamplerAdvancedCacheable,
    "Setter": To,
    "Getter": From,
    "ImageGetter": ImageGetter,
    "FlowBuilder": FlowBuilder,
    "FlowBuilder (advanced)": FlowBuilder,
    "FlowBuilder (adv)": FlowBuilder,

    "FlowBuilderSetter": FlowBuilderSetter,
    "FlowBuilder (advanced) Setter": FlowBuilderSetter,
    "FlowBuilderSetter (adv)": FlowBuilderSetter,

    "CachedGetter": CachedGetter,
    "DragNUWAImageCanvas": DragNUWAImageCanvas,

    "ImageCropByRatio": ImageCropByRatio,
    "ImageCropByRatioAndResize": ImageCropByRatioAndResize,
    "ImagesCropByRatioAndResizeBatch": ImagesCropByRatioAndResizeBatch,
    "BatchCreativeInterpolationNodeDynamicSettings": BatchCreativeInterpolationNodeDynamicSettings,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "YouTubeVideoLoader": "YouTube Video Loader",
    "ImageMerger": "Image Merger",
    "UltimateVideoLoader": "🎥Ultimate Video Loader🎥",
    "UltimateVideoLoader (simple)": "🎥Ultimate Video Loader (simple)🎥",
    "KSamplerCacheable": "KSampler (cacheable)",
    "KSamplerAdvancedCacheable": "KSamplerAdvanced (cacheable)",
    "Setter": "Setter",
    "Getter": "Getter",
    "CachedGetter": "CachedGetter",
    "ImageGetter": "ImageGetter",
    "FlowBuilder": END_EMOJI + " FlowBuilder",
    "FlowBuilder (advanced)": END_EMOJI + "(adv) FlowBuilder",
    # "FlowBuilder (adv)": END_EMOJI + "(adv) FlowBuilder",

    "FlowBuilderSetter": END_EMOJI + "FlowBuilderSetter",
    "FlowBuilder (advanced) Setter": END_EMOJI + "(adv) FlowBuilderSetter",
    # "FlowBuilderSetter (adv)": END_EMOJI + "(adv) FlowBuilder",

    "DragNUWAImageCanvas": "DragNUWAImageCanvas",
    "ImageCropByRatio": "ImageCropByRatio",
    "ImageCropByRatioAndResize": "ImageCropByRatioAndResize",
    "ImagesCropByRatioAndResizeBatch": "ImagesCropByRatioAndResizeBatch",
    "BatchCreativeInterpolationNodeDynamicSettings": "BatchCreativeInterpolationNodeDynamicSettings",
}


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