ComfyUI_Text_Processor/__init__.py at main · rookiestar28/ComfyUI_Text_Processor · 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
from .advanced_text_filter import AdvancedTextFilter
from .text_input import TextInput
from .text_scraper import TextScraper
from .text_storage import NODE_CLASS_MAPPINGS as TEXT_STORAGE_CLASS_MAPPINGS
from .text_storage import NODE_DISPLAY_NAME_MAPPINGS as TEXT_STORAGE_NAME_MAPPINGS
from .wildcards import WildcardsNode
from .add_text_to_image import AddTextToImage
from .simple_eval import EvaluateInts, EvaluateFloats, EvaluateStrs

from .advanced_image_saver import AdvancedImageSaver

from .image_cropper import NODE_CLASS_MAPPINGS as IMAGE_CLASS_MAPPINGS
from .image_cropper import NODE_DISPLAY_NAME_MAPPINGS as IMAGE_DISPLAY_NAME_MAPPINGS

from .mask_nodes import NODE_CLASS_MAPPINGS as MASK_CLASS_MAPPINGS
from .mask_nodes import NODE_DISPLAY_NAME_MAPPINGS as MASK_DISPLAY_NAME_MAPPINGS

from .Image_concat_advanced import NODE_CLASS_MAPPINGS as CONCAT_CLASS_MAPPINGS
from .Image_concat_advanced import NODE_DISPLAY_NAME_MAPPINGS as CONCAT_DISPLAY_NAME_MAPPINGS

NODE_CLASS_MAPPINGS = {
    "AdvancedTextFilter": AdvancedTextFilter,
    "TextInput": TextInput,
    "TextScraper": TextScraper,
    "WildcardsNode": WildcardsNode,
    "AddTextToImage": AddTextToImage,
    "EvaluateInts": EvaluateInts,
    "EvaluateFloats": EvaluateFloats,
    "EvaluateStrs": EvaluateStrs,
    "AdvancedImageSaver": AdvancedImageSaver,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "AdvancedTextFilter": "Advanced Text Filter",
    "TextInput": "Text Input",
    "TextScraper": "Text Scraper",
    "WildcardsNode": "Wildcards Processor",
    "AddTextToImage": "Add text to image",
    "EvaluateInts": "Simple Eval Integers",
    "EvaluateFloats": "Simple Eval Floats",
    "EvaluateStrs": "Simple Eval Strings",
    "AdvancedImageSaver": "Advanced Image Saver (Aesthetic)",
}

NODE_CLASS_MAPPINGS.update(TEXT_STORAGE_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(TEXT_STORAGE_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(IMAGE_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(IMAGE_DISPLAY_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(MASK_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(MASK_DISPLAY_NAME_MAPPINGS)

NODE_CLASS_MAPPINGS.update(CONCAT_CLASS_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(CONCAT_DISPLAY_NAME_MAPPINGS)

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']