import os

import folder_paths

from .HumanParserLIPCustomNode import HumanParserLIPCustomNode
from .HumanParserATRCustomNode import HumanParserATRCustomNode
from .HumanParserPascalCustomNode import HumanParserPascalCustomNode

# Register the SCHP model folder with ComfyUI so folder_paths can resolve it.
folder_paths.add_model_folder_path("schp", os.path.join(folder_paths.models_dir, "schp"))

NODE_CLASS_MAPPINGS = {
  "Cozy Human Parser LIP" : HumanParserLIPCustomNode,
  "Cozy Human Parser ATR" : HumanParserATRCustomNode,
  "Cozy Human Parser Pascal" : HumanParserPascalCustomNode,
}

__all__ = ['NODE_CLASS_MAPPINGS']