__init__.py

from __future__ import annotations

import importlib.util
from pathlib import Path


def _load_kie_nano_banana_2():
    module_path = Path(__file__).resolve().parent / "nodes" / "kie_nano_banana_2.py"
    spec = importlib.util.spec_from_file_location("comfyui_kieapi_kie_nano_banana_2", module_path)
    if spec is None or spec.loader is None:
        raise ImportError(f"Could not load node module from {module_path}")
    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    return module.KieNanoBanana2


KieNanoBanana2 = _load_kie_nano_banana_2()


NODE_CLASS_MAPPINGS = {
    "KieNanoBanana2": KieNanoBanana2,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "KieNanoBanana2": "Kie Nano Banana 2",
}