"""ComfyUI-AgeShift: face age detection + age-conditioned prompt building."""

from .nodes.detector_loader import AgeDetectorLoader
from .nodes.detect import AgeDetect
from .nodes.prompt_builder import AgePromptBuilder, AgeRangeBatch

NODE_CLASS_MAPPINGS = {
    "AgeDetectorLoader": AgeDetectorLoader,
    "AgeDetect": AgeDetect,
    "AgePromptBuilder": AgePromptBuilder,
    "AgeRangeBatch": AgeRangeBatch,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "AgeDetectorLoader": "🎂 Age Detector Loader",
    "AgeDetect": "🎂 Age Detect",
    "AgePromptBuilder": "🎂 Age Prompt Builder",
    "AgeRangeBatch": "🎂 Age Range Batch",
}

WEB_DIRECTORY = None
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]