mirror of
https://github.com/Zyin055/Config-Presets.git
synced 2026-01-26 19:19:54 +00:00
809 lines
40 KiB
Python
809 lines
40 KiB
Python
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_mod_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_mod_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_mod_radio
|
|
#txt2img_controlnet_ControlNet-2_controlnet_resize_mode_radio
|
|
#txt2img_controlnet_ControlNet-2_controlnet_automatically_send_generated_images_checkbox
|
|
"""
|
|
|
|
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_mod_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_mod_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_mod_radio
|
|
#img2img_controlnet_ControlNet-2_controlnet_resize_mode_radio
|
|
#img2img_controlnet_ControlNet-2_controlnet_automatically_send_generated_images_checkbox
|
|
"""
|
|
|
|
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 ------ 512x512, steps: 20, batch size: 4, DPM++ 2S a Karras": {
|
|
"txt2img_sampling": "DPM++ 2S a Karras",
|
|
"txt2img_steps": 20,
|
|
"txt2img_width": 512,
|
|
"txt2img_height": 512,
|
|
"txt2img_enable_hr": False,
|
|
"txt2img_batch_count": 1,
|
|
"txt2img_batch_size": 4,
|
|
# "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_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_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++ 2S a Karras, [Upscale by: 2.5, Denoising: 0.4, Hires steps: 10]": {
|
|
# 2x 960x536, 2.5x 768x432, 3x 640x360
|
|
"txt2img_steps": 20,
|
|
"txt2img_sampling": "DPM++ 2S a 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++ 2S a Karras, [Upscale by: 3.3334, Denoising: 0.3, Hires steps: 10]": {
|
|
# 2x 1024x720, 2.5x 1024x576, 3.3334x 768x432, 4x 640x360
|
|
"txt2img_steps": 25,
|
|
"txt2img_sampling": "DPM++ 2S a 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_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:
|
|
print(f"[ERROR][Config-Presets] The 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 or is an invalid component. This extension will not work until this issue is resolved.")
|
|
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
|
|
print("[ERROR][CRITICAL][Config-Presets] 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():
|
|
return gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
|
|
|
|
# def save_button_click(save_text):
|
|
# if save_text == "":
|
|
# return gr.update(), gr.update()
|
|
# return gr.update(visible=True), gr.update(visible=False)
|
|
|
|
def cancel_button_click():
|
|
return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
|
|
|
|
add_remove_button.click(
|
|
fn=add_remove_button_click,
|
|
inputs=[],
|
|
outputs=[collapsable_column, collapsable_row, add_remove_button_column],
|
|
)
|
|
# save_button.click(
|
|
# fn=save_button_click,
|
|
# inputs=[save_textbox],
|
|
# outputs=[add_remove_button_column, collapsable_column],
|
|
# )
|
|
cancel_button.click(
|
|
fn=cancel_button_click,
|
|
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)
|