from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilitySearchAndRecolor, StabilityControlStyle, StabilityFastUpscale, StabilityReplaceRelight

NODE_CLASS_MAPPINGS = {
    "Stability Conservative Upscale": StabilityConservativeUpscale,
    "Stability Creative Upscale": StabilityCreativeUpscale,
    "Stability Remove Background": StabilityRemoveBackground,
    "Stability Erase": StabilityErase,
    "Stability Inpainting": StabilityInpainting,
    "Stability Image Core": StabilityCore,
    "Stability Image Ultra": StabilityImageUltra,
    "Stability Search and Replace": StabilitySearchAndReplace,
    "Stability Outpainting": StabilityOutpainting,
    "Stability SD3": StabilitySD3,
    "Stability Control Sketch": StabilityControlSketch,
    "Stability Control Structure": StabilityControlStructure,
    "Stability Search And Recolor": StabilitySearchAndRecolor,
    "Stability Control Style": StabilityControlStyle,
    "Stability Fast Upscale": StabilityFastUpscale,
    "Stability Replace Background and Relight": StabilityReplaceRelight,
}