__init__.py

loaded = []
not_loaded = []
exceptions = []

try:
    from .api_nodes import NODE_CLASS_MAPPINGS as api

    loaded.append(f"api nodes ({len([*api])})")
except Exception as e:
    api = {}
    not_loaded.append("api nodes")
    exceptions.append(e)

try:
    from .simple_utilities import NODE_CLASS_MAPPINGS as simple

    loaded.append(f"simple utility nodes ({len([*simple])})")
except Exception as e:
    simple = {}
    exceptions.append(e)
    not_loaded.append("simple utility nodes")

try:
    from .cv_nodes import NODE_CLASS_MAPPINGS as cv_nodes

    loaded.append(f"CV nodes ({len([*cv_nodes])})")
except Exception as e:
    cv_nodes = {}
    not_loaded.append("CV nodes")
    exceptions.append(e)

try:
    from .color_clip_ade20k import NODE_CLASS_MAPPINGS as cc_ade20k

    loaded.append("color clip ade20k node (1)")
except Exception as e:
    cc_ade20k = {}
    exceptions.append(e)
    not_loaded.append("color clip ade20k node")

try:
    from .extension_dependant_utilities import NODE_CLASS_MAPPINGS as extended

    loaded.append(f"extension dependent nodes ({len([*extended])})")
except Exception as e:
    extended = {}
    not_loaded.append("extension dependent nodes")
    exceptions.append(e)

import nodes

NODE_CLASS_MAPPINGS = {**api, **simple, **cv_nodes, **cc_ade20k, **extended}
WEB_DIRECTORY = "js"

__all__ = ['NODE_CLASS_MAPPINGS']


print('\n\033[1mFrom Bmad Custom Nodes\033[0m')

if not hasattr(nodes, "EXTENSION_WEB_DIRS"):
    exceptions.append(f"The version of ComfyUI is outdated. Some nodes won't work as intended.")

print(f' \033[92mLoaded {len([*NODE_CLASS_MAPPINGS])} nodes:')
for m in loaded:
    print(f'  + {m}')

if len(not_loaded) > 0:
    print(f' \033[93mNot loaded:')
    for m in not_loaded:
        print(f'  * {m}')

if len(exceptions) > 0:
    print(' \033[91mProblems:')
    for e in exceptions:
        print(f'  ! {e}')

print('\033[0m')