from typing import TYPE_CHECKING

from ..utils import (
    DIFFUSERS_SLOW_IMPORT,
    OptionalDependencyNotAvailable,
    _LazyModule,
    get_objects_from_module,
    is_torch_available,
    is_transformers_available,
)


# These modules contain pipelines from multiple libraries/frameworks
_dummy_objects = {}
_import_structure = {}

try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from ..utils import dummy_pt_objects  # noqa F403

    _dummy_objects.update(get_objects_from_module(dummy_pt_objects))
else:
    _import_structure["modular_pipeline"] = [
        "ModularPipelineBlocks",
        "ModularPipeline",
        "AutoPipelineBlocks",
        "SequentialPipelineBlocks",
        "ConditionalPipelineBlocks",
        "LoopSequentialPipelineBlocks",
        "PipelineState",
        "BlockState",
    ]
    _import_structure["modular_pipeline_utils"] = [
        "ComponentSpec",
        "ConfigSpec",
        "InputParam",
        "OutputParam",
        "InsertableDict",
    ]
    _import_structure["stable_diffusion_xl"] = ["StableDiffusionXLAutoBlocks", "StableDiffusionXLModularPipeline"]
    _import_structure["stable_diffusion_3"] = ["StableDiffusion3AutoBlocks", "StableDiffusion3ModularPipeline"]
    _import_structure["wan_animate_2"] = [
        "WanAnimate2Blocks",
        "WanAnimate2DistilledBlocks",
        "WanAnimate2DistilledModularPipeline",
        "WanAnimate2ModularPipeline",
    ]
    _import_structure["wan"] = [
        "WanBlocks",
        "Wan22Blocks",
        "WanImage2VideoAutoBlocks",
        "Wan22Image2VideoBlocks",
        "WanModularPipeline",
        "Wan22ModularPipeline",
        "WanImage2VideoModularPipeline",
        "Wan22Image2VideoModularPipeline",
    ]
    _import_structure["helios"] = [
        "HeliosAutoBlocks",
        "HeliosModularPipeline",
        "HeliosPyramidAutoBlocks",
        "HeliosPyramidDistilledAutoBlocks",
        "HeliosPyramidDistilledModularPipeline",
        "HeliosPyramidModularPipeline",
    ]
    _import_structure["flux"] = [
        "FluxAutoBlocks",
        "FluxModularPipeline",
        "FluxKontextAutoBlocks",
        "FluxKontextModularPipeline",
    ]
    _import_structure["flux2"] = [
        "Flux2AutoBlocks",
        "Flux2KleinAutoBlocks",
        "Flux2KleinBaseAutoBlocks",
        "Flux2ModularPipeline",
        "Flux2KleinModularPipeline",
        "Flux2KleinBaseModularPipeline",
    ]
    _import_structure["ideogram4"] = [
        "Ideogram4AutoBlocks",
        "Ideogram4ModularPipeline",
    ]
    _import_structure["krea2"] = [
        "Krea2AutoBlocks",
        "Krea2ModularPipeline",
        "Krea2TurboAutoBlocks",
        "Krea2TurboModularPipeline",
    ]
    _import_structure["qwenimage"] = [
        "QwenImageAutoBlocks",
        "QwenImageModularPipeline",
        "QwenImageEditModularPipeline",
        "QwenImageEditAutoBlocks",
        "QwenImageEditPlusModularPipeline",
        "QwenImageEditPlusAutoBlocks",
        "QwenImageLayeredModularPipeline",
        "QwenImageLayeredAutoBlocks",
    ]
    _import_structure["anima"] = [
        "AnimaAutoBlocks",
        "AnimaModularPipeline",
    ]
    _import_structure["cosmos"] = [
        "Cosmos3DistilledBlocks",
        "Cosmos3DistilledModularPipeline",
        "Cosmos3OmniBlocks",
        "Cosmos3OmniModularPipeline",
    ]
    _import_structure["ernie_image"] = [
        "ErnieImageAutoBlocks",
        "ErnieImageModularPipeline",
    ]
    _import_structure["hunyuan_video1_5"] = [
        "HunyuanVideo15AutoBlocks",
        "HunyuanVideo15ModularPipeline",
    ]
    _import_structure["ltx"] = [
        "LTXAutoBlocks",
        "LTXModularPipeline",
    ]
    _import_structure["ltx2"] = [
        "LTX2AutoBlocks",
        "LTX25AutoBlocks",
        "LTX2ModularPipeline",
        "LTX25ModularPipeline",
    ]
    _import_structure["minimax_h3"] = [
        "MiniMaxH3Blocks",
        "MiniMaxH3ModularPipeline",
    ]
    _import_structure["minimax_music3"] = [
        "MiniMaxMusic3Blocks",
        "MiniMaxMusic3ModularPipeline",
    ]
    _import_structure["z_image"] = [
        "ZImageAutoBlocks",
        "ZImageModularPipeline",
    ]
    _import_structure["components_manager"] = ["ComponentsManager"]

if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
    try:
        if not is_torch_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from ..utils.dummy_pt_objects import *  # noqa F403
    else:
        from .anima import AnimaAutoBlocks, AnimaModularPipeline
        from .components_manager import ComponentsManager
        from .cosmos import (
            Cosmos3DistilledBlocks,
            Cosmos3DistilledModularPipeline,
            Cosmos3OmniBlocks,
            Cosmos3OmniModularPipeline,
        )
        from .ernie_image import ErnieImageAutoBlocks, ErnieImageModularPipeline
        from .flux import FluxAutoBlocks, FluxKontextAutoBlocks, FluxKontextModularPipeline, FluxModularPipeline
        from .flux2 import (
            Flux2AutoBlocks,
            Flux2KleinAutoBlocks,
            Flux2KleinBaseAutoBlocks,
            Flux2KleinBaseModularPipeline,
            Flux2KleinModularPipeline,
            Flux2ModularPipeline,
        )
        from .helios import (
            HeliosAutoBlocks,
            HeliosModularPipeline,
            HeliosPyramidAutoBlocks,
            HeliosPyramidDistilledAutoBlocks,
            HeliosPyramidDistilledModularPipeline,
            HeliosPyramidModularPipeline,
        )
        from .hunyuan_video1_5 import (
            HunyuanVideo15AutoBlocks,
            HunyuanVideo15ModularPipeline,
        )
        from .ideogram4 import (
            Ideogram4AutoBlocks,
            Ideogram4ModularPipeline,
        )
        from .krea2 import (
            Krea2AutoBlocks,
            Krea2ModularPipeline,
            Krea2TurboAutoBlocks,
            Krea2TurboModularPipeline,
        )
        from .ltx import LTXAutoBlocks, LTXModularPipeline
        from .ltx2 import LTX2AutoBlocks, LTX2ModularPipeline, LTX25AutoBlocks, LTX25ModularPipeline
        from .minimax_h3 import (
            MiniMaxH3Blocks,
            MiniMaxH3ModularPipeline,
        )
        from .minimax_music3 import (
            MiniMaxMusic3Blocks,
            MiniMaxMusic3ModularPipeline,
        )
        from .modular_pipeline import (
            AutoPipelineBlocks,
            BlockState,
            ConditionalPipelineBlocks,
            LoopSequentialPipelineBlocks,
            ModularPipeline,
            ModularPipelineBlocks,
            PipelineState,
            SequentialPipelineBlocks,
        )
        from .modular_pipeline_utils import ComponentSpec, ConfigSpec, InputParam, InsertableDict, OutputParam
        from .qwenimage import (
            QwenImageAutoBlocks,
            QwenImageEditAutoBlocks,
            QwenImageEditModularPipeline,
            QwenImageEditPlusAutoBlocks,
            QwenImageEditPlusModularPipeline,
            QwenImageLayeredAutoBlocks,
            QwenImageLayeredModularPipeline,
            QwenImageModularPipeline,
        )
        from .stable_diffusion_3 import StableDiffusion3AutoBlocks, StableDiffusion3ModularPipeline
        from .stable_diffusion_xl import StableDiffusionXLAutoBlocks, StableDiffusionXLModularPipeline
        from .wan import (
            Wan22Blocks,
            Wan22Image2VideoBlocks,
            Wan22Image2VideoModularPipeline,
            Wan22ModularPipeline,
            WanBlocks,
            WanImage2VideoAutoBlocks,
            WanImage2VideoModularPipeline,
            WanModularPipeline,
        )
        from .wan_animate_2 import (
            WanAnimate2Blocks,
            WanAnimate2DistilledBlocks,
            WanAnimate2DistilledModularPipeline,
            WanAnimate2ModularPipeline,
        )
        from .z_image import ZImageAutoBlocks, ZImageModularPipeline
else:
    import sys

    sys.modules[__name__] = _LazyModule(
        __name__,
        globals()["__file__"],
        _import_structure,
        module_spec=__spec__,
    )
    for name, value in _dummy_objects.items():
        setattr(sys.modules[__name__], name, value)