from .background import ProcessImageBackground, RemoveImageBackground
from .balloon import (
    ActivateBalloonTool,
    ExtractBalloon,
)
from .debug import DebugBen2, DebugDa3, DebugMoge2
from .depth_plane import (
    ConvertToDepthPlane,
    ExtractOriginalFrames,
    GenerateDa3Depth,
    GenerateMoge2Depth,
    GenerateOriginalBackground,
)
from .models import ClearModels, DownloadModel, DownloadRequiredModels
from .surface import ConvertToSurface, GenerateDa3Surface, GenerateMoge2Surface


CLASSES = (
    DownloadModel,
    DownloadRequiredModels,
    DebugDa3,
    DebugMoge2,
    DebugBen2,
    GenerateDa3Depth,
    GenerateMoge2Depth,
    GenerateDa3Surface,
    GenerateMoge2Surface,
    ProcessImageBackground,
    GenerateOriginalBackground,
    ExtractOriginalFrames,
    ClearModels,
    ExtractBalloon,
    ActivateBalloonTool,
    RemoveImageBackground,
    ConvertToDepthPlane,
    ConvertToSurface,
)