"""ComfyUI Krea Reference custom nodes.

This package registers the Krea 2 V9 and V10 reference-stack nodes and
the Concept Slider V1 nodes.
"""

from .kg_krea_v9 import KGKrea2ImageGuideCardV9, KGTextEncodeKreaImageReferencesV9
from .kg_krea_v10 import KGKrea2ImageGuideCardV10, KGTextEncodeKreaImageReferencesV10
from .kg_krea_slider import KGKrea2ConceptSliderCardV1, KGKrea2ConceptSliderStackV1

WEB_DIRECTORY = "./web"

NODE_CLASS_MAPPINGS = {
    "KGKrea2ImageGuideCardV9": KGKrea2ImageGuideCardV9,
    "KGTextEncodeKreaImageReferencesV9": KGTextEncodeKreaImageReferencesV9,
    "KGKrea2ImageGuideCardV10": KGKrea2ImageGuideCardV10,
    "KGTextEncodeKreaImageReferencesV10": KGTextEncodeKreaImageReferencesV10,
    "KGKrea2ConceptSliderCardV1": KGKrea2ConceptSliderCardV1,
    "KGKrea2ConceptSliderStackV1": KGKrea2ConceptSliderStackV1,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "KGKrea2ImageGuideCardV9": "KG Krea 2 Image Guide Card V9",
    "KGTextEncodeKreaImageReferencesV9": "KG Krea 2 Reference Stack Encoder V9",
    "KGKrea2ImageGuideCardV10": "KG Krea 2 Image Guide Card V10",
    "KGTextEncodeKreaImageReferencesV10": "KG Krea 2 Reference Stack Encoder V10",
    "KGKrea2ConceptSliderCardV1": "KG Krea 2 Concept Slider Card V1",
    "KGKrea2ConceptSliderStackV1": "KG Krea 2 Concept Slider Stack V1",
}

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

from .kg_krea_v11 import KGKrea2ImageGuideCardV11, KGTextEncodeKreaImageReferencesV11, KGKrea2ConceptSliderCardV11, KGKrea2ConceptSliderStackV11

NODE_CLASS_MAPPINGS.update({
    "KGKrea2ImageGuideCardV11": KGKrea2ImageGuideCardV11,
    "KGTextEncodeKreaImageReferencesV11": KGTextEncodeKreaImageReferencesV11,
    "KGKrea2ConceptSliderCardV11": KGKrea2ConceptSliderCardV11,
    "KGKrea2ConceptSliderStackV11": KGKrea2ConceptSliderStackV11,
})
NODE_DISPLAY_NAME_MAPPINGS.update({
    "KGKrea2ImageGuideCardV11": "KG Krea 2 Image Guide Card V11",
    "KGTextEncodeKreaImageReferencesV11": "KG Krea 2 Reference Stack Encoder V11",
    "KGKrea2ConceptSliderCardV11": "KG Krea 2 Concept Slider Card V11",
    "KGKrea2ConceptSliderStackV11": "KG Krea 2 Concept Slider Stack V11",
})

from .kg_krea_v12 import (
    KGKrea2PrepareNativeContextV12, KGKrea2NativeAxisV12, KGKrea2CombinedNativeStudiesV12,
    KGKrea2NativePhaseGuiderV12, KGKrea2PrepareOwnedGroundedContextV12, KGKrea2RemoveOwnedReferenceV12,
    KGKrea2NativeEnhancerV12, KGKrea2PrepareNativeTargetV12,
    KGKrea2BoundedNativeDiagnosticV12, KGKrea2NativeDiagnosticReceiptV12,
)

NODE_CLASS_MAPPINGS.update({
    "KGKrea2PrepareNativeContextV12": KGKrea2PrepareNativeContextV12,
    "KGKrea2NativeAxisV12": KGKrea2NativeAxisV12,
    "KGKrea2CombinedNativeStudiesV12": KGKrea2CombinedNativeStudiesV12,
    "KGKrea2NativePhaseGuiderV12": KGKrea2NativePhaseGuiderV12,
    "KGKrea2PrepareOwnedGroundedContextV12": KGKrea2PrepareOwnedGroundedContextV12,
    "KGKrea2RemoveOwnedReferenceV12": KGKrea2RemoveOwnedReferenceV12,
    "KGKrea2NativeEnhancerV12": KGKrea2NativeEnhancerV12,
    "KGKrea2PrepareNativeTargetV12": KGKrea2PrepareNativeTargetV12,
    "KGKrea2BoundedNativeDiagnosticV12": KGKrea2BoundedNativeDiagnosticV12,
    "KGKrea2NativeDiagnosticReceiptV12": KGKrea2NativeDiagnosticReceiptV12,
})
NODE_DISPLAY_NAME_MAPPINGS.update({
    "KGKrea2PrepareNativeContextV12": "KG Krea 2 Prepare Native Context V12",
    "KGKrea2NativeAxisV12": "KG Krea 2 Native Axis V12",
    "KGKrea2CombinedNativeStudiesV12": "KG Krea 2 Combined Native Studies V12",
    "KGKrea2NativePhaseGuiderV12": "KG Krea 2 Native Phase Guider V12",
    "KGKrea2PrepareOwnedGroundedContextV12": "KG Krea 2 Prepare Owned Grounded Context V12",
    "KGKrea2RemoveOwnedReferenceV12": "KG Krea 2 Remove Owned Reference V12",
    "KGKrea2NativeEnhancerV12": "KG Krea 2 Native Enhancer V12",
    "KGKrea2PrepareNativeTargetV12": "KG Krea 2 Prepare Native Target V12",
    "KGKrea2BoundedNativeDiagnosticV12": "KG Krea 2 Bounded Native Diagnostic V12",
    "KGKrea2NativeDiagnosticReceiptV12": "KG Krea 2 Native Diagnostic Receipt V12",
})

from .kg_krea_v13 import KGKrea2NativeCarrierStudiesV13, KGKrea2NativeCarrierPhaseGuiderV13

NODE_CLASS_MAPPINGS.update({
    "KGKrea2NativeCarrierStudiesV13": KGKrea2NativeCarrierStudiesV13,
    "KGKrea2NativeCarrierPhaseGuiderV13": KGKrea2NativeCarrierPhaseGuiderV13,
})
NODE_DISPLAY_NAME_MAPPINGS.update({
    "KGKrea2NativeCarrierStudiesV13": "KG Krea 2 Native Carrier Studies V13 (Experimental)",
    "KGKrea2NativeCarrierPhaseGuiderV13": "KG Krea 2 Native Carrier Phase Guider V13 (Experimental)",
})