"""ComfyUI-Prompt-Calibratror — VLM-as-judge prompt calibration loop."""

from .nodes.qwen_judge import (
    NODE_CLASS_MAPPINGS as _JUDGE_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _JUDGE_NAMES,
)
from .nodes.receptor import (
    NODE_CLASS_MAPPINGS as _RECEPTOR_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _RECEPTOR_NAMES,
)
from .nodes.audio_guide import (
    NODE_CLASS_MAPPINGS as _AUDIO_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _AUDIO_NAMES,
)
from .nodes.audio_wave_segments import (
    NODE_CLASS_MAPPINGS as _WAVE_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _WAVE_NAMES,
)

NODE_CLASS_MAPPINGS = {**_JUDGE_CLASSES, **_RECEPTOR_CLASSES, **_AUDIO_CLASSES, **_WAVE_CLASSES}
NODE_DISPLAY_NAME_MAPPINGS = {**_JUDGE_NAMES, **_RECEPTOR_NAMES, **_AUDIO_NAMES, **_WAVE_NAMES}

WEB_DIRECTORY = "./web"  # serves web/audio_wave.js (the waveform + segment widget)
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]