from .remove_background import RemoveImageBackground, RunBackgroundRemoval
from .cutout_tool import (
    CutoutSelectionToShape,
    SelectCutoutSelection,
)
from .image_edit_tool import CLASSES as IMAGE_EDIT_TOOL_CLASSES
from .convert_to_plane import (
    ConvertToDepthPlane,
    ConvertToPlane,
    GenerateDepthPlane,
)
from .ai_setup import (
    ClearModels,
    DownloadModel,
    DownloadRequiredModels,
    OpenAIEnvironmentSettings,
    SetupAIEnvironment,
)
from .upscale import UpscaleImage


CLASSES = (
    SetupAIEnvironment,
    OpenAIEnvironmentSettings,
    DownloadModel,
    DownloadRequiredModels,
    GenerateDepthPlane,
    CutoutSelectionToShape,
    SelectCutoutSelection,
    RunBackgroundRemoval,
    ClearModels,
    *IMAGE_EDIT_TOOL_CLASSES,
    RemoveImageBackground,
    UpscaleImage,
    ConvertToPlane,
    ConvertToDepthPlane,
)