ComfyUI-ExternalAPI-Helpers/__init__.py at main · Aryan185/ComfyUI-ExternalAPI-Helpers · 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
from dotenv import load_dotenv
load_dotenv()

import importlib

modules = [
    "flux_kontext_replicate",
    "gemini_node",
    "gemini_diarisation",
    "gpt_image_edit",
    "imagen",
    "imagen_edit",
    "veo",
    "veo_api",
    "gemini_segment",
    "nano_banana",
    "gemini_tts",
    "elevenlabs_tts",
    "flux2_replicate",
    "openai_tts",
    "openai_node",
    "sora",
    "cleanup",
    "tripoTextToModel",
    "tripoImageToModel",
    "grok",
    "groq_node",
    "groq_orpheus",
]

NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}

for module_name in modules:
    module = importlib.import_module(f".{module_name}", package=__name__)
    NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
    NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]