config_presets.py

import traceback
import modules.sd_samplers
import modules.scripts as scripts
import gradio as gr
import json
import os
import platform
import subprocess as sp


BASEDIR = scripts.basedir()     #C:\path\to\Stable Diffusion\extensions\Config-Presets   needs to be set in global space to get the extra 'extensions\Config-Presets' path
CONFIG_TXT2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME = "config-txt2img-custom-tracked-components.txt"
CONFIG_IMG2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME = "config-img2img-custom-tracked-components.txt"
CONFIG_TXT2IMG_FILE_NAME = "config-txt2img.json"
CONFIG_IMG2IMG_FILE_NAME = "config-img2img.json"


def load_txt2img_custom_tracked_component_ids() -> list[str]:
    txt2img_custom_tracked_components_ids = []
    try:
        with open(f"{BASEDIR}/{CONFIG_TXT2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME}", "r") as file:
            for line in file:
                line = line.strip()
                if not line.startswith("#") and line != "":  # ignore lines that start with # or are empty
                    txt2img_custom_tracked_components_ids.append(line)
                    #print(f"Added txt2img custom tracked component: {line}")

    except FileNotFoundError:
        # config file not found
        # First time running the extension or it was deleted, so fill it with default values
        txt2img_custom_tracked_components_default_text = """# Put custom txt2img tracked component IDs here. This will allow those fields to be saved as a config preset.
# Lines starting with a # are ignored.
# Component IDs can be found in the HTML (id="..."), in modules/ui.py (elem_id="..."), or in an extensions python code. IDs like "component-5890" won't work because the number at the end will change each startup.
# Entering an invalid component ID here will cause this extension to error and not load. Components that do not have a value associated with them, such as tabs and accordions, are not supported.
# Note that components on the top row of the UI cannot be added here, such as "setting_sd_model_checkpoint", "setting_sd_vae", and "setting_CLIP_stop_at_last_layers".

# Other fields:
#txt2img_prompt
#txt2img_neg_prompt
#txt2img_styles
#txt2img_subseed_show
#txt2img_subseed
#txt2img_subseed_strength
#txt2img_seed_resize_from_w
#txt2img_seed_resize_from_h
#txt2img_tiling
#txt2img_hr_resize_x
#txt2img_hr_resize_y

# Script dropdown:
#script_list

# X/Y/Z plot (script):
#script_txt2txt_xyz_plot_x_type
#script_txt2txt_xyz_plot_y_type
#script_txt2txt_xyz_plot_z_type
#script_txt2txt_xyz_plot_x_values
#script_txt2txt_xyz_plot_y_values
#script_txt2txt_xyz_plot_z_values

# Latent Couple (extension):
#cd_txt2img_divisions
#cd_txt2img_positions
#cd_txt2img_weights
#cd_txt2img_end_at_this_step

# ControlNet (extension):
#txt2img_controlnet_ControlNet-0_controlnet_enable_checkbox
#txt2img_controlnet_ControlNet-0_controlnet_low_vram_checkbox
#txt2img_controlnet_ControlNet-0_controlnet_pixel_perfect_checkbox
#txt2img_controlnet_ControlNet-0_controlnet_preprocessor_preview_checkbox
#txt2img_controlnet_ControlNet-0_controlnet_preprocessor_dropdown
#txt2img_controlnet_ControlNet-0_controlnet_model_dropdown
#txt2img_controlnet_ControlNet-0_controlnet_control_weight_slider
#txt2img_controlnet_ControlNet-0_controlnet_start_control_step_slider
#txt2img_controlnet_ControlNet-0_controlnet_ending_control_step_slider
#txt2img_controlnet_ControlNet-0_controlnet_control_mode_radio
#txt2img_controlnet_ControlNet-0_controlnet_resize_mode_radio
#txt2img_controlnet_ControlNet-0_controlnet_automatically_send_generated_images_checkbox

#txt2img_controlnet_ControlNet-1_controlnet_enable_checkbox
#txt2img_controlnet_ControlNet-1_controlnet_low_vram_checkbox
#txt2img_controlnet_ControlNet-1_controlnet_pixel_perfect_checkbox
#txt2img_controlnet_ControlNet-1_controlnet_preprocessor_preview_checkbox
#txt2img_controlnet_ControlNet-1_controlnet_preprocessor_dropdown
#txt2img_controlnet_ControlNet-1_controlnet_model_dropdown
#txt2img_controlnet_ControlNet-1_controlnet_control_weight_slider
#txt2img_controlnet_ControlNet-1_controlnet_start_control_step_slider
#txt2img_controlnet_ControlNet-1_controlnet_ending_control_step_slider
#txt2img_controlnet_ControlNet-1_controlnet_control_mode_radio
#txt2img_controlnet_ControlNet-1_controlnet_resize_mode_radio
#txt2img_controlnet_ControlNet-1_controlnet_automatically_send_generated_images_checkbox

#txt2img_controlnet_ControlNet-2_controlnet_enable_checkbox
#txt2img_controlnet_ControlNet-2_controlnet_low_vram_checkbox
#txt2img_controlnet_ControlNet-2_controlnet_pixel_perfect_checkbox
#txt2img_controlnet_ControlNet-2_controlnet_preprocessor_preview_checkbox
#txt2img_controlnet_ControlNet-2_controlnet_preprocessor_dropdown
#txt2img_controlnet_ControlNet-2_controlnet_model_dropdown
#txt2img_controlnet_ControlNet-2_controlnet_control_weight_slider
#txt2img_controlnet_ControlNet-2_controlnet_start_control_step_slider
#txt2img_controlnet_ControlNet-2_controlnet_ending_control_step_slider
#txt2img_controlnet_ControlNet-2_controlnet_control_mode_radio
#txt2img_controlnet_ControlNet-2_controlnet_resize_mode_radio
#txt2img_controlnet_ControlNet-2_controlnet_automatically_send_generated_images_checkbox

# Tiled Diffusion (extension)
#MD-t2i-enabled
#MD-t2i-overwrite-image-size
#MD-overwrite-width-t2i
#MD-overwrite-height-t2i
#MD-t2i-method
#MD-t2i-control-tensor-cpu
#MD-t2i-latent-tile-width
#MD-t2i-latent-tile-height
#MD-t2i-latent-tile-overlap
#MD-t2i-latent-tile-batch-size
# Tiled Diffusion - Region Prompt Control
#MD-t2i-enable-bbox-control
#MD-t2i-draw-background
#MD-t2i-cfg-name
# Tiled Diffusion - Region Prompt Control - Region 1
#MD-bbox-t2i-0-enable
#MD-t2i-0-blend-mode
#MD-t2i-0-feather
#MD-t2i-0-x
#MD-t2i-0-y
#MD-t2i-0-w
#MD-t2i-0-h
#MD-t2i-0-prompt
#MD-t2i-0-neg-prompt
#MD-t2i-0-seed
# Tiled Diffusion - Region Prompt Control - Region 2
#MD-bbox-t2i-1-enable
#MD-t2i-1-blend-mode
#MD-t2i-1-feather
#MD-t2i-1-x
#MD-t2i-1-y
#MD-t2i-1-w
#MD-t2i-1-h
#MD-t2i-1-prompt
#MD-t2i-1-neg-prompt
#MD-t2i-1-seed
# Tiled Diffusion - Region Prompt Control - Region 3
#MD-bbox-t2i-2-enable
#MD-t2i-2-blend-mode
#MD-t2i-2-feather
#MD-t2i-2-x
#MD-t2i-2-y
#MD-t2i-2-w
#MD-t2i-2-h
#MD-t2i-2-prompt
#MD-t2i-2-neg-prompt
#MD-t2i-2-seed
# Tiled Diffusion - Tiled VAE
#tiledvae-t2i-enable
#tiledvae-t2i-vae2gpu
#tiledvae-t2i-enc-size
#tiledvae-t2i-dec-size
#tiledvae-t2i-fastenc
#tiledvae-t2i-fastenc-colorfix
#tiledvae-t2i-fastdec

# ADetailer (extension)
#script_txt2img_adetailer_ad_enable
# ADetailer - 1st tab
#script_txt2img_adetailer_ad_model
#script_txt2img_adetailer_ad_prompt
#script_txt2img_adetailer_ad_negative_prompt
# ADetailer - 1st tab - Detection
#script_txt2img_adetailer_ad_confidence
#script_txt2img_adetailer_ad_mask_min_ratio
#script_txt2img_adetailer_ad_mask_max_ratio
# ADetailer - 1st tab - Mask Preprocessing
#script_txt2img_adetailer_ad_x_offset
#script_txt2img_adetailer_ad_y_offset
#script_txt2img_adetailer_ad_dilate_erode
#script_txt2img_adetailer_ad_mask_merge_invert
# ADetailer - 1st tab - Inpainting
#script_txt2img_adetailer_ad_mask_blur
#script_txt2img_adetailer_ad_denoising_strength
#script_txt2img_adetailer_ad_inpaint_full_res
#script_txt2img_adetailer_ad_inpaint_full_res_padding
#script_txt2img_adetailer_ad_use_inpaint_width_height
#script_txt2img_adetailer_ad_inpaint_width
#script_txt2img_adetailer_ad_inpaint_height
#script_txt2img_adetailer_ad_use_steps
#script_txt2img_adetailer_ad_steps
#script_txt2img_adetailer_ad_use_cfg_scale
#script_txt2img_adetailer_ad_cfg_scale
#script_txt2img_adetailer_ad_restore_face
# ADetailer - 1st tab - ControlNet
#script_txt2img_adetailer_ad_controlnet_model
#script_txt2img_adetailer_ad_controlnet_weight
#script_txt2img_adetailer_ad_controlnet_guidance_start
#script_txt2img_adetailer_ad_controlnet_guidance_end
# ADetailer - 2nd tab
#script_txt2img_adetailer_ad_model_2nd
#script_txt2img_adetailer_ad_prompt_2nd
#script_txt2img_adetailer_ad_negative_prompt_2nd
# ADetailer - 2nd tab - Detection
#script_txt2img_adetailer_ad_confidence_2nd
#script_txt2img_adetailer_ad_mask_min_ratio_2nd
#script_txt2img_adetailer_ad_mask_max_ratio_2nd
# ADetailer - 2nd tab - Mask Preprocessing
#script_txt2img_adetailer_ad_x_offset_2nd
#script_txt2img_adetailer_ad_y_offset_2nd
#script_txt2img_adetailer_ad_dilate_erode_2nd
#script_txt2img_adetailer_ad_mask_merge_invert_2nd
# ADetailer - 2nd tab - Inpainting
#script_txt2img_adetailer_ad_mask_blur_2nd
#script_txt2img_adetailer_ad_denoising_strength_2nd
#script_txt2img_adetailer_ad_inpaint_full_res_2nd
#script_txt2img_adetailer_ad_inpaint_full_res_padding_2nd
#script_txt2img_adetailer_ad_use_inpaint_width_height_2nd
#script_txt2img_adetailer_ad_inpaint_width_2nd
#script_txt2img_adetailer_ad_inpaint_height_2nd
#script_txt2img_adetailer_ad_use_steps_2nd
#script_txt2img_adetailer_ad_steps_2nd
#script_txt2img_adetailer_ad_use_cfg_scale_2nd
#script_txt2img_adetailer_ad_cfg_scale_2nd
#script_txt2img_adetailer_ad_restore_face_2nd
# ADetailer - 2nd tab - ControlNet
#script_txt2img_adetailer_ad_controlnet_model_2nd
#script_txt2img_adetailer_ad_controlnet_weight_2nd
#script_txt2img_adetailer_ad_controlnet_guidance_start_2nd
#script_txt2img_adetailer_ad_controlnet_guidance_end_2nd
"""

        write_text_to_file(txt2img_custom_tracked_components_default_text, CONFIG_TXT2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME)
        print(f"[Config Presets] txt2img custom tracked components config file not found, created default config at {BASEDIR}/{CONFIG_TXT2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME}")

    return txt2img_custom_tracked_components_ids



def load_img2img_custom_tracked_component_ids() -> list[str]:
    img2img_custom_tracked_components_ids = []
    try:
        with open(f"{BASEDIR}/{CONFIG_IMG2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME}", "r") as file:
            for line in file:
                line = line.strip()
                if not line.startswith("#") and line != "":  # ignore lines that start with # or are empty
                    img2img_custom_tracked_components_ids.append(line)
                    #print(f"Added img2img custom tracked component: {line}")

    except FileNotFoundError:
        # config file not found
        # First time running the extension or it was deleted, so fill it with default values
        img2img_custom_tracked_components_ids = """# Put custom img2img tracked component IDs here. This will allow those fields to be saved as a config preset.
# Lines starting with a # are ignored.
# Component IDs can be found in the HTML (id="..."), in modules/ui.py (elem_id="..."), or in an extensions python code. IDs like "component-5890" won't work because the number at the end will change each startup.
# Entering an invalid component ID here will cause this extension to error and not load. Components that do not have a value associated with them, such as tabs and accordions, are not supported.
# Note that components on the top row of the UI cannot be added here, such as "setting_sd_model_checkpoint", "setting_sd_vae", and "setting_CLIP_stop_at_last_layers".

# Other fields:
#img2img_prompt
#img2img_neg_prompt
#img2img_mask_mode
#img2img_mask_blur
#img2img_mask_alpha
#img2img_inpainting_fill
#img2img_inpaint_full_res
#img2img_inpaint_full_res_padding
#resize_mode
#img2img_scale
#img2img_seed
#img2img_subseed_show
#img2img_subseed
#img2img_subseed_strength
#img2img_seed_resize_from_w
#img2img_seed_resize_from_h
#img2img_restore_faces
#img2img_tiling
#img2img_batch_input_dir
#img2img_batch_output_dir
#img2img_batch_inpaint_mask_dir

# Script dropdown:
#script_list

# X/Y/Z plot (script):
#script_img2txt_xyz_plot_x_type
#script_img2txt_xyz_plot_y_type
#script_img2txt_xyz_plot_z_type
#script_img2txt_xyz_plot_x_values
#script_img2txt_xyz_plot_y_values
#script_img2txt_xyz_plot_z_values

# Loopback (script):
#script_loopback_loops
#script_loopback_final_denoising_strength

# SD upscale (script):
#script_sd_upscale_overlap
#script_sd_upscale_scale_factor
#script_sd_upscale_upscaler_index

# Latent Couple (extension):
#cd_img2img_divisions
#cd_img2img_positions
#cd_img2img_weights
#cd_img2img_end_at_this_step

# ControlNet (extension):
#img2img_controlnet_ControlNet-0_controlnet_enable_checkbox
#img2img_controlnet_ControlNet-0_controlnet_low_vram_checkbox
#img2img_controlnet_ControlNet-0_controlnet_pixel_perfect_checkbox
#img2img_controlnet_ControlNet-0_controlnet_preprocessor_preview_checkbox
#img2img_controlnet_ControlNet-0_controlnet_preprocessor_dropdown
#img2img_controlnet_ControlNet-0_controlnet_model_dropdown
#img2img_controlnet_ControlNet-0_controlnet_control_weight_slider
#img2img_controlnet_ControlNet-0_controlnet_start_control_step_slider
#img2img_controlnet_ControlNet-0_controlnet_ending_control_step_slider
#img2img_controlnet_ControlNet-0_controlnet_control_mode_radio
#img2img_controlnet_ControlNet-0_controlnet_resize_mode_radio
#img2img_controlnet_ControlNet-0_controlnet_automatically_send_generated_images_checkbox

#img2img_controlnet_ControlNet-1_controlnet_enable_checkbox
#img2img_controlnet_ControlNet-1_controlnet_low_vram_checkbox
#img2img_controlnet_ControlNet-1_controlnet_pixel_perfect_checkbox
#img2img_controlnet_ControlNet-1_controlnet_preprocessor_preview_checkbox
#img2img_controlnet_ControlNet-1_controlnet_preprocessor_dropdown
#img2img_controlnet_ControlNet-1_controlnet_model_dropdown
#img2img_controlnet_ControlNet-1_controlnet_control_weight_slider
#img2img_controlnet_ControlNet-1_controlnet_start_control_step_slider
#img2img_controlnet_ControlNet-1_controlnet_ending_control_step_slider
#img2img_controlnet_ControlNet-1_controlnet_control_mode_radio
#img2img_controlnet_ControlNet-1_controlnet_resize_mode_radio
#img2img_controlnet_ControlNet-1_controlnet_automatically_send_generated_images_checkbox

#img2img_controlnet_ControlNet-2_controlnet_enable_checkbox
#img2img_controlnet_ControlNet-2_controlnet_low_vram_checkbox
#img2img_controlnet_ControlNet-2_controlnet_pixel_perfect_checkbox
#img2img_controlnet_ControlNet-2_controlnet_preprocessor_preview_checkbox
#img2img_controlnet_ControlNet-2_controlnet_preprocessor_dropdown
#img2img_controlnet_ControlNet-2_controlnet_model_dropdown
#img2img_controlnet_ControlNet-2_controlnet_control_weight_slider
#img2img_controlnet_ControlNet-2_controlnet_start_control_step_slider
#img2img_controlnet_ControlNet-2_controlnet_ending_control_step_slider
#img2img_controlnet_ControlNet-2_controlnet_control_mode_radio
#img2img_controlnet_ControlNet-2_controlnet_resize_mode_radio
#img2img_controlnet_ControlNet-2_controlnet_automatically_send_generated_images_checkbox

# Tiled Diffusion (extension)
#MD-i2i-enabled
#MD-i2i-keep-input-size
#MD-i2i-method
#MD-i2i-control-tensor-cpu
#MD-i2i-latent-tile-width
#MD-i2i-latent-tile-height
#MD-i2i-latent-tile-overlap
#MD-i2i-latent-tile-batch-size
#MD-i2i-upscaler-index
#MD-i2i-upscaler-factor
# Tiled Diffusion - Noise Inversion
#MD-i2i-noise-inverse
#MD-i2i-noise-inverse-steps
#MD-i2i-noise-inverse-retouch
#MD-i2i-noise-inverse-renoise-strength
#MD-i2i-noise-inverse-renoise-kernel
# Tiled Diffusion - Region Prompt Control
#MD-i2i-enable-bbox-control
#MD-i2i-draw-background
#MD-i2i-cfg-name
# Tiled Diffusion - Region Prompt Control - Region 1
#MD-bbox-i2i-0-enable
#MD-i2i-0-blend-mode
#MD-i2i-0-feather
#MD-i2i-0-x
#MD-i2i-0-y
#MD-i2i-0-w
#MD-i2i-0-h
#MD-i2i-0-prompt
#MD-i2i-0-neg-prompt
#MD-i2i-0-seed
# Tiled Diffusion - Region Prompt Control - Region 2
#MD-bbox-i2i-1-enable
#MD-i2i-1-blend-mode
#MD-i2i-1-feather
#MD-i2i-1-x
#MD-i2i-1-y
#MD-i2i-1-w
#MD-i2i-1-h
#MD-i2i-1-prompt
#MD-i2i-1-neg-prompt
#MD-i2i-1-seed
# Tiled Diffusion - Region Prompt Control - Region 3
#MD-bbox-i2i-2-enable
#MD-i2i-2-blend-mode
#MD-i2i-2-feather
#MD-i2i-2-x
#MD-i2i-2-y
#MD-i2i-2-w
#MD-i2i-2-h
#MD-i2i-2-prompt
#MD-i2i-2-neg-prompt
#MD-i2i-2-seed
# Tiled Diffusion - Tiled VAE
#tiledvae-i2i-enable
#tiledvae-i2i-vae2gpu
#tiledvae-i2i-enc-size
#tiledvae-i2i-dec-size
#tiledvae-i2i-fastenc
#tiledvae-i2i-fastenc-colorfix
#tiledvae-i2i-fastdec

# StableSR (extension)
#SR Model does not have an ID as of June 1 2023
#StableSR-scale
#StableSR-color-fix
#StableSR-save-original
#StableSR-pure-noise

# ADetailer (extension)
#script_img2img_adetailer_ad_enable
# ADetailer - 1st tab
#script_img2img_adetailer_ad_model
#script_img2img_adetailer_ad_prompt
#script_img2img_adetailer_ad_negative_prompt
# ADetailer - 1st tab - Detection
#script_img2img_adetailer_ad_confidence
#script_img2img_adetailer_ad_mask_min_ratio
#script_img2img_adetailer_ad_mask_max_ratio
# ADetailer - 1st tab - Mask Preprocessing
#script_img2img_adetailer_ad_x_offset
#script_img2img_adetailer_ad_y_offset
#script_img2img_adetailer_ad_dilate_erode
#script_img2img_adetailer_ad_mask_merge_invert
# ADetailer - 1st tab - Inpainting
#script_img2img_adetailer_ad_mask_blur
#script_img2img_adetailer_ad_denoising_strength
#script_img2img_adetailer_ad_inpaint_full_res
#script_img2img_adetailer_ad_inpaint_full_res_padding
#script_img2img_adetailer_ad_use_inpaint_width_height
#script_img2img_adetailer_ad_inpaint_width
#script_img2img_adetailer_ad_inpaint_height
#script_img2img_adetailer_ad_use_steps
#script_img2img_adetailer_ad_steps
#script_img2img_adetailer_ad_use_cfg_scale
#script_img2img_adetailer_ad_cfg_scale
#script_img2img_adetailer_ad_restore_face
# ADetailer - 1st tab - ControlNet
#script_img2img_adetailer_ad_controlnet_model
#script_img2img_adetailer_ad_controlnet_weight
#script_img2img_adetailer_ad_controlnet_guidance_start
#script_img2img_adetailer_ad_controlnet_guidance_end
# ADetailer - 2nd tab
#script_img2img_adetailer_ad_model_2nd
#script_img2img_adetailer_ad_prompt_2nd
#script_img2img_adetailer_ad_negative_prompt_2nd
# ADetailer - 2nd tab - Detection
#script_img2img_adetailer_ad_confidence_2nd
#script_img2img_adetailer_ad_mask_min_ratio_2nd
#script_img2img_adetailer_ad_mask_max_ratio_2nd
# ADetailer - 2nd tab - Mask Preprocessing
#script_img2img_adetailer_ad_x_offset_2nd
#script_img2img_adetailer_ad_y_offset_2nd
#script_img2img_adetailer_ad_dilate_erode_2nd
#script_img2img_adetailer_ad_mask_merge_invert_2nd
# ADetailer - 2nd tab - Inpainting
#script_img2img_adetailer_ad_mask_blur_2nd
#script_img2img_adetailer_ad_denoising_strength_2nd
#script_img2img_adetailer_ad_inpaint_full_res_2nd
#script_img2img_adetailer_ad_inpaint_full_res_padding_2nd
#script_img2img_adetailer_ad_use_inpaint_width_height_2nd
#script_img2img_adetailer_ad_inpaint_width_2nd
#script_img2img_adetailer_ad_inpaint_height_2nd
#script_img2img_adetailer_ad_use_steps_2nd
#script_img2img_adetailer_ad_steps_2nd
#script_img2img_adetailer_ad_use_cfg_scale_2nd
#script_img2img_adetailer_ad_cfg_scale_2nd
#script_img2img_adetailer_ad_restore_face_2nd
# ADetailer - 2nd tab - ADetailer ControlNet
#script_img2img_adetailer_ad_controlnet_model_2nd
#script_img2img_adetailer_ad_controlnet_weight_2nd
#script_img2img_adetailer_ad_controlnet_guidance_start_2nd
#script_img2img_adetailer_ad_controlnet_guidance_end_2nd
"""

        write_text_to_file(img2img_custom_tracked_components_ids, CONFIG_IMG2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME)
        print(f"[Config Presets] img2img custom tracked components config file not found, created default config at {BASEDIR}/{CONFIG_TXT2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME}")

    return img2img_custom_tracked_components_ids


def load_txt2img_config_file():
    try:
        with open(f"{BASEDIR}/{CONFIG_TXT2IMG_FILE_NAME}") as file:
            txt2img_config_presets = json.load(file)

    except FileNotFoundError:
        # txt2img config file not found
        # First time running the extension or it was deleted, so fill it with default values
        txt2img_config_presets = {
            "None": {},
            "Low quality ------ 512x512, steps: 10, batch size: 8, DPM++ 2M Karras": {
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_steps": 10,
                "txt2img_width": 512,
                "txt2img_height": 512,
                "txt2img_enable_hr": False,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 8,
                # "txt2img_cfg_scale": 7,
            },
            "Medium quality - 512x512, steps: 15, batch size: 4, DPM++ 2M Karras": {
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_steps": 15,
                "txt2img_width": 512,
                "txt2img_height": 512,
                "txt2img_enable_hr": False,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 4,
                # "txt2img_cfg_scale": 7,
            },
            "High quality ------ 512x768, steps: 20, batch size: 4, DPM++ 2M Karras": {
                "txt2img_prompt": "masterpiece, best quality, ultra-detailed, 1girl, solo",
                "txt2img_neg_prompt": "verybadimagenegative_v1.3, ng_deepnegative_v1_75t, easynegative, badhandv4, sketch by bad-artist, bad anatomy, disfigured, deformed, extra hands, extra legs, extra arms, text, error, cropped, worst quality, low quality, jpeg artifacts, signature, watermark, username, blurry",
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_steps": 20,
                "txt2img_width": 512,
                "txt2img_height": 768,
                "txt2img_enable_hr": False,
                "txt2img_batch_count": 2,
                "txt2img_batch_size": 2,
                # "txt2img_cfg_scale": 7,
            },
            "Low quality ------ 768x768, steps: 10, batch size: 8, DPM++ 2M Karras": {
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_steps": 10,
                "txt2img_width": 768,
                "txt2img_height": 768,
                "txt2img_enable_hr": False,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 8,
                # "txt2img_cfg_scale": 7,
            },
            "Medium quality - 768x768, steps: 15, batch size: 4, DPM++ 2M Karras": {
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_steps": 15,
                "txt2img_width": 768,
                "txt2img_height": 768,
                "txt2img_enable_hr": False,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 4,
                # "txt2img_cfg_scale": 7,
            },
            "High quality ------ 768x768, steps: 20, batch size: 4, DPM++ 2S a Karras": {
                "txt2img_prompt": "masterpiece, best quality, ultra-detailed, 1girl, solo",
                "txt2img_neg_prompt": "verybadimagenegative_v1.3, ng_deepnegative_v1_75t, easynegative, badhandv4, sketch by bad-artist, bad anatomy, disfigured, deformed, extra hands, extra legs, extra arms, text, error, cropped, worst quality, low quality, jpeg artifacts, signature, watermark, username, blurry",
                "txt2img_sampling": "DPM++ 2S a Karras",
                "txt2img_steps": 20,
                "txt2img_width": 768,
                "txt2img_height": 768,
                "txt2img_enable_hr": False,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 4,
                # "txt2img_cfg_scale": 7,
            },
            "High res -------- 1024x1024, steps: 30, batch size: 1, DPM++ 2M Karras, [Upscale by: 2, Denoising: 0.4, Hires steps: 10]": {
                "txt2img_prompt": "masterpiece, best quality, ultra-detailed, 1girl, solo",
                "txt2img_neg_prompt": "verybadimagenegative_v1.3, ng_deepnegative_v1_75t, easynegative, badhandv4, sketch by bad-artist, bad anatomy, disfigured, deformed, extra hands, extra legs, extra arms, text, error, cropped, worst quality, low quality, jpeg artifacts, signature, watermark, username, blurry",
                "txt2img_steps": 30,
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_width": 512,
                "txt2img_height": 512,
                "txt2img_enable_hr": True,
                "txt2img_hr_scale": 2,
                "txt2img_hires_steps": 10,
                "txt2img_denoising_strength": 0.4,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 1,
                # "txt2img_cfg_scale": 7,
            },
            "1080p ----------- 1920x1080, steps: 20, batch size: 1, DPM++ 2M Karras, [Upscale by: 2.5, Denoising: 0.4, Hires steps: 10]": {
                # 2x 960x536, 2.5x 768x432, 3x 640x360
                "txt2img_prompt": "masterpiece, best quality, ultra-detailed, 1girl, solo",
                "txt2img_neg_prompt": "verybadimagenegative_v1.3, ng_deepnegative_v1_75t, easynegative, badhandv4, sketch by bad-artist, bad anatomy, disfigured, deformed, extra hands, extra legs, extra arms, text, error, cropped, worst quality, low quality, jpeg artifacts, signature, watermark, username, blurry",
                "txt2img_steps": 20,
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_width": 768,
                "txt2img_height": 432,
                "txt2img_enable_hr": True,
                "txt2img_hr_scale": 2.5,
                "txt2img_hires_steps": 10,
                "txt2img_denoising_strength": 0.4,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 1,
                # "txt2img_cfg_scale": 7,
            },
            "1440p ----------- 2560x1440, steps: 25, batch size: 1, DPM++ 2M Karras, [Upscale by: 3.3334, Denoising: 0.3, Hires steps: 10]": {
                # 2x 1024x720, 2.5x 1024x576, 3.3334x 768x432, 4x 640x360
                "txt2img_prompt": "masterpiece, best quality, ultra-detailed, 1girl, solo",
                "txt2img_neg_prompt": "verybadimagenegative_v1.3, ng_deepnegative_v1_75t, easynegative, badhandv4, sketch by bad-artist, bad anatomy, disfigured, deformed, extra hands, extra legs, extra arms, text, error, cropped, worst quality, low quality, jpeg artifacts, signature, watermark, username, blurry",
                "txt2img_steps": 25,
                "txt2img_sampling": "DPM++ 2M Karras",
                "txt2img_width": 768,
                "txt2img_height": 432,
                "txt2img_enable_hr": True,
                "txt2img_hr_scale": 3.3334,
                "txt2img_hires_steps": 10,
                "txt2img_denoising_strength": 0.3,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 1,
                # "txt2img_cfg_scale": 7,
            },
            "4k ---------------- 3840x2160, steps: 30, batch size: 1, DPM++ 2S a Karras, [Upscale by: 5, Denoising: 0.3, Hires steps: 15]": {
                # 2x 1420x1080, 2.5x 1536x864, 3x 1280x720, 5x 768x432, 6x 640x360
                "txt2img_prompt": "masterpiece, best quality, ultra-detailed, 1girl, solo",
                "txt2img_neg_prompt": "verybadimagenegative_v1.3, ng_deepnegative_v1_75t, easynegative, badhandv4, sketch by bad-artist, bad anatomy, disfigured, deformed, extra hands, extra legs, extra arms, text, error, cropped, worst quality, low quality, jpeg artifacts, signature, watermark, username, blurry",
                "txt2img_steps": 30,
                "txt2img_sampling": "DPM++ 2S a Karras",
                "txt2img_width": 768,
                "txt2img_height": 432,
                "txt2img_enable_hr": True,
                "txt2img_hr_scale": 5,
                "txt2img_hires_steps": 15,
                "txt2img_denoising_strength": 0.3,
                "txt2img_batch_count": 1,
                "txt2img_batch_size": 1,
                # "txt2img_cfg_scale": 7,
            },
        }

        write_json_to_file(txt2img_config_presets, CONFIG_TXT2IMG_FILE_NAME)
        print(f"[Config Presets] txt2img config file not found, created default config at {BASEDIR}/{CONFIG_TXT2IMG_FILE_NAME}")

    return txt2img_config_presets


def load_img2img_config_file():
    try:
        with open(f"{BASEDIR}/{CONFIG_IMG2IMG_FILE_NAME}") as file:
            img2img_config_presets = json.load(file)

    except FileNotFoundError:
        # img2img config file not found
        # First time running the extension or it was deleted, so fill it with default values
        img2img_config_presets = {
            "None": {},
            "Low denoising ------- 512x512, denoising: 0.25, steps: 10, DPM++ 2M Karras": {
                "img2img_sampling": "DPM++ 2M Karras",
                "img2img_steps": 10,
                "img2img_width": 512,
                "img2img_height": 512,
                # "img2img_batch_count": 1,
                # "img2img_batch_size": 1,
                # "img2img_cfg_scale": 7,
                "img2img_denoising_strength": 0.25,
            },
            "Medium denoising -- 512x512, denoising: 0.50, steps: 15, DPM++ 2M Karras": {
                "img2img_sampling": "DPM++ 2M Karras",
                "img2img_steps": 15,
                "img2img_width": 512,
                "img2img_height": 512,
                # "img2img_batch_count": 1,
                # "img2img_batch_size": 1,
                # "img2img_cfg_scale": 7,
                "img2img_denoising_strength": 0.50,
            },
            "High denoising ------- 512x512, denoising: 0.75, steps: 20, DPM++ 2M Karras": {
                "img2img_sampling": "DPM++ 2M Karras",
                "img2img_steps": 20,
                "img2img_width": 512,
                "img2img_height": 512,
                # "img2img_batch_count": 1,
                # "img2img_batch_size": 1,
                # "img2img_cfg_scale": 7,
                "img2img_denoising_strength": 0.75,
            },
        }

        write_json_to_file(img2img_config_presets, CONFIG_IMG2IMG_FILE_NAME)
        print(f"[Config Presets] img2img config file not found, created default config at {BASEDIR}/{CONFIG_IMG2IMG_FILE_NAME}")

    return img2img_config_presets


# workaround function for not being able to select new dropdown values after new choices are added to the dropdown in Gradio v3.28.1 (Automatic1111 v1.1.0)
# it's possible they will fix this in Gradio v4
# see: https://github.com/Zyin055/Config-Presets/pull/41
#def get_config_preset_dropdown_choices(new_config_presets) -> list[str]:
def get_config_preset_dropdown_choices(new_config_presets: list[str]) -> list[str]:
    new_choices = []
    if len(new_config_presets) > 0:
        # if isinstance(new_config_presets, dict):
        #     new_choices.extend(new_config_presets.keys())
        # else: # List assumed.
        #     new_choices.extend(new_config_presets)
        new_choices.extend(new_config_presets)
    return new_choices


class Script(scripts.Script):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        # Load custom tracked components
        txt2img_custom_tracked_components_ids = load_txt2img_custom_tracked_component_ids()
        img2img_custom_tracked_components_ids = load_img2img_custom_tracked_component_ids()

        # These are the settings from the UI that are saved for each preset
        self.txt2img_component_ids = [
            "txt2img_sampling",
            "txt2img_steps",
            "txt2img_width",
            "txt2img_height",
            "txt2img_batch_count",
            "txt2img_batch_size",
            "txt2img_restore_faces",
            "txt2img_enable_hr",
            "txt2img_hr_scale",
            "txt2img_hr_upscaler",
            "txt2img_hires_steps",
            "txt2img_denoising_strength",
            "txt2img_cfg_scale",
        ]
        self.txt2img_component_ids += txt2img_custom_tracked_components_ids # add the custom tracked components

        self.img2img_component_ids = [
            "img2img_sampling",
            "img2img_steps",
            "img2img_width",
            "img2img_height",
            "img2img_batch_count",
            "img2img_batch_size",
            "img2img_cfg_scale",
            "img2img_denoising_strength",
            "img2img_restore_faces",
        ]
        self.img2img_component_ids += img2img_custom_tracked_components_ids # add the custom tracked components

        # Mapping between component labels and the actual components in ui.py
        self.txt2img_component_map = {k: None for k in self.txt2img_component_ids}  # gets filled up in the after_component() method
        self.img2img_component_map = {k: None for k in self.img2img_component_ids}  # gets filled up in the after_component() method

        # Load txt2img and img2img config files
        self.txt2img_config_presets = load_txt2img_config_file()
        self.img2img_config_presets = load_img2img_config_file()



    def title(self):
        return "Config Presets"

    def show(self, is_img2img):
        return scripts.AlwaysVisible    # hide this script in the Scripts dropdown

    def after_component(self, component, **kwargs):
        # to generalize the code, detect if we are in txt2img tab or img2img tab, and then use the corresponding self variables
        # so we can use the same code for both tabs
        component_map = None
        component_ids = None
        config_file_name = None
        custom_tracked_components_config_file_name = None
        if self.is_txt2img:
            component_map = self.txt2img_component_map
            component_ids = self.txt2img_component_ids
            config_file_name = CONFIG_TXT2IMG_FILE_NAME
            custom_tracked_components_config_file_name = CONFIG_TXT2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME
        else:
            component_map = self.img2img_component_map
            component_ids = self.img2img_component_ids
            config_file_name = CONFIG_IMG2IMG_FILE_NAME
            custom_tracked_components_config_file_name = CONFIG_IMG2IMG_CUSTOM_TRACKED_COMPONENTS_FILE_NAME

        #if component.label in self.component_map:
        if component.elem_id in component_map:
            component_map[component.elem_id] = component
            #print(f"[Config-Presets][DEBUG]: found component: {component.elem_id} {component}")

        #if component.elem_id == "script_list": #bottom of the script dropdown
        #if component.elem_id == "txt2img_style2_index": #doesn't work, need to be added after all the components we edit are loaded
        #if component.elem_id == "open_folder": #bottom of the image gallery
        if component.elem_id == "txt2img_generation_info_button" or component.elem_id == "img2img_generation_info_button": #very bottom of the txt2img/img2img image gallery

            #print("Creating dropdown values...")
            #print("key/value pairs in component_map:")
            # before we create the dropdown, we need to check if each component was found successfully to prevent errors from bricking the Web UI
            for component_name, component in component_map.items():
                #print(component_name, component_type)
                if component is None:
                    log_error(f"The {'txt2img' if self.is_txt2img else 'img2img'} component '{component_name}' could not be processed. This may be because you are running an outdated version of the Config-Presets extension, or you included a component ID in the custom tracked components config file that does not exist, no longer exists (if you updated an extension), or is an invalid component (if this is the case, you need to manually edit the config file at {BASEDIR}\\{custom_tracked_components_config_file_name} or just delete it so it resets to defaults). This extension will not work until this issue is resolved.")

                    if "controlnet_control_mod_radio" in component_name:
                        # 5/26/2023 special error message for ControlNet users letting them know their config file has been automatically updated
                        # https://github.com/Mikubill/sd-webui-controlnet/commit/0d1c252cad9c37a75e839d52f9ea8207adb8aa46
                        replace_text_in_file("controlnet_control_mod_radio", "controlnet_control_mode_radio", custom_tracked_components_config_file_name)
                        log(f"'{component_name}' is from an outdated version of the ControlNet extension. Your config file has been automatically fixed to replace it with the correct ID ('control_mode_radio'). Please reload the Web UI to load the fix.")

                    return

            # Mark components with type "index" to be transform
            index_type_components = []
            for component in component_map.values():
                #print(component)
                if getattr(component, "type", "No type attr") == "index":
                    # print(component.elem_id)
                    index_type_components.append(component.elem_id)

            preset_values = []
            config_presets: dict[str, any] = None
            if self.is_txt2img:
                config_presets = self.txt2img_config_presets
            else:
                config_presets = self.img2img_config_presets

            preset_values: list[str] = list(config_presets.keys())
            # for dropdownValue in config_presets:
            #     preset_values.append(dropdownValue)
            #     #print(f"Config Presets: added \"{dropdownValue}\"")

            fields_checkboxgroup = gr.CheckboxGroup(choices=component_ids,
                                                    value=component_ids,    #check all checkboxes by default
                                                    label="Fields to save",
                                                    show_label=True,
                                                    interactive=True,
                                                    elem_id="script_config_preset_fields_to_save",
                                                    ).unrender() #we need to define this early on so that it can be used as an input for another function

            with gr.Column(min_width=600, elem_id="config_preset_wrapper_txt2img" if self.is_txt2img else "config_preset_wrapper_img2img"):  # pushes our stuff onto a new row at 1080p screen resolution
                with gr.Row():
                    with gr.Column(scale=8, min_width=100) as dropdown_column:


                        def config_preset_dropdown_change(dropdown_value, *components_value):
                            config_preset = config_presets[dropdown_value]
                            print(f"[Config-Presets] Changed to: {dropdown_value}")

                            # update component values with user preset
                            current_components = dict(zip(component_map.keys(), components_value))
                            #print("Components before:", current_components)
                            current_components.update(config_preset)

                            # transform necessary components from index to value
                            for component_name, component_value in current_components.items():
                                #print(component_name, component_value)
                                if component_name in index_type_components and type(component_value) == int:
                                        current_components[component_name] = component_map[component_name].choices[component_value]

                            #print("Components after :", current_components)
                            
                            return list(current_components.values())

                        config_preset_dropdown = gr.Dropdown(
                            label="Config Presets",
                            #choices=preset_values,
                            choices=get_config_preset_dropdown_choices(preset_values),
                            elem_id="config_preset_txt2img_dropdown" if self.is_txt2img else "config_preset_img2img_dropdown",
                        )
                        config_preset_dropdown.style(container=False) #set to True to give it a white box to sit in

                        #self.txt2img_config_preset_dropdown = config_preset_dropdown

                        try:
                            components = list(component_map.values())
                            config_preset_dropdown.change(
                                fn=config_preset_dropdown_change,
                                show_progress=False,
                                inputs=[config_preset_dropdown, *components],
                                outputs=components
                                )
                        except AttributeError:
                            print(traceback.format_exc())   # prints the exception stacktrace
                            log_critical_error("The Config-Presets extension encountered a fatal error. A component required by this extension no longer exists in the Web UI. This is most likely due to the A1111 Web UI being updated. Try updating the Config-Presets extension. If that doesn't work, please post a bug report at https://github.com/Zyin055/Config-Presets/issues and delete your extensions/Config-Presets folder until an update is published.")

                        # No longer needed after the bump to Gradio 3.23
                        # config_preset_dropdown.change(
                        #     fn=None,
                        #     inputs=[],
                        #     outputs=[],
                        #     _js="function() { config_preset_dropdown_change() }",   # JS is used to update the Hires fix row to show/hide it
                        # )
                    with gr.Column(scale=8, min_width=100, visible=False) as collapsable_column:
                        with gr.Row():
                            with gr.Column(scale=1, min_width=10):

                                def delete_selected_preset(config_preset_name):
                                    if config_preset_name in config_presets.keys():
                                        del config_presets[config_preset_name]
                                        print(f'[Config-Presets] deleted: "{config_preset_name}"')

                                        write_json_to_file(config_presets, config_file_name)

                                        preset_keys = list(config_presets.keys())
                                        return gr.Dropdown.update(value=preset_keys[len(preset_keys)-1],
                                                                  #choices=preset_values,
                                                                  choices=get_config_preset_dropdown_choices(preset_keys),
                                                                  )
                                    return gr.Dropdown.update() # do nothing if no value is selected

                                trash_button = gr.Button(
                                    value="\U0001f5d1\ufe0f", #🗑
                                    elem_id="script_config_preset_trash_button",
                                )
                                trash_button.click(
                                    fn=delete_selected_preset,
                                    inputs=[config_preset_dropdown],
                                    outputs=[config_preset_dropdown],
                                )

                            with gr.Column(scale=2, min_width=190):
                                def open_file(f):
                                    path = os.path.normpath(f)

                                    if not os.path.exists(path):
                                        print(f'Config Presets: The file at "{path}" does not exist.')
                                        return

                                    # copied from ui.py:538
                                    if platform.system() == "Windows":
                                        os.startfile(path)
                                    elif platform.system() == "Darwin":
                                        sp.Popen(["open", path])
                                    else:
                                        sp.Popen(["xdg-open", path])

                                open_config_file_button = gr.Button(
                                    value="📂 Open config file...",
                                    elem_id="script_config_preset_open_config_file_button",
                                )
                                open_config_file_button.click(
                                    fn=lambda: open_file(f"{BASEDIR}/{config_file_name}"),
                                    inputs=[],
                                    outputs=[],
                                )

                            with gr.Column(scale=2, min_width=50):
                                cancel_button = gr.Button(
                                    value="Cancel",
                                    elem_id="script_config_preset_cancel_save_button",
                                )

                    with gr.Column(scale=1, min_width=120, visible=True) as add_remove_button_column:
                        add_remove_button = gr.Button(
                            value="Add/Remove...",
                            elem_id="script_config_preset_add_button",
                        )

                with gr.Row() as collapsable_row:
                    collapsable_row.visible = False
                    with gr.Column():
                        with gr.Row():
                            with gr.Column(scale=10, min_width=100):
                                save_textbox = gr.Textbox(
                                    label="New preset name",
                                    placeholder="Ex: Low quality",
                                    # value="My Preset",
                                    max_lines=1,
                                    elem_id="script_config_preset_save_textbox",
                                )
                            with gr.Column(scale=2, min_width=200):
                                save_button = gr.Button(
                                    # value="Create",
                                    value="💾 Save",
                                    variant="primary",
                                    elem_id="script_config_preset_save_button",
                                )

                                save_button.click(
                                    fn=save_config(config_presets, component_map, config_file_name),
                                    inputs=list(
                                        [save_textbox] + [fields_checkboxgroup] + [component_map[comp_name] for comp_name in
                                                                                   component_ids if
                                                                                   component_map[comp_name] is not None]),
                                    outputs=[config_preset_dropdown, save_textbox],
                                )

                                def add_remove_button_click(save_textbox_text: str, config_preset_dropdown_value: str):
                                    if save_textbox_text == "" or save_textbox_text is None:
                                        if config_preset_dropdown_value != "" and config_preset_dropdown_value is not None:
                                            # save textbox is empty, and we have a dropdown value selected
                                            # auto-populate the save textbox so it's easier to overwrite existing config preset
                                            return gr.Textbox.update(value=config_preset_dropdown_value)
                                    return gr.Textbox.update()


                                def expand_edit_ui():
                                    return gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)

                                def collapse_edit_ui():
                                    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)

                                add_remove_button.click(
                                    fn=add_remove_button_click,
                                    inputs=[save_textbox, config_preset_dropdown],
                                    outputs=[save_textbox],
                                )
                                add_remove_button.click(
                                    fn=expand_edit_ui,
                                    inputs=[],
                                    outputs=[collapsable_column, collapsable_row, add_remove_button_column],
                                )

                                cancel_button.click(
                                    fn=collapse_edit_ui,
                                    inputs=[],
                                    outputs=[collapsable_column, collapsable_row, add_remove_button_column],
                                )

                        with gr.Row():
                            fields_checkboxgroup.render()

                        with gr.Row():
                            with gr.Column(scale=1):
                                open_custom_tracked_components_config_file_button = gr.Button(
                                    value="📂 Add custom fields...",
                                    elem_id="script_config_preset_open_custom_tracked_components_config",
                                )
                                open_custom_tracked_components_config_file_button.click(
                                    fn=lambda: open_file(f"{BASEDIR}/{custom_tracked_components_config_file_name}"),
                                    inputs=[],
                                    outputs=[],
                                )
                            with gr.Column(scale=2):
                                pass


    def ui(self, is_img2img):
        pass

    def run(self, p, *args):
        pass


# Save the current values on the UI to a new entry in the config file
def save_config(config_presets, component_map, config_file_name):
    #print("save_config()")
    # closure keeps path in memory, it's a hack to get around how click or change expects values to be formatted
    def func(new_setting_name, fields_to_save_list, *new_setting):
        #print(f"save_config() func() new_setting_name={new_setting_name} *new_setting={new_setting}")
        #print(f"config_presets()={config_presets}")
        #print(f"component_map()={component_map}")
        #print(f"config_file_name()={config_file_name}")

        if new_setting_name == "":
            return gr.Dropdown.update(), "" # do nothing if no label entered in textbox

        new_setting_map = {}    # dict[str, Any]    {"txt2img_steps": 10, ...}

        #print(f"component_map={component_map}")
        #print(f"new_setting={new_setting}")

        for i, component_id in enumerate(component_map.keys()):

            if component_id not in fields_to_save_list:
                #print(f"[Config-Presets] New preset '{new_setting_name}' will not include {component_id}")
                continue

            if component_map[component_id] is not None:
                new_value = new_setting[i]  # this gives the index when the component is a dropdown
                if component_id == "txt2img_sampling":
                    new_setting_map[component_id] = modules.sd_samplers.samplers[new_value].name
                elif component_id == "img2img_sampling":
                    new_setting_map[component_id] = modules.sd_samplers.samplers_for_img2img[new_value].name
                else:
                    new_setting_map[component_id] = new_value

                #print(f"Saving '{component_id}' as: {new_setting_map[component_id]} ({new_value})")

        #print(f"new_setting_map = {new_setting_map}")

        config_presets.update({new_setting_name: new_setting_map})
        write_json_to_file(config_presets, config_file_name)

        # print(f"self.txt2img_config_preset_dropdown.choices before =\n{self.txt2img_config_preset_dropdown.choices}")
        # self.txt2img_config_preset_dropdown.choices = list(config_presets.keys())
        # print(f"self.txt2img_config_preset_dropdown.choices after =\n{self.txt2img_config_preset_dropdown.choices}")

        print(f"[Config-Presets] Added new preset: {new_setting_name}")
        #print(f"[Config-Presets] Restarting UI...") # done in _js
        return gr.Dropdown.update(value=new_setting_name,   # update the dropdown with the new config preset
                                  #choices=list(config_presets.keys()),
                                  choices=get_config_preset_dropdown_choices(config_presets.keys()),
                                  ), "" # clear the 'New preset name' textbox

    return func


def write_json_to_file(json_data, file_name: str):
    with open(f"{BASEDIR}/{file_name}", "w") as file:
        file.write(json.dumps(json_data, indent=4))


def write_text_to_file(text, file_name: str):
    with open(f"{BASEDIR}/{file_name}", "w") as file:
        file.write(text)


def replace_text_in_file(old: str, new: str, file_name: str):
    with open(f"{BASEDIR}/{file_name}", "r") as file:
        content = file.read()

    with open(f"{BASEDIR}/{file_name}", "w") as file:
        file.write(content.replace(old, new))


def log(text: str):
    print(f"[Config Presets] {text}")


def log_error(text: str):
    print(f"[ERROR][Config Presets] {text}")


def log_critical_error(text: str):
    print(f"[ERROR][CRITICAL][Config Presets] {text}")