try:
    import smea
    from smea import sample_spawner_smea, sample_spawner_smea_beta, sample_spawner_smea_dyn_beta, sample_spawner_smea_dyn_beta1, sample_spawner_rk2_smea_d_clamp, sample_spawner_smea_dyn_exp

    if smea.BACKEND == "WebUI":
        from modules import scripts, sd_samplers_common, sd_samplers
        from modules.sd_samplers_kdiffusion import sampler_extra_params, KDiffusionSampler

        class SMEA(scripts.Script):
            def title(self):
                "Spawner SMEA Samplers"

            def show(self, is_img2img):
                return False

            def __init__(self):
                if not smea.INITIALIZED:
                    samplers_smea = [
                        ("Spawner SMEA", sample_spawner_smea, ["k_spawner_smea"], {"uses_ensd": True}),
                        ("Spawner SMEA (beta)", sample_spawner_smea_beta, ["k_spawner_smea_beta"], {"uses_ensd": True}),
                        ("Spawner SMEA Dyn (beta)", sample_spawner_smea_dyn_beta, ["k_spawner_smea_dyn_beta"], {"uses_ensd": True}),
                        ("Spawner SMEA Dyn (beta1)", sample_spawner_smea_dyn_beta1, ["k_spawner_smea_dyn_beta1"], {"uses_ensd": True}),
                        ("Spawner SMEA Dyn exp", sample_spawner_smea_dyn_exp, ["k_spawner_smea_dyn_exp"], {}),
                        ('Spawner rk2 smea d clamp', sample_spawner_rk2_smea_d_clamp, ['spawner_rk2_smea_d_clamp'], {}),
                    ]
                    samplers_data_smea = [
                        sd_samplers_common.SamplerData(label, lambda model, funcname=funcname: KDiffusionSampler(funcname, model), aliases, options)
                        for label, funcname, aliases, options in samplers_smea
                        if callable(funcname)
                    ]
                    sampler_extra_params["sample_spawner_smea"] = ['s_noise', 'eta']
                    sampler_extra_params["sample_spawner_smea_beta"] = ['eta', 's_noise', 'beta']
                    sampler_extra_params["sample_spawner_smea_dyn_beta"] = ['eta_start', 'eta_end', 'eta_exponent', 's_noise', 'beta', 'sigma_max_for_dyn_eta']
                    sampler_extra_params["sample_spawner_smea_dyn_beta1"] = ['eta_start', 'eta_end', 'eta_exponent', 's_noise', 'beta', 'sigma_max_for_dyn_eta']
                    
                    sd_samplers.all_samplers.extend(samplers_data_smea)
                    sd_samplers.all_samplers_map = {x.name: x for x in sd_samplers.all_samplers}
                    sd_samplers.set_samplers()
                    smea.INITIALIZED = True

except ImportError as _:
    pass