spaghetti_ai_script.py

import gradio as gr
import torch
import modin.pandas as pd
from PIL import Image
from diffusers import DiffusionPipeline
import os
import random
import torchsde
from math import floor, copysign
import time

##########

#
#    Spaghetti AI by MagicFixesEverything
#        https://huggingface.co/magicfixeseverything
#
#    This app is based on scripts by:
#        https://huggingface.co/Manjushri
#    This app has been adapted from that person's versions.
#

##########

#
#    For instructions on how to install this app offline, see the instructions
#    here:
#        https://huggingface.co/spaces/magicfixeseverything/ai_image_creation/blob/main/Instructions.txt
#
#    The display was tested with gradio version 4.11.0.
#
#    To launch this script, use the following in the command prompt, taking off
#    the # at the start. (You will need to adjust the start of the path if you
#    have changed the location)
#
#cd C:\Diffusers && .venv\Scripts\activate.bat && py .venv\ai_image_creation\app.py
#
#    You must have a NVIDIA graphics card in your computer with Cuda installed
#    to use this script. It will not work on just a CPU on Windows.
#
#    If not using "enable_model_cpu_offload" or "enable_sequential_cpu_offload",
#    memory usage will remain high until command prompt is closed. (whether
#    image is being created or not)
#
###############################################################################
###############################################################################
#
#
#
#    Begin Configurations
#
#
#
###############################################################################
###############################################################################

#
#    Main Directory
#
#        This is where everything goes. Your Python virtual environment should
#        be here. Model data will be stored here. (unless you change the next
#        configuration) If configured, imagery will also be automatically be
#        saved here.
#

main_dir = "C:/Spaghetti_AI"

####################

#
#    Only Use Local Files (IMPORTANT)
#
#        Please read the section below the configuration. The app will not work
#        unless this is set to "0" in order for it to download model data.
#
#        0    False (Model data will be downloaded)
#        1    True (Model data will not be downloaded, even if missing)
#

only_use_local_files = 0

#
#        This is an important value. HuggingFace doesn't just download a model
#        once and never try again. If something is updated, it will download it
#        again. It will not delete the older version. This could eventually
#        allow it to use all the space on your drive as that could potentially
#        add 5 to 15 gigabytes of data each time a model updates. This variable
#        forces the script not to check for model data. However, when set to
#        "1", the app will not download any data for the models, including for
#        refining and upscaling. That would mean the app will not work. You need
#        to set this to "0" when you need to download data. While there are ways
#        to not download as much data, it is not as simple. I prefer this
#        method. You could check occasionally and manually delete model data,
#        but if you're like me, you would forget.
#
#        To download data you can use each model in the app when this is set to
#        "0" and it will download the needed data. That does require you to use
#        each model. Then you could set this back to "1". However, if you want
#        to download all the data, without having to use each model, including
#        using refining and upscaling at some point at least once, there are
#        two options.
#
#        To download all default model data, meaning the default model selected
#        for each base model (set in
#        "base_model_model_configuration_defaults_object"), including data for
#        the refiner and upscaler, add this to the end of the url for option 1:
#
#            ?download_data=1
#
#        Like this for option 1:
#            http://127.0.0.1:7860/?download_data=1
#
#        To download all model data, for everything in the
#        "model_configuration_links_object", including data for the refiner and
#        upscaler, add this to the end of the url for option 2:
#
#            ?download_data=2
#
#        Both options will download dozens of gigabytes of data, most especially
#        the second option, so you may not want to do that. Before you do that,
#        make sure you have removed from the configurations the models you do
#        not want. For option 1, remove the base models you do not want to use
#        in "base_model_array". For option 2, remove the model configurations
#        you do not want to use in
#        "base_model_object_of_model_configuration_arrays".
#
#        To have model data download, this variable must be set to 0. You must
#        also set "HF_HUB_OFFLINE" to "0" in "spaghetti_ai_launcher.bat" if you
#        use that script. If you use that script, and either is not that, model
#        data will not download.
#

####################

#
#    Use Custom Hugging Face Cache Directory
#
#        The folder where model data is stored can get huge. I choose to add it
#        to a place where I am more likely to notice it more often. If you use
#        other Hugging Face things however, and will use these models in those
#        other things, then you might want to consider not having this here as
#        it would duplicate the model data.
#
#        If set to 1, the data would be here:
#            C:\Diffusers\model_data
#
#        If set to 0, the data would be here:
#            %USERPROFILE%/.cache/huggingface/hub
#            Which would look like this, where {Username} is the username of
#            your Windows account:
#            C:\Users\{Username}\.cache\huggingface\hub
#
#        You need to clean out the folder occasionally as this folder will get
#        extremely large. Eventually, it would take up all the space on your
#        computer.
#

use_custom_hugging_face_cache_dir = 1

#####

#
#    Name of Model Data Folder
#
#        This is where all the model data will go. (unless you changed it in the
#        previous configuration) This folder will get very large. You need to
#        clean it out manually occasionally.
#

cache_directory_folder_name = "model_data"

####################

#
#    Default Base Model
#
#        This will automatically be SDXL Turbo if you are running this on a CPU.
#

default_base_model = "sdxl"

####################

#
#    Use Safety Checker
#
#        This can block some NSFW images for the models that allow it.
#
#        0    No
#        1    Yes
#

use_safety_checker = 0

#####

#
#    Auto Save Imagery
#
#        You can automatically save the image file, and a text file with the
#        prompt details.
#

auto_save_imagery = 1

#####

#
#        Name of Saved Images Folder
#
#            You can change the name of this folder if you want. Imagery will be
#            saved in a folder called "saved_images" in the directory configured
#            in "main_dir". (the saved images folder will be created
#            automatically) A directory for each day will be created in this
#            folder. Imagery will then be placed in each folder.
#

saved_images_folder_name = "saved_images"

####################

#
#    Auto Open Browser From Command Prompt
#

auto_open_browser = 0

####################

#
#    Allow Image Generation Cancellation
#
#        This allows canceling the image generation. It will not stop
#        immediately. It will stop after completing the current step it is on.
#

enable_image_generation_cancellation = 1

####################

#
#    Include Close Command Prompt / Cancel Button
#
#        This will likely be removed in the future.
#
#        This doesn't work well at all. It just closes the command prompt. I
#        might remove this eventually.
#

enable_close_command_prompt_button = 0

####################

#
#    Use Denoising Start In Base Model When Using Refiner
#
#        If set to "1", refining will end at the percent (expressed as decimal)
#        defined in the denoising start for the refiner. If the steps set are
#        100, and the denoising start value is 0.75, the base model will run for
#        75 steps. The refiner will then run for 25 steps.
#

default_use_denoising_start_in_base_model_when_using_refiner = 1

####################

#
#    Base Model Output To Refiner Is In Latent Space
#
#        If set to "1", base model output is in latent space instead of PIL
#        image when sent to refiner.
#

default_base_model_output_to_refiner_is_in_latent_space = 1

####################

#
#    Log Generation Times
#
#        Log generation times to saved text output. The initial time it takes to
#        load a model is not included in the generation time.
#

log_generation_times = 1

####################

#
#    Use Image Gallery
#

use_image_gallery = 1

####################

#
#    Show Image Creation Progress Log
#
#        This adds the current step that image generation is on.
#

show_image_creation_progress_log = 1

####################

#
#    Show Messages In Command Prompt
#
#        Messages will be printed in command prompt.
#

show_messages_in_command_prompt = 1

####################

#
#    Show Messages In Modal On Page
#
#        A popup appears in the top right corner on the page.
#

show_messages_in_modal_on_page = 0

####################

#
#    Suppress Hugging Face Hub Offline Status
#
#        By default, we add messages about the current setting of
#        "HF_HUB_OFFLINE".
#

suppress_hugging_face_hub_offline_status = 0

####################

#
#    Add Seed Into Pipe
#
#        To make generation deterministic. I add the option because the online
#        configuration for the PhotoReal site doesn't do that and it changes
#        things.
#

default_add_seed_into_pipe = 1

####################

#
#    Use torch.manual_seed But Do Not Add To Pipe
#
#        We need this to match the PhotoReal site to make it work.
#

default_use_torch_manual_seed_but_do_not_add_to_pipe = 0

####################

#
#    Save Base Image When Using Refiner
#
#        The image will be shown on the page. If you save images automatically,
#        it will also be saved.
#

default_save_base_image_when_using_refiner = 1

####################

#
#    Save Refined Image When Using Upscaler
#
#        The image will be shown on the page. If you save images automatically,
#        it will also be saved.
#

default_save_refined_image_when_using_upscaler = 1

####################

#
#    Max Queue Size
#

max_queue_size_if_cpu = 3
max_queue_size_if_torch = 20

####################

#
#    Allow Other Model Versions
#
#        This allows models other than the default versions.
#

allow_other_model_versions = 1

####################

#
#    Image Preview
#

enable_image_preview = 1
use_image_preview = 1
image_preview_step_interval = 10
image_preview_seconds_interval = 60
load_image_preview_frequency_in_seconds = 2

####################

sdxl_link = "https://huggingface.co/spaces/Manjushri/SDXL-1.0"
photoreal_link = "https://huggingface.co/spaces/Manjushri/PhotoReal-V3.8.1"
sdxl_turbo_link = "https://huggingface.co/spaces/diffusers/unofficial-SDXL-Turbo-i2i-t2i"

####################

#
#    Up Next Is Various Configuration Arrays and Objects
#

####################

base_model_array = [
    "sdxl",
    "photoreal",
    "sdxl_turbo",
    "sd_1_5_runwayml"
]

base_model_names_object = {
    "sdxl":            "Stable Diffusion XL",
    "photoreal":       "PhotoReal",
    "sdxl_turbo":      "Stable Diffusion XL Turbo",
    "sd_1_5_runwayml": "Stable Diffusion 1.5"
}

####################

#
#    To match very closely older the older defaults from these two links:
#        https://huggingface.co/spaces/Manjushri/SDXL-1.0
#        https://huggingface.co/spaces/Manjushri/PhotoReal-V3.8.1
#
#    The PhotoReal site doesn't explicitly add the seed to the generation, so
#    that option needs to be unchecked to match that imagery.
#
#    You can use the following customizations...
#
##########
#
#    Stable Diffusion XL
#
#          The refiner always runs for SDXL at this link.
#
#        - Valid from November 12th to present.
#          Number of steps in upscaler changed from 5 to 15.
#
#          Valid from September 5th to November 12th.
#          Number of steps in upscaler was 5 during this period.
#          There were changes on this date.
#
#          http://127.0.0.1:7860/?model=sdxl&model_config=sdxl_1-0&scheduler=model_default&width=768&height=768&guidance=10&steps=50&prompt=&neg_prompt=&seed=&add_seed=yes&refiner=yes&denoise_start=0.95&refiner_steps=100&use_denoise_end=no&latent_space_before_refiner=yes&upscaler=no&upscaler_steps=15
#
##########
#
#    PhotoReal 3.8.1
#
#        - Valid from December 29th to present.
#          Base model was changed from 3.7.5 to 3.8.1: "circulus/canvers-real-v3.8.1"
#
#          For some reason, I haven't been able to match this version of the
#          configuration yet. That means this link doesn't work:
#
#          http://127.0.0.1:7860/?model=photoreal&model_config=photoreal_3-8-1&scheduler=model_default&width=768&height=768&guidance=5&steps=50&prompt=&neg_prompt=&seed=&add_seed=no&use_torch_manual_seed_but_not_in_generator=yes&refiner=no&denoise_start=0.95&refiner_steps=&use_denoise_end=no&latent_space_before_refiner=no&upscaler=no&upscaler_steps=
#
####################
#
#    PhotoReal 3.7.5
#
#        - Valid from November 12th to December 29th.
#          Base model was changed from 3.6 to 3.7.5: "circulus/canvers-real-v3.7.5"
#
#          http://127.0.0.1:7860/?model=photoreal&model_config=photoreal_3-7-5&scheduler=model_default&width=768&height=768&guidance=7&steps=50&prompt=&neg_prompt=&seed=&add_seed=no&use_torch_manual_seed_but_not_in_generator=yes&refiner=no&denoise_start=0.95&refiner_steps=&use_denoise_end=no&latent_space_before_refiner=no&upscaler=no&upscaler_steps=
#
####################
#
#    PhotoReal 3.6
#
#        - Valid from September 1st to November 12th.
#          "circulus/canvers-realistic-v3.6" was already in effect.
#          But there were changes on this date.
#
#          http://127.0.0.1:7860/?model=photoreal&model_config=photoreal_3-6&scheduler=model_default&width=768&height=768&guidance=7&steps=50&prompt=&neg_prompt=&seed=&add_seed=no&use_torch_manual_seed_but_not_in_generator=yes&refiner=no&denoise_start=0.95&refiner_steps=&use_denoise_end=no&latent_space_before_refiner=no&upscaler=no&upscaler_steps=
#
####################

base_model_object_of_model_configuration_arrays = {
    "sdxl": [
        "sdxl_default",
        "sdxl_1-0"
    ],
    "photoreal": [
        "photoreal_default",
        "photoreal_3-8-1",
        "photoreal_3-8",
        "photoreal_3-7-5",
        "photoreal_3-6"
    ],
    "sdxl_turbo": [
        "sdxl_turbo_default",
        "sdxl_turbo_initial"
    ],
    "sd_1_5_runwayml": [
        "sd_1_5_runwayml_default"
    ]
}

####################

model_configuration_names_object = {
    "sdxl_default":            "Default (currently 1.0)",
    "sdxl_1-0":                "1.0",
    "photoreal_default":       "Default (currently 3.6)",
    "photoreal_3-8-1":         "3.8.1",
    "photoreal_3-8":           "3.8",
    "photoreal_3-7-5":         "3.7.5",
    "photoreal_3-6":           "3.6",
    "sdxl_turbo_default":      "Default (currently Initial Release)",
    "sdxl_turbo_initial":      "Initial Release",
    "sd_1_5_runwayml_default": "Default"
}

model_configuration_links_object = {
    "sdxl_default":            "stabilityai/stable-diffusion-xl-base-1.0",
    "sdxl_1-0":                "stabilityai/stable-diffusion-xl-base-1.0",
    "photoreal_default":       "circulus/canvers-realistic-v3.6",
    "photoreal_3-8-1":         "circulus/canvers-real-v3.8.1",
    "photoreal_3-8":           "circulus/canvers-real-v3.8",
    "photoreal_3-7-5":         "circulus/canvers-real-v3.7.5",
    "photoreal_3-6":           "circulus/canvers-realistic-v3.6",
    "sdxl_turbo_default":      "stabilityai/sdxl-turbo",
    "sdxl_turbo_initial":      "stabilityai/sdxl-turbo",
    "sd_1_5_runwayml_default": "runwayml/stable-diffusion-v1-5"
}

####################

base_models_not_supporting_denoising_end_for_base_model_object = {
    "photoreal":       1,
    "sd_1_5_runwayml": 1
}

####################

hugging_face_refiner_partial_path = "stabilityai/stable-diffusion-xl-refiner-1.0"
hugging_face_upscaler_partial_path = "stabilityai/sd-x2-latent-upscaler"

####################

base_model_model_configuration_defaults_object = {
    "sdxl":            "sdxl_default",
    "photoreal":       "photoreal_default",
    "sdxl_turbo":      "sdxl_turbo_default",
    "sd_1_5_runwayml": "sd_1_5_runwayml_default"
}

####################

#
#    Links:
#
#        SD-XL 1.0-base Model Card
#            https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0
#
#        SD-XL 1.0-refiner Model Card
#            https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0
#
#        Stable Diffusion x2 latent upscaler model card
#            https://huggingface.co/stabilityai/sd-x2-latent-upscaler
#
#        PhotoReal
#            3.8.1: https://huggingface.co/circulus/canvers-real-v3.8.1
#            3.8:   https://huggingface.co/circulus/circulus/canvers-real-v3.8
#            3.7.5: https://huggingface.co/circulus/canvers-real-v3.7.5
#            3.6:   https://huggingface.co/circulus/canvers-realistic-v3.6
#
#        SDXL Turbo
#            https://huggingface.co/stabilityai/sdxl-turbo
#
#        Stable Diffusion v1-5 (runwayml)
#            https://huggingface.co/runwayml/stable-diffusion-v1-5
#

####################



default_scheduler = "model_default"



schedulers_array = [
    "model_default",
    "ddim",
    "ddpm",
    "dpm_solver_multistep",
    "dpm_solver_multistep_karras_sigmas_true",
    "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp",
    "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp",
    "dpm_solver_singlestep",
    "dpm_solver_singlestep_karras_sigmas_true",
    "kdpm2_discrete",
    "kdpm2_discrete_karras_sigmas_true",
    "kdpm2_ancestral_discrete",
    "kdpm2_ancestral_discrete_karras_sigmas_true",
    "euler_discrete",
    "euler_ancestral_discrete",
    "heun_discrete",
    "lms_discrete",
    "lms_discrete_karras_sigmas_true",
    "pndm",
    "pndm_skip_prk_steps_true",
    "deis_multistep",
    "dpm_solver_sde",
    "uni_pc_multistep"
]



scheduler_long_names_object = {
    "model_default":                                                           "Model Default",
    "ddim":                                                                    "DDIM",
    "ddpm":                                                                    "DDPM",
    "dpm_solver_multistep":                                                    "DPM++ 2M (DPMSolverMultistep)",
    "dpm_solver_multistep_karras_sigmas_true":                                 "DPM++ 2M Karras (DPMSolverMultistep with use_karras_sigmas=True)",
    "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp":                    "DPM++ 2M SDE (DPMSolverMultistep with algorithm_type=\"sde-dpmsolver++\")",
    "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp": "DPM++ 2M SDE Karras (DPMSolverMultistep with use_karras_sigmas=True & algorithm_type=\"sde-dpmsolver++\")",
    "dpm_solver_singlestep":                                                   "DPM++ SDE (DPMSolverSinglestep)",
    "dpm_solver_singlestep_karras_sigmas_true":                                "DPM++ SDE Karras (DPMSolverSinglestep with use_karras_sigmas=True)",
    "kdpm2_discrete":                                                          "DPM2 (KDPM2Discrete)",
    "kdpm2_discrete_karras_sigmas_true":                                       "DPM2 Karras (KDPM2Discrete with use_karras_sigmas=True)",
    "kdpm2_ancestral_discrete":                                                "DPM2 a (KDPM2AncestralDiscrete)",
    "kdpm2_ancestral_discrete_karras_sigmas_true":                             "DPM2 a Karras (KDPM2AncestralDiscrete with use_karras_sigmas=True)",
    "euler_discrete":                                                          "Euler (EulerDiscrete)",
    "euler_ancestral_discrete":                                                "Euler a (EulerAncestralDiscrete)",
    "heun_discrete":                                                           "Heun (HeunDiscrete)",
    "lms_discrete":                                                            "LMS (LMSDiscrete)",
    "lms_discrete_karras_sigmas_true":                                         "LMS Karras (LMSDiscrete with use_karras_sigmas=True)",
    "pndm":                                                                    "PNDM",
    "pndm_skip_prk_steps_true":                                                "PNDM (with skip_prk_steps=True) - Close to PLMS",
    "deis_multistep":                                                          "DEISMultistep",
    "dpm_solver_sde":                                                          "DPMSolverSDE",
    "uni_pc_multistep":                                                        "UniPCMultistep"
}



scheduler_short_names_object = {
    "ddim":                                                                    "DDIM",
    "ddpm":                                                                    "DDPM",
    "dpm_solver_multistep":                                                    "DPM++ 2M",
    "dpm_solver_multistep_karras_sigmas_true":                                 "DPM++ 2M Karras",
    "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp":                    "DPM++ 2M SDE",
    "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp": "DPM++ 2M SDE Karras",
    "dpm_solver_singlestep":                                                   "DPM++ SDE",
    "dpm_solver_singlestep_karras_sigmas_true":                                "DPM++ SDE Karras",
    "kdpm2_discrete":                                                          "DPM2",
    "kdpm2_discrete_karras_sigmas_true":                                       "DPM2 Karras",
    "kdpm2_ancestral_discrete":                                                "DPM2 a",
    "kdpm2_ancestral_discrete_karras_sigmas_true":                             "DPM2 a Karras",
    "euler_discrete":                                                          "Euler",
    "euler_ancestral_discrete":                                                "Euler a",
    "heun_discrete":                                                           "Heun",
    "lms_discrete":                                                            "LMS",
    "lms_discrete_karras_sigmas_true":                                         "LMS Karras",
    "pndm":                                                                    "PNDM",
    "pndm_skip_prk_steps_true":                                                "PNDM (with skip_prk_steps=True) - Close to PLMS",
    "deis_multistep":                                                          "DEISMultistep",
    "dpm_solver_sde":                                                          "DPMSolverSDE",
    "uni_pc_multistep":                                                        "UniPCMultistep"
}



scheduler_name_to_identifier_in_app_object = {
    "DDIMScheduler":                   "ddim",
    "DDPMScheduler":                   "ddpm",
    "DPMSolverMultistepScheduler":     "dpm_solver_multistep",
    "DPMSolverSinglestepScheduler":    "dpm_solver_singlestep",
    "KDPM2DiscreteScheduler":          "kdpm2_discrete",
    "KDPM2AncestralDiscreteScheduler": "kdpm2_ancestral_discrete",
    "EulerDiscreteScheduler":          "euler_discrete",
    "EulerAncestralDiscreteScheduler": "euler_ancestral_discrete",
    "HeunDiscreteScheduler":           "heun_discrete",
    "LMSDiscreteScheduler":            "lms_discrete",
    "PNDMScheduler":                   "pndm",
    "DEISMultistepScheduler":          "deis_multistep",
    "DPMSolverSDEScheduler":           "dpm_solver_sde",
    "UniPCMultistepScheduler":         "uni_pc_multistep"
}



####################

#
#    Determine automatically if on CPU or GPU
#
#        CPU will not work on Windows.
#

device = "cpu"

if torch.cuda.is_available():

    device = "cuda"

    PYTORCH_CUDA_ALLOC_CONF = {
        "max_split_size_mb": 8000
    }
    torch.cuda.max_memory_allocated(
        device = device
    )
    torch.cuda.empty_cache()

if device == "cpu":

    default_base_model = "sdxl_turbo"

####################

#
#    Determine if running on HuggingFace
#

try:
    if (str(os.uname()).find("magicfixeseverything") >= 0):
        script_being_run_on_hugging_face = 1
except:
    script_being_run_on_hugging_face = 0

if script_being_run_on_hugging_face == 1:

    allow_other_model_versions = 0

####################



default_prompt = ""
default_negative_prompt = ""



default_width = 768
default_height = 768

minimum_width = 64
minimum_height = 64

maximum_width = 2048 # 1024
maximum_height = 2048 # 1024



default_base_model_steps = 50
default_base_model_steps_for_sdxl_turbo = 2

maximum_base_model_steps = 150 # 100
maximum_base_model_steps_for_sdxl_turbo = 25



default_guidance_scale = 7.5

minimum_guidance_scale = 0
maximum_guidance_scale = 30
#    Must be greater than 0
guidance_scale_input_slider_steps = 0.25



#
#    To have the seed be random, set this to:
#
#        "random"
#

default_seed_value = "random"
maximum_seed = 1000000000000000000



add_generation_information_to_image = 1



#    If you turn off the refiner it will not be available in the display unless
#    you select an online configuration option that requires it.

enable_refiner = 1
enable_upscaler = 1

#    Selected on form as a default?

default_refiner_selected = 0
default_upscaler_selected = 0

#    Accordion visible on load?
#
#        0    If selected as default, will be open. Otherwise, closed.
#        1    Always starts open

default_refiner_accordion_open = 1
default_upscaler_accordion_open = 1



default_refiner_denoise_start = 0.95

minimum_refiner_denoise_start = 0.01
maximum_refiner_denoise_start = 0.99
#    Must be greater than 0
refiner_denoise_start_input_slider_steps = 0.01



#    Only for SDXL online configuration. We just use base model steps and denoise start normally
default_refining_steps_for_online_config_field = 100
maximum_refining_steps_for_online_config_field = 100



#    Upscaler Options

maximum_upscaler_steps = 150
default_upscaler_steps = 50



#    xFormers:
#
#        https://huggingface.co/docs/diffusers/optimization/xformers

use_xformers = 1



#    Scaled dot product attention (SDPA) is used by default for PyTorch 2.0. To
#    use default instead, set this to 1.
#
#        https://huggingface.co/docs/diffusers/optimization/torch2.0#scaled-dot-product-attention

use_default_attn_processor = 0



display_xformers_usage_in_prompt_info = 1
include_transformers_version_in_prompt_info = 1
display_diffusers_version_in_prompt_info = 1
display_default_attn_processor_usage_in_prompt_info = 1



#    You can't select both sequential and model cpu offloading. If you select
#    both, model cpu offloading will be used.

use_sequential_cpu_offload_for_base_model = 1
use_sequential_cpu_offload_for_refiner = 1
use_sequential_cpu_offload_for_upscaler = 1

use_model_cpu_offload_for_base_model = 0
use_model_cpu_offload_for_refiner = 0
use_model_cpu_offload_for_upscaler = 0

#    Doesn't work on Windows as of writing this

use_torch_compile_for_base_model = 0
use_torch_compile_for_refiner = 0
use_torch_compile_for_upscaler = 0

if default_base_model == "sdxl":

    #    SDXL

    default_width = 1024
    default_height = 1024

#elif default_base_model == "photoreal":

    #    PhotoReal

#elif default_base_model == "sdxl_turbo":

    #    SDXL Turbo

#elif default_base_model == "sd_1_5_runwayml":

    #    SD 1.5



#    Must be multiple of 8

width_and_height_input_slider_steps = 8



maximum_prompt_characer_count = 1000



opening_html = ""
ending_html = ""

max_queue_size = max_queue_size_if_torch

if device == "cpu":

    if script_being_run_on_hugging_face == 1:

        opening_html = "<span style=\"font-weight: bold; color: #c00;\">This app is extremely slow. Do not use it yet.</span> This app is not running on a GPU. The first time it loads after the space is rebuilt it might take 10 minutes to generate a SDXL Turbo image. It may take around 3 minutes after that point to do two steps. (with no refining or upscaling) For other models, it may take an hour to create a single image. Want apps that work fast? Use these which this app is based on: <a href=\"" + sdxl_link + "\" target=\"_blank\">Stable Diffusion XL</a>, <a href=\"" + photoreal_link + "\" target=\"_blank\">PhotoReal with SDXL 1.0 Refiner</a> and <a href=\"" + sdxl_turbo_link + "\" target=\"_blank\">SDXL Turbo Unofficial Demo</a>. This app is designed to give more options, but it's too slow to operate and test on a CPU. There are some features that are either not available, or more limited, in the online version of this app. (such as smaller allowed image dimensions and less steps allowed) This app is still in development. A future addition will be the ability to create a <a href=\"https://huggingface.co/spaces/magicfixeseverything/ai_image_creation?model=sdxl_turbo&model_config=sdxl_default&scheduler=model_default&width=768&height=768&guidance=7.5&steps=2&add_seed=1&refiner=no&upscaler=no&prompt=cat&neg_prompt=dog&seed=3648905360627576\" target=\"_blank\">link like this</a>."

    else:

        opening_html = "<span style=\"font-weight: bold; color: #c00;\">This app is currently running on a CPU.</span> If you have a NVIDIA graphics card, make sure you have torch installed so that you can use your GPU to create imagery. If you don't, it will work extremely slowly."

    max_queue_size = max_queue_size_if_cpu



ending_html += """<img src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAFUAAABkCAMAAADADYxgAAABelBMVEVzQShrNiBOLB2FSSlaNSPm4uCVZ0RhQjVCJhtcKBc3HhN2TkN8WE2DYVtwRjl8TjOHWjtHGRA4DQivgltQHxLz6OD68+onGRYvWZGlWibswHv78dxOOzKSVDLwzY+idlNrU0ytZDCaUiPZrkrYnFe6jmLhpVvVro7Cl2zWomq4bjIOX7T158vAg1HAdzfR1dzntm6sdk4fc8jWk0rHgUC80OMda7vFoHXkrWWBrtvxyITSqHbOikW2dULcsHydXzSIUDvLj1OjbEP01Zvdq3BGCAqWdnKVamWRWUXJlF/w3ruVc1pGRk2sw9rcvZzeuYydhoAyKSjLo4KKbWjMoUCubjxaT1Lnv5y3knEaX6k8PEGkfGvDlzt2HBpaEBB+WSLy2q3kzam2jDSWbimQKyUUVJhHXH/DlX2nOTCnfS4mSG+slJBniaK5UkAnfdMBZszt7e5FjdRoods2hNTk6Oxcmtl2qNuuyeXJ2emRuN+gwOFSlNfb4+vS3ur8OZTmAAAAbnRSTlPj4+Pj4/vk4+Pj4+Pk4+Pj5OPj4+P7++P14+Pj5OPj4+Tj4+Pj4+Pj4+Pj++Pj4/vj4/7j4/775OP+4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4/zj4+Tk4+Pj4+bj4/nl5OPj4+Pj4+Pj4/Xx4+Pj7+Py4w/umxAAAA4ISURBVGje3NJBbgUhCAbgXoqFlyKAioB69+LbvMxmpi9NNyUZk39IPtH4hX9R/1aVxZfMnlkUsdP8hUpXtVNHHJa81huVjWgVrDuMmp/hiHZK0cjMUvX8L4jeTltzVWi0a193s0abXIxr2yqDBGH0vgMLuTgd9WRApdILSVXSymD8cAOFFoRgbQVzcZwe0ALHzh4clV8nHm2M/N43cK/WWWKTvFQm5xbaLdVVr+qSrPlTdWzpetSlAjQ7KZcWqYEPeqtKLmVJYs5ni6dZ3dZ2rC1gbcU6lnkEogwIMOyQqmfWvUwx+yYoBjiBH19WqnrJMmada+Bdfa7yICLgT9Xnqt+00TFrg0AYxvE3Koe6GKtDe5ALoj0SoWKIUDGGoOmSPW6BQjv1+3+BPseRiIkp7dB/Lg4Ov/d4xff6g/r4H5H78EO+Lrzk/ybXJfcu24vOuRCpR98Y6vtku7eur442HY/NPPKmnnOdVh38bsfQKAtSm22SSFnmec7f2kPOn8kZzRgUhnR8N4Zuf8/Q2db1OkESlcA5ZxM0EHHZwQA1guLj54AFqU2gXVHUOGsd/JILsVFwnzWxLPxxLhkWxXH8Zdn2mXVdbU4WK7bNiizNsjRNi6apkUwkr0S1CYKAkUrDdBXexOiDcF14OEBt2zYMdoi6LoqiLnpSRWlRYxGllDmvKsGBL6beaOQBjlUntdwHwBBhojZod7tdu2oZM01zFm1fmiZZljlcybEHUXERTO9Fp1illqszVBa1KzYz2XyO0cxkGIDrZkWyTEq4HO5+L8QruxN901W/rYkDQRjAN3GZBBOTPfKinKeuLabaIxZFwWKrxT8vTgMV5ArCFVp7r+77f4F7ZrcxXuGGtNCW/jp9dnby+v7TsH8kNEtKpalJSI5IITalKlxXV1+/fl+sOw+j67s6R7BqrZCwrTQ99CaT/b7b7T49PR1F1flj2ffXD1MkvT0JhXgAR/haOPgzRM1m8+vlLdodjzpwW2h39bbdbrNTfSsKN9b9/fITMFKQQNXhkBJIPkklI4yJ4wolHFGJFdFyOBt31uuOZYGyUVapgn19fwHLIybpmCWUAJUoISPXhStwrjxBODZ6fJ49sIlRwJlts/+oKKTwYkdBTrYZDh0G9+250udQHLBw8Vktk3SDOUa2gzrYz81mtqzKKSACq2qYglXu03fhmr5RDKvhEGN2e8sNo1aIFdm+HY9HHBMOaz+ZHA5WtSm8/0YCXS1AojyXSd/MsOuFHverCCUqV8+LPl+LwRT97ia9XpomtrTWhBJfqoVr2pPO+YWu+vyTi4uLalQLIhEoZjngy03/O3K442zT0uNSKIGFWC3KN6rRfS7bKasXbp6HkccJk+bb9nwzY/YabAqx8FB8aQWWbulyguciV6PBbjX38jAIpStEhdDacgaWB3f6ccMK0BZWItiiV9tmlRlfNhr8eJ7Hu6fW8H4FQR65Dthl/2Y+n4PtYBLuP5H4Z/Fh3AKVLNgsTMxe1A7bERIOIz/MoyAPsEorNOxvbtb1+c2iU5/OqTSZRJkV+cEatNwwxgzCWq0Wgo1DV/4KPMpjyW5ztlkPBrhf9WmdDFqaiMmMvE2B20OdVJhBCNWwQkUuUewRhfhVpPvY74wwslgyBNaobBpDYGmwi0PzoRaotGgI1BSPq/IEFqNSpCHERMP7RWc0wO5KmD2pLqvL5RKblLstWsXDqoyCQkUFzAodqJgURiAn0r3dfDFmdkeWhQrBqEPU0rBFq9KiXgC1bDYS2GGIQMGLKZ10u7v5ff9hzatrwgMLtlB9kaDAcrbSObWKiqxauIHg5aIh6wS5xthel/37xWiAl80KGyAlJYRVfVZxiYdUsSq+g4dRpFqytlmwmgTugOAxcCo4sPn6uo5XTfYt2x5UbFX/Q01LtQwAKuqsWbAgVZyksZlHgZfkeN1CAi3sw6xL/6iJ7dWBKq3qQI2gcrXP2EjECSmNdzafDYb28ZI3DF40mVUtyirqpNpYJXr1CrU8MbDcLNGEkC6vL0Q7W2DDTN+y7JioM1Wzqoln1rEqn9VJDUuV2VgdNE0SBRZHl/bS2QJTUF/t9gkp4ZSqZvSHVSUadRFAqX4eL6RKacqdauS67G/GnXrrjrchxUb1jZprraHGgll7q4TZAGBLF9Vue42GK9Fp0ktiokTHmILZ5mHQutsBLVtl9W9d9v7TNBTFAfy2HW23tWu1Ao5s7hEm7cgcqGwu0zGM4kiYMI0vjP7kKz6GApKg8X/3e85pN0D8KsapfDg99/bcDhHewghUKlQjFezUxby2bZlkKqwF7ZqpsG3DklF8toEnOjwnTncrq/HI9WAoPljhMiD7QEwZjzrfGbh+Khb3QdDuho3bEVTsVUJpRxZSOqlujOKwgwpX2fOcjK6DJZQKoLNHqZzXR4lmO8SZ6RlBuNXrRRhcO4wiVhmN0jPKQfCVclheX8WqTuGjldicAoik+bEuVwqDftBum+hqsbG18QCPnSs7ptIEdZ8+dS1dOajGU5bFqsZtZUR+lWLRG2lLAX03w3BgdsPQ9IpGdSPawCAIg7agOXdzd3ezrKtsDrEQjyYvVJWCiLqu8EOLsFrKlgN83k4NwnbJDLrdmmka1cpWD/vV9HcUq1l388uXXbegICL0dXxqLKkY4fHpmMYVO05Wpaix8/OjQiGjD8InH4P73agdmOvVfH0RDzCB6nkyO5zNXaiWgkiRWlElqiV0HgGrazi4HAuqbtkjqFbWDPHcE9S63aC0Xq3W7+KZvq1CHwsixe6SmuLwuef51E0UBpPYEW2mrPsUB0GKZi5M/KtSuPT4xVKjF9WCCql4x3TNq5lQudinm5uk6gijjvL4KCdVt0fzI9uOVadsyTGucln3NY6BFy+ghvef5aGuXF/psSosbwIVv6KJ54maTsvsLtBf0TW5wqI5qMV9/WQJ7FYURevP8Ex/e2FxpSYdkGJRRU5RFbR5HalVYzVOvKxPMVjK5ZzFTwePXuJp8slON4q66xWoVGugdak6Xp6y69JqFRBSy2V5VwYVf80mohx385EMrHKW0Ncflx4vPVnqQa1AvXH3zsJKSXV5spCa5WO+bImKJKqmWZPkWEWzyHVcVnH2kfpAVLxPMr2Q6pAFAlQ4pVoKrGJV5eTeUKKCpQCF+vIJ9sBO70EU3YvVaxiLokozn2/HqgX0rCqshYveJdahJCz2QO/B4kaiLgQ1H6qkYD2HarHKIVOaAJVuf5RK6hewOFw89FV6wOrGjaQDUZcWK1EL29tQKRereOXSAqHYTewWjEqj3x8MzODjzotedKOer+Bd3a3rC42FbjKwR3ufr75//ypWC/TBpqhggWoKd3F/gLhQNWUgcM1g6cWDjXplqoaE7n1+/2pZopiUnFI5GhyOUoZHpcsrsOFO1KhM1MWFml74/J641s2bYwSqBM2FyncBq3LoMqQhWWfKFs2gRxsAap3U6wu7n1mUb4seHHdUgiKJCpQtCf8O8xAsvWJ2tbtVn6gr7a+t5WUif3eGzRmKKkwDjVR8CMrjO80dwaKB5fJZjRoTNfra4us++MWgqJgiNtbOLuztyYZNVIRUJEWoS7XaeqziTVxe1NdiHjfF+yNqhjMa7YmqmFNQwclU4I2m0JlL85mUqI3GaiVfhVv/0Grh0jszZ6PmKSN7b297G6rG1RJOyuUr2qmkMrZlAYV6r5FaJ9W8SuhBc3butHnEKh2e24jUSmqOa9b1y2nFX8P3MbLo8TunSF1/FugKqv8K6364BlJUMeNageLmZdWAZmie4xELN2NlffQUEysLE62FSR0w9VS+6hD6c2YO5PlaUSmpZbBSK/1wiVXUVVsHlsPBZdEUyHqsVhZ9W8+XX2E7dWbn5uZmJ+pwOFyT1RIVA8yKTVU1cTAY0mX+vh+mDFBSpdRFZae40hOgCNeLNPf393/Eqk1T0cGsj2s1qvdMJwuVIvcvHphY7fNa3YO6R5UOZ/+j0skHlT7HVxBZvdHAK2GpYsRj1e/zWlUic0Tor9lYnf1HRej9k494YEW9bSLcW0FxZCdqsbJeiYL3y3T5iKCiSl+nKhUDFSJMUldLpdVVg0tNUKTfN4pIZesrLxQys/br5KQznJkVtYnIHWsTKxWdVlcbjXsGt5RJF/EFNSq7rRa2FMyf+3EO1/5ZLTs+/A1R8ZxzG596+8Y6UDYFdQg1EBPoIaFjFiXD02r8XE3REzVfryutkqcRyypcbipMzgfcpmu4fKp0/BP/e3STtfOqxDY4eah5TasaMl49Th8x4pSxUY+pp8DGM7RSTRQ9vljNkKqMKqlVVJlOC3s6Rfp5la8fauf4eDjH6/8/NYPOylrR4ORvQ0ClnDGR58ut8a85qPEe6Bz8AHqBqmtv3n1rBIa0FaxKQ03moBIxydXl8Y8ZqNhMncN9yQVqJv0J+dZYFRQt4FLPqFO4TKVyneimeIcH5zugY3hq75BPgyKriKFIvcQq31xEJvCH5ZvoKuU37dMhLdfsj3Mq3sb302+Qd0qJil+AkiqNFTVxvVctbACwsgXiQXB4Xq0Fg/RbYjWNVWxTlIokKo+xifsc9+raRD0U9YT6gJInaikIzPRbSkqDh+2qYvWKqOpMrWjAmD6PXGrrz7Vm82TM3aUtlqi+aQ7SD4E+TE9U7bQKVNRpAwiF2oEhuUnu96mKaeR7+kOKLmVVtbOqEXegyD+z2AFDRvHjODY7RydUtqi/WUXOqAapmqjnS5W2NsWkvdU57pw0+fY6OjqaQ68RvIBqWbEK7SLVmPaV9xUuESx9UCBOMvmDv2JFnMdVezSBAAAAAElFTkSuQmCC" alt="Spaghetti AI Logo" title="Spaghetti AI Logo" style="float: left; padding: 15px 15px 15px 0px;">

Tokens are not individual characters. If the prompt length is too long, the display will notify you what part of the prompt wasn't used. Changing just the image dimensions alone will change the image generated. For some models, trying to make a large image, such as 1024x1024, may add extra people and come out worse than using smaller dimensions.

The original script for this app was written by <a href=\"https://huggingface.co/Manjushri\" target=\"_blank\">Manjushri</a>."""



refiner_on_text = "Refiner is on. "
refiner_off_text = "Refiner is off. "

upscaler_on_text = "Upscaler is on. "
upscaler_off_text = "Upscaler is off. "

number_of_reserved_tokens = 2



generate_image_button_normal_text = "Generate Image"
generate_image_button_in_progress_text = "Generating..."

cancel_image_button_text = "Cancel"
cancel_image_button_in_progress_text = "Cancelling..."



gradio_image_component_height = 300
gradio_image_gallery_component_height = 350



canceled_image_in_queue_message = "Due to how the queue works in this app, you need to reload the page after canceling an image. Otherwise, you will not be able to generate another image until you reach the position in the queue you were in originally. At that time, the button to generate an image will appear again."

canceled_image_in_process_of_being_generated = "<div style=\"text-align: center;\">Image generation will be canceled once the current step completes.</div>"



###############################################################################
###############################################################################
#
#
#
#    End Configurations
#
#
#
###############################################################################
###############################################################################



if default_add_seed_into_pipe == 1:

    default_use_torch_manual_seed_but_do_not_add_to_pipe = 0



if script_being_run_on_hugging_face == 1:

    #    If on HuggingFace, I change some things.

    use_custom_hugging_face_cache_dir = 0
    auto_save_imagery = 0
    show_messages_in_modal_on_page = 0
    show_messages_in_command_prompt = 1

    only_use_local_files = 0

    if device == "cpu":

        #    If on CPU at HuggingFace, I reduce what is available amd do other
        #    things.

        use_torch_compile_for_base_model = 0
        use_torch_compile_for_refiner = 0
        use_torch_compile_for_upscaler = 0

        show_image_creation_progress_log = 1

        minimum_width = 256
        minimum_height = 256

        maximum_width = 768
        maximum_height = 768

        minimum_guidance_scale = 1
        maximum_guidance_scale = 15

        maximum_base_model_steps = 30

        maximum_base_model_steps_for_sdxl_turbo = 5

        minimum_refiner_denoise_start = 0.70

        maximum_upscaler_steps = 15
        default_upscaler_steps = 10



    ending_html = """
If you would like to download this app to run offline on a Windows computer that has a NVIDIA graphics card, click <a href=\"https://huggingface.co/spaces/magicfixeseverything/ai_image_creation/resolve/main/spaghetti_ai.zip\">here</a> to download it.

""" + ending_html



if default_width < minimum_width:
    default_width = minimum_width

if default_height < minimum_height:
    default_height = minimum_height



if default_width > maximum_width:
    default_width = maximum_width

if default_height > maximum_height:
    default_height = maximum_height



if default_base_model_steps > maximum_base_model_steps:
    default_base_model_steps = maximum_base_model_steps

if default_base_model_steps_for_sdxl_turbo > maximum_base_model_steps_for_sdxl_turbo:
    default_base_model_steps_for_sdxl_turbo = maximum_base_model_steps_for_sdxl_turbo



if default_guidance_scale < minimum_guidance_scale:
    default_guidance_scale = minimum_guidance_scale

if default_guidance_scale > maximum_guidance_scale:
    default_guidance_scale = maximum_guidance_scale



if default_upscaler_steps > maximum_upscaler_steps:
    default_upscaler_steps = maximum_upscaler_steps



only_use_local_files_bool = False

if only_use_local_files == 1:

    only_use_local_files_bool = True



if allow_other_model_versions == 0:

    base_model_array = [
        "sdxl",
        "photoreal",
        "sdxl_turbo"
    ]

    base_model_object_of_model_configuration_arrays = {
        "sdxl": [
            "sdxl_default"
        ],
        "photoreal": [
            "photoreal_default"
        ],
        "sdxl_turbo": [
            "sdxl_turbo_default"
        ]
    }

    base_model_model_configuration_defaults_object = {
        "sdxl":            "sdxl_default",
        "photoreal":       "photoreal_default",
        "sdxl_turbo":      "sdxl_turbo_default"
    }



hugging_face_hub_is_offline = 0

if script_being_run_on_hugging_face == 0:

    if (
        ("HF_HUB_OFFLINE" in os.environ) and
        (int(os.environ["HF_HUB_OFFLINE"]) == 1)
    ):

        hugging_face_hub_is_offline = 1

        only_use_local_files = 1

    if suppress_hugging_face_hub_offline_status == 1:

        if hugging_face_hub_is_offline == 0:

            print ("Note: The Hugging Face cache directory does not automatically delete older data. Over time, it could eventually grow to use all the space on the drive it is on. You either need to manually clean out the folder occasionally or see Instructons.txt on how to not automatically update data once you have downloaded everything you need.")

        else:

            print ("You are working offline. Data will not be downloaded. See \"ai_image_creation.bat\" or \"Instructions.txt\" for more info.")



saved_images_dir = main_dir + "/" + saved_images_folder_name

hugging_face_cache_dir = main_dir + "/" + cache_directory_folder_name

if not os.path.exists(hugging_face_cache_dir):
    os.makedirs(hugging_face_cache_dir)



if auto_save_imagery == 1:

    from datetime import datetime



if device == "cpu":

    use_sequential_cpu_offload_for_base_model = 0
    use_sequential_cpu_offload_for_refiner = 0
    use_sequential_cpu_offload_for_upscaler = 0

    use_model_cpu_offload_for_base_model = 0
    use_model_cpu_offload_for_refiner = 0
    use_model_cpu_offload_for_upscaler = 0

    use_xformers = 0



if (
    (use_sequential_cpu_offload_for_base_model == 1) and
    (use_model_cpu_offload_for_base_model == 1)
):
    
    use_sequential_cpu_offload_for_base_model = 0

if (
    (use_sequential_cpu_offload_for_refiner == 1) and
    (use_model_cpu_offload_for_refiner == 1)
):
    
    use_sequential_cpu_offload_for_refiner = 0

if (
    (use_sequential_cpu_offload_for_upscaler == 1) and
    (use_model_cpu_offload_for_upscaler == 1)
):
    
    use_sequential_cpu_offload_for_upscaler = 0



def error_function(
    text_message
):

    print (text_message)

    raise Exception(text_message)



additional_prompt_info_html = ""

if auto_save_imagery == 1:

    additional_prompt_info_html = " The image, and a text file with generation information, will be saved automatically."



if use_xformers == 1:

    from xformers.ops import MemoryEfficientAttentionFlashAttentionOp

if use_default_attn_processor == 1:

    from diffusers.models.attention_processor import AttnProcessor



if (
    default_base_model and
    (default_base_model in base_model_object_of_model_configuration_arrays) and
    (default_base_model in base_model_model_configuration_defaults_object)
):

    default_model_configuration = base_model_model_configuration_defaults_object[default_base_model]

    if default_model_configuration in model_configuration_names_object:

        default_model_configuration_choices_array = []

        for this_model_configuration in base_model_object_of_model_configuration_arrays[default_base_model]:

            if this_model_configuration in model_configuration_names_object:

                default_model_configuration_choices_array.append(
                    model_configuration_names_object[this_model_configuration]
                )

            else:

                error_function("A default model version must be properly named in the code.")

    else:

        error_function("A default model version must be properly configured in the code.")

else:

    error_function("A default base model must be properly configured in the code.")



default_base_model_nicely_named_value = base_model_names_object[default_base_model]

default_model_configuration_nicely_named_value = model_configuration_names_object[default_model_configuration]



if not (
    default_scheduler and
    default_scheduler in scheduler_long_names_object
):

    error_function("A default scheduler must be properly configured in the code.")

default_scheduler_nicely_named_value = scheduler_long_names_object[default_scheduler]



if enable_refiner != 1:

    default_refiner_selected = 0

if enable_upscaler != 1:

    default_upscaler_selected = 0



default_refine_option = "No"

if default_refiner_selected == 1:

    default_refine_option = "Yes"

default_upscale_option = "No"

if default_upscaler_selected == 1:

    default_upscale_option = "Yes"



default_refiner_and_upscaler_status_text = ""



default_use_denoising_start_in_base_model_when_using_refiner_is_selected = False

if default_use_denoising_start_in_base_model_when_using_refiner == 1:

    default_use_denoising_start_in_base_model_when_using_refiner_is_selected = True

default_base_model_output_to_refiner_is_in_latent_space_is_selected = False

if default_base_model_output_to_refiner_is_in_latent_space == 1:

    default_base_model_output_to_refiner_is_in_latent_space_is_selected = True



refiner_accordion_visible = True

if enable_refiner != 1:

    refiner_accordion_visible = False

refiner_accordion_open = False

if default_refiner_accordion_open == 1:

    refiner_accordion_open = True



refiner_group_visible = False

if enable_refiner == 1:

    refiner_group_visible = True

    if default_refiner_selected == 1:

        default_refiner_and_upscaler_status_text += refiner_on_text

    else:

        default_refiner_and_upscaler_status_text += refiner_off_text



upscaler_accordion_open = False

if (
    (default_upscaler_selected == 1) or
    (default_upscaler_accordion_open == 1)
):

    upscaler_accordion_open = True

upscaler_group_visible = False

if enable_upscaler == 1:

    upscaler_group_visible = True

    if default_upscaler_selected == 1:

        default_refiner_and_upscaler_status_text += upscaler_on_text

    else:

        default_refiner_and_upscaler_status_text += upscaler_off_text



default_negative_prompt_field_row_visibility = True
default_negative_prompt_for_sdxl_turbo_field_row_visibility = False
default_base_model_steps_field_row_visibility = True
default_base_model_steps_field_for_sdxl_turbo_field_row_visibility = False
default_guidance_scale_field_row_visibility = True
default_guidance_scale_for_sdxl_turbo_field_row_visibility = False

if default_base_model == "sdxl_turbo":

    default_negative_prompt_field_row_visibility = False
    default_negative_prompt_for_sdxl_turbo_field_row_visibility = True
    default_base_model_steps_field_row_visibility = False
    default_base_model_steps_field_for_sdxl_turbo_field_row_visibility = True
    default_guidance_scale_field_row_visibility = False
    default_guidance_scale_for_sdxl_turbo_field_row_visibility = True



default_refining_use_denoising_start_in_base_model_when_using_refiner_field_row_visibility = True

if default_base_model in base_models_not_supporting_denoising_end_for_base_model_object:

    default_refining_use_denoising_start_in_base_model_when_using_refiner_field_row_visibility = False



default_add_seed_into_pipe_is_selected = False

if default_add_seed_into_pipe == 1:

    default_add_seed_into_pipe_is_selected = True



default_use_torch_manual_seed_but_do_not_add_to_pipe_is_selected = False

if default_use_torch_manual_seed_but_do_not_add_to_pipe == 1:

    default_use_torch_manual_seed_but_do_not_add_to_pipe_is_selected = True



default_save_base_image_when_using_refiner_is_selected = False

if default_save_base_image_when_using_refiner == 1:

    default_save_base_image_when_using_refiner_is_selected = True



default_save_refined_image_when_using_upscaler_is_selected = False

if default_save_base_image_when_using_refiner == 1:

    default_save_refined_image_when_using_upscaler_is_selected = True



default_base_model_choices_array = []

stored_model_configuration_names_object = {}

for this_base_model in base_model_array:

    default_base_model_choices_array.append(
        base_model_names_object[this_base_model]
    )

    stored_model_configuration = base_model_model_configuration_defaults_object[this_base_model]

    stored_model_configuration_names_object[this_base_model] = model_configuration_names_object[stored_model_configuration]



default_scheduler_choices_array = []

for this_scheduler in schedulers_array:

    default_scheduler_choices_array.append(
        scheduler_long_names_object[this_scheduler]
    )



if enable_image_preview != 1:

    use_image_preview = 0



make_seed_selection_a_textbox = 1

if maximum_seed <= 9007199254740992:

    make_seed_selection_a_textbox = 0



current_preview_image = ""
current_preview_image_user_id = 0



current_image_generation_id_in_progress = 0



cancel_image_generation_ids_object = {}
cancel_image_generation_times_object = {}
seconds_to_store_cancellations_in_cancel_image_generation_times_object = 86400



###############################################################################
###############################################################################
#
#
#
#
#
#
#    Functions
#
#
#
#
#
#
###############################################################################
###############################################################################







#####################
#
#    Rounded Number
#
#        A better, and seemingly more accurate, way to round.
#
#        https://realpython.com/python-rounding/

#####################

def rounded_number(n, decimals=0):
    n = float(n)
    multiplier = 10**decimals
    rounded_abs = (floor(abs(n) * multiplier + 0.5) / multiplier)
    rounded_value = round(copysign(rounded_abs, n), decimals)
    return rounded_value







#####################
#
#    Rounded Number
#
#        Format number to a certain number of decimal places and output it to a string.
#
#        https://stackoverflow.com/questions/1995615/how-can-i-format-a-decimal-to-always-show-2-decimal-places

#####################

def formatted_number(n, decimals=0):
    rounded_value = rounded_number(n, decimals)
    formatted_value = '{:.{prec}f}'.format(rounded_value, prec=decimals)
    return formatted_value







#####################
#
#    Show Message
#
#        Display message to user in model on web form and/or command prompt.
#
#####################

def generate_random_seed():

    maximum_seed_for_random = maximum_seed

    if maximum_seed_for_random > 9007199254740992:

        #    If above this number, seeds may not be able to be entered into
        #    slider properly.

        maximum_seed_for_random = 9007199254740992

    actual_seed = int(random.randrange(0, 10**len(str(maximum_seed_for_random))))

    return actual_seed







#####################
#
#    Show Message
#
#        Display message to user in model on web form and/or command prompt.
#
#####################

def show_message(
    message_to_display
):

    if show_messages_in_command_prompt == 1:

        print (message_to_display)

    if show_messages_in_modal_on_page == 1:

        gr.Info(message_to_display)







#####################
#
#    Nice Elapsed Time
#
#        Formatted nicely from seconds.
#
#####################

def nice_elapsed_time(
    seconds
):
    #    Google AI Code

    hours = seconds // 3600
    minutes = (seconds % 3600) // 60
    seconds = seconds % 60

    if hours > 0:

        hours_text = "hr"

        if hours > 1:

            hours_text = "hrs"

        time_html = str(int(hours)) + " " + hours_text + ". " + str(int(minutes)) + " min. " + str(round(seconds, 1)) + " sec."

    elif minutes > 0:

        time_html = str(int(minutes)) + " min. " + str(round(seconds, 1)) + " sec."

    else:

        time_html = str(round(seconds, 2)) + " sec."

    return time_html







#####################
#
#    Base Model Valid
#
#        Return True if valid.
#
#####################

def base_model_valid(base_model_name_value):
    try:
        base_model_name_value_str = str(base_model_name_value).lower()
        if (
            (base_model_name_value_str in base_model_object_of_model_configuration_arrays) and
            (base_model_name_value_str in base_model_model_configuration_defaults_object)
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Model Configuration Valid
#
#        Return True if valid.
#
#####################

def model_configuration_valid(
    base_model_name_value,
    model_configuration_name_value
):
    try:
        base_model_name_value_str = str(base_model_name_value).lower()
        model_configuration_name_value_str = str(model_configuration_name_value).lower()
        for this_base_model in base_model_array:
            for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:
                if (
                    (base_model_name_value_str == this_base_model) and
                    (model_configuration_name_value_str == this_model_configuration)
                ):
                    return True
        return False
    except ValueError:
        return False







#####################
#
#    Prompt Valid
#
#        Return True if valid.
#
#####################

def prompt_valid(prompt_field):
    try:
        prompt_field_str = str(prompt_field)
        if len(prompt_field_str) <= maximum_prompt_characer_count:
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Negative Prompt Valid
#
#        Return True if valid.
#
#####################

def negative_prompt_valid(negative_prompt_field):
    try:
        negative_prompt_field_str = str(negative_prompt_field)
        if len(negative_prompt_field_str) <= maximum_prompt_characer_count:
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Scheduler/Sampler Valid
#
#        Return True if valid.
#
#####################

def scheduler_valid(scheduler_field):
    try:
        scheduler_str = str(scheduler_field).lower()
        if scheduler_str in scheduler_long_names_object:
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Width Valid
#
#        Return True if valid.
#
#####################

def width_valid(width_num_str):
    try:
        width_num = int(width_num_str)
        if (
            (width_num >= int(minimum_width)) and
            (width_num <= int(maximum_width)) and
            (width_num % int(width_and_height_input_slider_steps)) == 0
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Height Valid
#
#        Return True if valid.
#
#####################

def height_valid(height_num_str):
    try:
        height_num = int(height_num_str)
        if (
            (height_num >= int(minimum_height)) and
            (height_num <= int(maximum_height)) and
            (height_num % int(width_and_height_input_slider_steps)) == 0
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Guidance Scale Valid
#
#        Return True if valid.
#
#####################

def guidance_scale_valid(guidance_scale_str):
    try:
        guidance_scale_num = float(guidance_scale_str)
        guidance_scale_num_times_100 = (guidance_scale_num * 100)
        guidance_scale_num_times_100_with_int = int(guidance_scale_num_times_100)
        guidance_scale_input_slider_steps_times_100 = (float(guidance_scale_input_slider_steps) * 100)
        if (
            (guidance_scale_num >= float(minimum_guidance_scale)) and
            (guidance_scale_num <= float(maximum_guidance_scale)) and
            (guidance_scale_num_times_100 == guidance_scale_num_times_100_with_int) and
            ((guidance_scale_num_times_100 % guidance_scale_input_slider_steps_times_100) == 0)
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Steps Valid
#
#        Return True if valid.
#
#####################

def steps_valid(
    steps_num_str,
    base_model_name_value
):
    try:
        steps_num = int(steps_num_str)
        base_model_name_value_str = str(base_model_name_value).lower()
        if steps_num > 0:
            if (base_model_name_value_str == "sdxl_turbo"):
                if steps_num <= int(maximum_base_model_steps_for_sdxl_turbo):
                    return True
            else:
                if steps_num <= int(maximum_base_model_steps):
                    return True
        return False
    except ValueError:
        return False







#####################
#
#    Seed Valid
#
#        Return True if valid.
#
#####################

def seed_valid(
    seed_num_str
):
    try:
        seed_num = int(seed_num_str)
        if (
            (seed_num >= 0) and
            (seed_num <= int(maximum_seed))
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Refiner Denoise Start
#
#        Return True if valid.
#
#####################

def refiner_denoise_start_valid(
    refiner_denoise_start_str
):
    try:
        refiner_denoise_start_num = float(refiner_denoise_start_str)
        refiner_denoise_start_num_times_100 = (refiner_denoise_start_num * 100)
        refiner_denoise_start_num_times_100_with_int = int(refiner_denoise_start_num_times_100)
        refiner_denoise_start_input_slider_steps_times_100 = (float(refiner_denoise_start_input_slider_steps) * 100)
        if (
            (refiner_denoise_start_num >= float(minimum_refiner_denoise_start)) and
            (refiner_denoise_start_num <= float(maximum_refiner_denoise_start)) and
            (refiner_denoise_start_num_times_100 == refiner_denoise_start_num_times_100_with_int) and
            ((refiner_denoise_start_num_times_100 % refiner_denoise_start_input_slider_steps_times_100) == 0)
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Refiner Steps
#
#        Return True if valid.
#
#####################

def refining_steps_valid(
    refining_steps_num_str
):
    try:
        refining_steps_num = int(refining_steps_num_str)
        if (
            (refining_steps_num > 0) and 
            (refining_steps_num <= int(maximum_refining_steps_for_online_config_field))
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Upscaler Steps
#
#        Return True if valid.
#
#####################

def upscaling_steps_valid(
    upscaling_steps_num_str
):
    try:
        upscaling_steps_num = int(upscaling_steps_num_str)
        if (
            (upscaling_steps_num > 0) and 
            (upscaling_steps_num <= int(maximum_upscaler_steps))
        ):
            return True
        else:
            return False
    except ValueError:
        return False







#####################
#
#    Numerical Bool
#
#        Return 1 for anything that is True/Yes/1. Everything else is False.
#
#####################

def numerical_bool(
    original_value
):

    new_value = 0

    if (
        (original_value == 1) or
        (original_value == "Yes") or
        (original_value == "True") or
        (original_value == True)
    ):

        new_value = 1

    return new_value







#####################
#
#    Truncate Prompt
#
#        Truncate a prompt. Get the actual prompt that will be used and save the
#        part of the prompt that will not be used.
#
#####################

def truncate_prompt (
    pipe,
    existing_prompt_text
):

    #    Only 77 tokens are allowed in the prompt. 2 are reserved, meaning it is
    #    truncated to 75. This happens automatically, but we want to tell people
    #    that

    tokenizer = pipe.tokenizer

    max_token_length_of_model = pipe.tokenizer.model_max_length - number_of_reserved_tokens

    prompt_text_words_array = existing_prompt_text.split(" ")

    prompt_text_words_array_length = len(prompt_text_words_array)

    prompt_text_words_index = 0

    prompt_text_substring = ""
    prompt_text_not_used_substring = ""

    for prompt_text_word in prompt_text_words_array:

        prompt_text_words_index += 1

        substring_to_test = prompt_text_substring

        if prompt_text_words_index > 1:

            substring_to_test += " "

        substring_to_test += prompt_text_word

        token_length_of_substring_to_test = len(tokenizer.tokenize(substring_to_test))

        if token_length_of_substring_to_test > max_token_length_of_model:

            prompt_text_not_used_substring += prompt_text_word + " "

        else:

            prompt_text_substring = substring_to_test

    return (
        prompt_text_substring,
        prompt_text_not_used_substring
    )







#####################
#
#    Construct Pipe
#
#        Prepare the base model.
#
#####################

def construct_pipe (
    base_model_name_value,
    model_configuration_name_value
):

    if device == "cuda":
        torch.cuda.empty_cache()

    base_model_kwargs = {}

    if (
        (base_model_name_value == "sdxl") or
        (base_model_name_value == "photoreal") or
        (base_model_name_value == "sdxl_turbo") or
        (base_model_name_value == "sd_1_5_runwayml")
    ):

        base_model_kwargs["use_safetensors"] = True

    if use_safety_checker == 0:

        if (
            (base_model_name_value == "photoreal") or
            (base_model_name_value == "sd_1_5_runwayml")
        ):

            base_model_kwargs = {
                "safety_checker": None,
                "requires_safety_checker": False
            }

    if device == "cuda":

        if (
            (base_model_name_value == "sdxl") or
            (base_model_name_value == "sdxl_turbo") or
            (base_model_name_value == "sd_1_5_runwayml")
        ):

            base_model_kwargs["variant"] = "fp16"

        base_model_kwargs["torch_dtype"] = torch.float16

    if use_custom_hugging_face_cache_dir == 1:

        base_model_kwargs["cache_dir"] = hugging_face_cache_dir

    pipe = DiffusionPipeline.from_pretrained(
        pretrained_model_name_or_path = model_configuration_links_object[model_configuration_name_value],
        local_files_only = only_use_local_files_bool,
        **base_model_kwargs
    )

    if use_model_cpu_offload_for_base_model == 1:
        pipe.enable_model_cpu_offload()

    if use_xformers == 1:
        pipe.enable_xformers_memory_efficient_attention()

    pipe = pipe.to(device)

    if use_sequential_cpu_offload_for_base_model == 1:
        pipe.enable_sequential_cpu_offload()

    if use_default_attn_processor == 1:
        pipe.unet.set_default_attn_processor()

    if use_torch_compile_for_base_model == 1:
        pipe.unet = torch.compile(
            pipe.unet,
            mode = "reduce-overhead",
            fullgraph = True
        )

    if device == "cuda":
        torch.cuda.empty_cache()

    return (
        pipe
    )







#####################
#
#    Configure Scheduler
#
#####################

def configure_scheduler (
    pipe,
    scheduler_value
):

    scheduler_config = pipe.scheduler.config

    scheduler = scheduler_value



    if scheduler_value == "model_default":

        scheduler_name = pipe.scheduler.config._class_name

        if scheduler_name in scheduler_name_to_identifier_in_app_object:

            scheduler = scheduler_name_to_identifier_in_app_object[scheduler_name]



    scheduler_used = scheduler



    if scheduler == "ddim":

        from diffusers import DDIMScheduler
        pipe.scheduler = DDIMScheduler.from_config(scheduler_config)

    elif scheduler == "ddpm":

        from diffusers import DDPMScheduler
        pipe.scheduler = DDPMScheduler.from_config(scheduler_config)

    elif scheduler == "dpm_solver_multistep":

        from diffusers import DPMSolverMultistepScheduler
        pipe.scheduler = DPMSolverMultistepScheduler.from_config(scheduler_config)

    elif scheduler == "dpm_solver_multistep_karras_sigmas_true":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"use_karras_sigmas": True})

        from diffusers import DPMSolverMultistepScheduler
        pipe.scheduler = DPMSolverMultistepScheduler.from_config(new_scheduler_config)

    elif scheduler == "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"algorithm_type": "sde-dpmsolver++"})

        from diffusers import DPMSolverMultistepScheduler
        pipe.scheduler = DPMSolverMultistepScheduler.from_config(new_scheduler_config)

    elif scheduler == "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"use_karras_sigmas": True})
        new_scheduler_config.update({"algorithm_type": "sde-dpmsolver++"})

        from diffusers import DPMSolverMultistepScheduler
        pipe.scheduler = DPMSolverMultistepScheduler.from_config(new_scheduler_config)

    elif scheduler == "dpm_solver_singlestep":

        from diffusers import DPMSolverSinglestepScheduler
        pipe.scheduler = DPMSolverSinglestepScheduler.from_config(scheduler_config)

    elif scheduler == "dpm_solver_singlestep_karras_sigmas_true":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"use_karras_sigmas": True})

        from diffusers import DPMSolverSinglestepScheduler
        pipe.scheduler = DPMSolverSinglestepScheduler.from_config(new_scheduler_config)

    elif scheduler == "kdpm2_discrete":

        from diffusers import KDPM2DiscreteScheduler
        pipe.scheduler = KDPM2DiscreteScheduler.from_config(scheduler_config)

    elif scheduler == "kdpm2_discrete_karras_sigmas_true":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"use_karras_sigmas": True})

        from diffusers import KDPM2DiscreteScheduler
        pipe.scheduler = KDPM2DiscreteScheduler.from_config(new_scheduler_config)

    elif scheduler == "kdpm2_ancestral_discrete":

        from diffusers import KDPM2AncestralDiscreteScheduler
        pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(scheduler_config)

    elif scheduler == "kdpm2_ancestral_discrete_karras_sigmas_true":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"use_karras_sigmas": True})

        from diffusers import KDPM2AncestralDiscreteScheduler
        pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(new_scheduler_config)

    elif scheduler == "euler_discrete":

        from diffusers import EulerDiscreteScheduler
        pipe.scheduler = EulerDiscreteScheduler.from_config(scheduler_config)

    elif scheduler == "euler_ancestral_discrete":

        from diffusers import EulerAncestralDiscreteScheduler
        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(scheduler_config)

    elif scheduler == "heun_discrete":

        from diffusers import HeunDiscreteScheduler
        pipe.scheduler = HeunDiscreteScheduler.from_config(scheduler_config)

    elif scheduler == "lms_discrete":

        from diffusers import LMSDiscreteScheduler
        pipe.scheduler = LMSDiscreteScheduler.from_config(scheduler_config)

    elif scheduler == "lms_discrete_karras_sigmas_true":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"use_karras_sigmas": True})

        from diffusers import LMSDiscreteScheduler
        pipe.scheduler = LMSDiscreteScheduler.from_config(new_scheduler_config)

    elif scheduler == "pndm":

        from diffusers import PNDMScheduler
        pipe.scheduler = PNDMScheduler.from_config(scheduler_config)

    elif scheduler == "pndm_skip_prk_steps_true":

        new_scheduler_config = dict(pipe.scheduler.config)
        new_scheduler_config.update({"skip_prk_steps": True})

        from diffusers import PNDMScheduler
        pipe.scheduler = PNDMScheduler.from_config(new_scheduler_config)

    elif scheduler == "deis_multistep":

        from diffusers import DEISMultistepScheduler
        pipe.scheduler = DEISMultistepScheduler.from_config(scheduler_config)

    elif scheduler == "dpm_solver_sde":

        from diffusers import DPMSolverSDEScheduler
        pipe.scheduler = DPMSolverSDEScheduler.from_config(scheduler_config)

    elif scheduler == "uni_pc_multistep":

        from diffusers import UniPCMultistepScheduler
        pipe.scheduler = UniPCMultistepScheduler.from_config(scheduler_config)

    else:

        from diffusers import PNDMScheduler
        pipe.scheduler = PNDMScheduler.from_config(scheduler_config)

        scheduler_used = "pndm"



    return (
        scheduler_used
    )







#####################
#
#    Construct Refiner
#
#        Prepare the refiner.
#
#####################

def construct_refiner ():

    refiner_kwargs = {
        "use_safetensors": True
    }

    if device == "cuda":

        refiner_kwargs["variant"] = "fp16"
        refiner_kwargs["torch_dtype"] = torch.float16

    if use_custom_hugging_face_cache_dir == 1:

        refiner_kwargs["cache_dir"] = hugging_face_cache_dir

    refiner = DiffusionPipeline.from_pretrained(
        pretrained_model_name_or_path = hugging_face_refiner_partial_path,
        local_files_only = only_use_local_files_bool,
        **refiner_kwargs
    )

    if use_model_cpu_offload_for_refiner == 1:

        refiner.enable_model_cpu_offload()

    if use_xformers == 1:

        refiner.enable_xformers_memory_efficient_attention()

    refiner = refiner.to(device)

    if use_sequential_cpu_offload_for_refiner == 1:

        refiner.enable_sequential_cpu_offload()

    if use_default_attn_processor == 1:

        refiner.unet.set_default_attn_processor()

    if use_torch_compile_for_refiner == 1:
        refiner.unet = torch.compile(
            refiner.unet,
            mode = "reduce-overhead",
            fullgraph = True
        )

    if device == "cuda":
        torch.cuda.empty_cache()

    return (
        refiner
    )







#####################
#
#    Construct Upscaler
#
#        Prepare the upscaler.
#
#####################

def construct_upscaler ():

    upscaler_kwargs = {
        "use_safetensors": True
    }

    if device == "cuda":

        upscaler_kwargs["torch_dtype"] = torch.float16

    if use_custom_hugging_face_cache_dir == 1:

        upscaler_kwargs["cache_dir"] = hugging_face_cache_dir

    upscaler = DiffusionPipeline.from_pretrained(
        pretrained_model_name_or_path = hugging_face_upscaler_partial_path,
        local_files_only = only_use_local_files_bool,
        **upscaler_kwargs
    )

    if use_model_cpu_offload_for_upscaler == 1:

        upscaler.enable_model_cpu_offload()

    if use_xformers == 1:

        upscaler.enable_xformers_memory_efficient_attention()

    upscaler = upscaler.to(device)

    if use_sequential_cpu_offload_for_upscaler == 1:

        upscaler.enable_sequential_cpu_offload()

    if use_default_attn_processor == 1:

        upscaler.unet.set_default_attn_processor()

    if use_torch_compile_for_refiner == 1:
        upscaler.unet = torch.compile(
            upscaler.unet,
            mode = "reduce-overhead",
            fullgraph = True
        )

    if device == "cuda":
        torch.cuda.empty_cache()

    return (
        upscaler
    )







#####################
#
#    Update Prompt Info From Gallery
#
#        If you select an image in the image gallery, display the prompt
#        information for that image.
#
#####################

def update_prompt_info_from_gallery (
    gallery_data: gr.SelectData,
    image_gallery_array_state_value
):

    gallery_data_index = gallery_data.index

    output_image_gallery_field_update = gr.Gallery(
        selected_index = gallery_data_index
    )

    output_text_field_update = image_gallery_array_state_value[gallery_data_index]

    return {
        output_image_gallery_field: output_image_gallery_field_update,
        output_text_field: output_text_field_update
    }







#####################
#
#    Save Image File
#
#        If configured to do so.
#
#####################

def save_image_file (
    saved_image_path_and_file,
    image_to_return,
    add_generation_information_to_image,
    info_to_save_in_image
):

    if add_generation_information_to_image == 1:

        from PIL.PngImagePlugin import PngInfo

        saved_image_metadata = PngInfo()
        saved_image_metadata.add_text(
            "parameters",
            info_to_save_in_image
        )

        image_to_return_file = image_to_return.save(
            saved_image_path_and_file,
            pnginfo = saved_image_metadata
        )

    else:

        image_to_return_file = image_to_return.save(
            saved_image_path_and_file
        )







#####################
#
#    Create Image Generation Information
#
#        This information will be displayed on the page and within the image too
#        if you were to open it in a text file.
#
#####################

def create_image_generation_information (
    base_model_name_value,
    model_configuration_name_value,
    scheduler_used,
    scheduler_value,
    prompt_text,
    prompt_text_not_used_substring,
    negative_prompt_text,
    negative_prompt_text_not_used_substring,
    image_width,
    image_height,
    actual_seed,
    add_seed_into_pipe,
    guidance_scale,
    base_model_steps,
    display_xformers_usage_in_prompt_info,
    display_default_attn_processor_usage_in_prompt_info,
    display_diffusers_version_in_prompt_info,
    use_refiner,
    refiner_error,
    refining_denoise_start_field_value,
    denoising_end_applicable,
    refining_use_denoising_start_in_base_model_when_using_refiner_field_value,
    refining_base_model_output_to_refiner_is_in_latent_space_field_value,
    refining_steps_for_older_configuration_field_value,
    use_upscaler,
    upscaler_error,
    upscaling_steps,
    upscaled_image_width,
    upscaled_image_height,
    current_actual_total_base_model_steps,
    current_actual_total_refiner_steps,
    current_actual_total_upscaler_steps,
    generation_start_time,
    image_has_been_canceled,
    which_image
):

    #
    #
    #
    #    Prompt Information
    #
    #
    #

    info_about_prompt_lines_array = []

    if image_has_been_canceled == 1:

        info_about_prompt_lines_array.extend([
            "Image was canceled before completion. Some details below may not be accurate."
        ])

    if refiner_error == 1:

        info_about_prompt_lines_array.extend([
            "Refiner Error: An error occurred in the refining progress and was skipped. Some details below will not be accurate."
        ])

    if upscaler_error == 1:

        info_about_prompt_lines_array.extend([
            "Upscaler Error: An error occurred in the upscaling progress and was skipped. Some details below will not be accurate."
        ])

    info_about_prompt_lines_array.extend([
        "Prompt: " + prompt_text
    ])

    if len(negative_prompt_text) > 0:

        info_about_prompt_lines_array.extend([
            "Negative Prompt: " + negative_prompt_text
        ])

    dimensions_title = "Dimensions"

    if use_upscaler == 1:

        dimensions_title = "Original Dimensions"

    info_about_prompt_lines_array.extend([
        dimensions_title + ": " + str(image_width) + "x" + str(image_height) + " px"
    ])

    if use_upscaler == 1:

        info_about_prompt_lines_array.extend([
            "Upscaled Dimensions: " + str(upscaled_image_width) + "x" + str(upscaled_image_height) + " px"
        ])

    info_about_prompt_lines_array.extend([
        "Seed: " + str(actual_seed)
    ])

    nice_seed_added_to_generation = "No"

    if add_seed_into_pipe == 1:

        nice_seed_added_to_generation = "Yes"

    info_about_prompt_lines_array.extend([
        "Seed added to generation? " + nice_seed_added_to_generation
    ])

    if int(guidance_scale) > 0:

        info_about_prompt_lines_array.extend([
            "Guidance Scale: " + str(guidance_scale)
        ])

    if (
        (image_has_been_canceled == 0) or
        (base_model_steps == current_actual_total_base_model_steps)
    ):

        info_about_prompt_lines_array.extend([
            "Steps: " + str(base_model_steps)
        ])

    else:

        info_about_prompt_lines_array.extend([
            "Selected Steps: " + str(base_model_steps),
            "Actual Base Model Steps: " + str(current_actual_total_base_model_steps)
        ])

    nice_model_name = base_model_names_object[base_model_name_value] + " (" + model_configuration_links_object[model_configuration_name_value] + ")"

    nice_scheduler_name = scheduler_short_names_object[scheduler_used]

    if scheduler_value == "model_default":

        nice_scheduler_name += " (model default)"

    info_about_prompt_lines_array.extend([
        "Model: " + nice_model_name,
        "Scheduler/Sampler: " + nice_scheduler_name
    ])

    if (
        (use_refiner == 1) or
        (refiner_error == 1)
    ):

        refiner_usage_text = "Yes"

        if refiner_error == 1:

            refiner_usage_text = "No (an error prevented it from refining)\nHowever, selected refining details are still included."

        info_about_prompt_lines_array.extend([
            "Refiner Used? " + refiner_usage_text
        ])

        nice_refiner_denoise_start = str(refining_denoise_start_field_value)

        if denoising_end_applicable == 1:

            if refining_use_denoising_start_in_base_model_when_using_refiner_field_value == 1:

                info_about_prompt_lines_array.extend([
                    "Set \"denoising_end\" in base model generation? Yes",
                    "Base model denoise end %: " + nice_refiner_denoise_start
                ])

                if current_actual_total_base_model_steps > 0:

                    info_about_prompt_lines_array.extend([
                        "Actual Base Model Steps: " + formatted_number(current_actual_total_base_model_steps)
                    ])

            else:

                info_about_prompt_lines_array.extend([
                    "Set \"denoising_end\" in base model generation? No",
            ])

        info_about_prompt_lines_array.extend([
            "Refiner denoise start %: " + nice_refiner_denoise_start
        ])

        if current_actual_total_refiner_steps > 0:

            info_about_prompt_lines_array.extend([
                "Actual Refining Steps: " + formatted_number(current_actual_total_refiner_steps)
            ])

        if refining_base_model_output_to_refiner_is_in_latent_space_field_value == 1:

            info_about_prompt_lines_array.extend([
                "Base model output in latent space before refining? Yes",
            ])

        else:

            info_about_prompt_lines_array.extend([
                "Base model output in latent space before refining? No",
            ])

    if (
        (use_upscaler == 1) or
        (upscaler_error == 1)
    ):

        if use_upscaler == 1:

            info_about_prompt_lines_array.extend([
                "Upscaled (2x)? Yes",
                "Upscaler Steps: " + str(upscaling_steps)
            ])

        elif upscaler_error:

            info_about_prompt_lines_array.extend([
                "Upscaled (2x)? No (an error prevented it from upscaling)",
                "Selected Upscaler Steps: " + str(upscaling_steps)
            ])

    if log_generation_times == 1:

        generation_end_time = time.time()

        generation_time_in_seconds = (generation_end_time - generation_start_time)

        nice_generation_time = nice_elapsed_time(generation_time_in_seconds)

        info_about_prompt_lines_array.extend([
            "Time: " + nice_generation_time
        ])

    if len(prompt_text_not_used_substring) > 0:

        info_about_prompt_lines_array.extend([
            "End of Prompt Truncated: " + prompt_text_not_used_substring
        ])

    if len(negative_prompt_text_not_used_substring) > 0:

        info_about_prompt_lines_array.extend([
            "End of Negative Prompt Truncated: " + negative_prompt_text_not_used_substring
        ])

    if display_xformers_usage_in_prompt_info == 1:

        nice_xformers_usage = "No"

        if use_xformers == 1:

            nice_xformers_usage = "Yes"

            if include_transformers_version_in_prompt_info == 1:

                import transformers

                nice_xformers_usage += " (version " + str(transformers.__version__) + ")"

        info_about_prompt_lines_array.extend([
            "xFormers Used?: " + nice_xformers_usage
        ])

    if display_default_attn_processor_usage_in_prompt_info == 1:

        nice_default_attn_processor_usage = "No"

        if use_default_attn_processor == 1:

            nice_default_attn_processor_usage = "Yes"

        info_about_prompt_lines_array.extend([
            "Default AttnProcessor Used? " + nice_default_attn_processor_usage
        ])

    if display_diffusers_version_in_prompt_info == 1:

        try:

            import diffusers

            diffusers_version = diffusers.__version__

        except:

            diffusers_version = ""

        if diffusers_version:

            info_about_prompt_lines_array.extend([
                "Diffusers Version: " + str(diffusers_version)
            ])

    info_about_prompt = "\n".join(info_about_prompt_lines_array)

    return info_about_prompt







#####################
#
#    Load Image Preview
#
#        If needed, this loads the image preview.
#
#####################

def load_image_preview (
    user_id_state
):

    user_id_state_value = user_id_state.value

#    print ("Image preview check run", current_preview_image_user_id, user_id_state_value)

    if (
        (user_id_state_value > 0) and
        (user_id_state_value == current_preview_image_user_id)
    ):

#        print ("Image preview shown")

        return {
            output_image_preview_field: current_preview_image[0]
        }

    else:

        return None







#####################
#
#    Before Create Image Function
#
#        This is loaded before the image creation begins.
#
#####################

def before_create_image_function ():

    generate_image_button_update = gr.Button(
        value = generate_image_button_in_progress_text,
        variant = "secondary",
        interactive = False
    )

    output_text_field_update = gr.Textbox(
        visible = False
    )

    prompt_truncated_field_group_update = gr.Group(
        visible = False
    )

    prompt_truncated_field_update = gr.HTML(
        value = ""
    )

    negative_prompt_truncated_field_group_update = gr.Group(
        visible = False
    )

    negative_prompt_truncated_field_update = gr.HTML(
        value = ""
    )

    error_text_field_update = gr.HTML(
        value = "",
        visible = False
    )



    image_generation_id = int(random.randrange(0, 1000000000))

    image_generation_id_state_update = gr.State(
        value = image_generation_id
    )



    before_create_image_object = {
        generate_image_button: generate_image_button_update,
        output_text_field: output_text_field_update,
        prompt_truncated_field_group: prompt_truncated_field_group_update,
        prompt_truncated_field: prompt_truncated_field_update,
        negative_prompt_truncated_field_group: negative_prompt_truncated_field_group_update,
        negative_prompt_truncated_field: negative_prompt_truncated_field_update,
        error_text_field: error_text_field_update,
        image_generation_id_state: image_generation_id_state_update
    }

    if use_image_preview == 1:

        output_image_field_update = gr.Image(
            height = 100
        )

        output_image_gallery_field_update = gr.Gallery(
            height = 100
        )

        output_image_preview_field_row_update = gr.Row(
            visible = True
        )

        before_create_image_object.update({
            output_image_field: output_image_field_update,
            output_image_preview_field_row: output_image_preview_field_row_update
        })

    if enable_image_generation_cancellation == 1:

        cancel_image_button_row_update = gr.Row(
            visible = True
        )

        cancel_image_button_update = gr.Button(
            interactive = True
        )

        before_create_image_object.update({
            cancel_image_button_row: cancel_image_button_row_update,
            cancel_image_button: cancel_image_button_update
        })

    return before_create_image_object







#####################
#
#    After Create Image Function
#
#        This is loaded once image creation has completed.
#
#####################

def after_create_image_function ():

    generate_image_button_update = gr.Button(
        value = generate_image_button_normal_text,
        variant = "primary",
        interactive = True
    )

    output_text_field_update = gr.Textbox(
        visible = True
    )

    after_create_image_object = {
        generate_image_button: generate_image_button_update,
        output_text_field: output_text_field_update
    }

    if use_image_preview == 1:

        output_image_field_update = gr.Image(
            height = gradio_image_component_height
        )

        output_image_gallery_field_update = gr.Gallery(
            height = gradio_image_gallery_component_height
        )

        output_image_preview_field_row_update = gr.Row(
            visible = False
        )

        after_create_image_object.update({
            output_image_field: output_image_field_update,
            output_image_gallery_field: output_image_gallery_field_update,
            output_image_preview_field_row: output_image_preview_field_row_update
        })

    if enable_image_generation_cancellation == 1:

        generate_image_button_row_update = gr.Row(
            visible = True
        )

        cancel_image_button_row_update = gr.Row(
            visible = False
        )

        cancel_image_button_update = gr.Button(
            value = cancel_image_button_text,
            interactive = False
        )

        cancel_image_message_field_row_update = gr.Button(
            visible = False
        )

        cancel_image_message_field_update = gr.Button(
            value = ""
        )

        after_create_image_object.update({
            generate_image_button_row: generate_image_button_row_update,
            cancel_image_button_row: cancel_image_button_row_update,
            cancel_image_button: cancel_image_button_update,
            cancel_image_message_field_row: cancel_image_message_field_row_update,
            cancel_image_message_field: cancel_image_message_field_update
        })

    return after_create_image_object







#####################
#
#    Remove From Cancel Object
#
#        Remove user id and generation id once something is canceled.
#
#####################

def remove_from_cancel_object (
    user_id_state_value,
    image_generation_id_state_value
):

    if (
        (user_id_state_value in cancel_image_generation_ids_object) and
        (image_generation_id_state_value in cancel_image_generation_ids_object[user_id_state_value])
    ):

        cancel_image_generation_ids_object[user_id_state_value].remove(image_generation_id_state_value)

    if len(cancel_image_generation_ids_object[user_id_state_value]) == 0:

        cancel_image_generation_ids_object.pop(user_id_state_value, None)

    if image_generation_id_state_value in cancel_image_generation_times_object:

        cancel_image_generation_times_object.pop(image_generation_id_state_value, None)







#####################
#
#    Image Processing Is Canceled
#
#        Return True if image processing has been canceled.
#
#####################

def image_processing_is_canceled (
    user_id_state_value,
    image_generation_id_state_value
):

    global cancel_image_generation_ids_object

    image_processing_was_canceled = False

    if (
        enable_image_generation_cancellation and
        (user_id_state_value in cancel_image_generation_ids_object) and
        (image_generation_id_state_value in cancel_image_generation_ids_object[user_id_state_value])
    ):

        image_processing_was_canceled = True

    return image_processing_was_canceled







#####################
#
#   Create Image from Latents
#
#        
#
#####################

def create_image_from_latents (
    model_to_use,
    pipe,
    latents,
    generator,
    is_final_image
):

    if (
        (model_to_use == "sdxl") or
        (model_to_use == "sdxl_turbo")
    ):

        #    https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py

        # make sure the VAE is in float32 mode, as it overflows in float16
        needs_upcasting = pipe.vae.dtype == torch.float16 and pipe.vae.config.force_upcast

        if needs_upcasting:
            pipe.upcast_vae()
            latents = latents.to(next(iter(pipe.vae.post_quant_conv.parameters())).dtype)

        image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False)[0]

        # cast back to fp16 if needed
        if needs_upcasting:
            pipe.vae.to(dtype=torch.float16)

        if int(is_final_image) == 1:

            # apply watermark if available
            if pipe.watermark is not None:
                image = pipe.watermark.apply_watermark(image)

        image = pipe.image_processor.postprocess(image, output_type="pil")

        return image

    elif (
        (model_to_use == "sd_1_5_runwayml") or
        (model_to_use == "photoreal")
    ):

        #    https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

        image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False, generator=generator)[0]

        do_denormalize = [True] * image.shape[0]

        image = pipe.image_processor.postprocess(image, output_type="pil", do_denormalize=do_denormalize)

        return image

    elif model_to_use == "upscaler":

        #    https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py

        # make sure the VAE is in float32 mode, as it overflows in float16
        needs_upcasting = pipe.vae.dtype == torch.float16 and pipe.vae.config.force_upcast

        if needs_upcasting:
            pipe.upcast_vae()

        # Ensure latents are always the same type as the VAE
        latents = latents.to(next(iter(pipe.vae.post_quant_conv.parameters())).dtype)
        image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False)[0]

        # cast back to fp16 if needed
        if needs_upcasting:
            pipe.vae.to(dtype=torch.float16)

        if int(is_final_image) == 1:

            # apply watermark if available
            if pipe.watermark is not None:
                image = pipe.watermark.apply_watermark(image)

        do_denormalize = [True] * image.shape[0]

        image = pipe.image_processor.postprocess(image, output_type="pil", do_denormalize=do_denormalize)

        return image

    return ""







#####################
#
#    Create Preview Image
#
#        Add to a global variable to display preview image
#
#####################

def create_preview_image (
    model_to_use,
    user_id_state,
    pipe,
    latents,
    generator,
    temporary_extra
):

    is_final_image = 0

    pil_image = create_image_from_latents(
        model_to_use,
        pipe,
        latents,
        generator,
        is_final_image
    )

    if pil_image:

        global current_preview_image
        global current_preview_image_user_id

        current_preview_image = pil_image
        current_preview_image_user_id = user_id_state

#    For testing
#        for i, img in enumerate(pil_image):
#            filename = str(temporary_extra) + "_" + str(i) + ".png"
#            img.info = {"parameters": "Preview"}
#            img.save(filename)

        return None







#####################
#
#    Create Image Function
#
#        This is the main image creation function.
#
#####################

def create_image_function (

    base_model_field_index,
    prompt_text,
    negative_prompt_text,

    scheduler_index,

    image_width,
    image_height,
    guidance_scale,
    base_model_steps,
    base_model_steps_field_for_sdxl_turbo,
    actual_seed,
    add_seed_into_pipe,
    use_torch_manual_seed_but_do_not_add_to_pipe_field,

    refining_selection_field_value,
    refining_denoise_start_field_value,
    refining_use_denoising_start_in_base_model_when_using_refiner_field_value,
    refining_base_model_output_to_refiner_is_in_latent_space_field_value,
    refining_steps_for_older_configuration_field_value,

    upscaling_selection_field_value,
    upscaling_steps,

    image_gallery_array_state_value,
    prompt_information_array_state_value,

    last_model_configuration_name_selected_state_value,
    last_refiner_name_selected_state_value,
    last_upscaler_name_selected_state_value,

    stored_pipe_state,
    stored_refiner_state,
    stored_upscaler_state,

    save_base_image_when_using_refiner_field_value,
    save_refined_image_when_using_upscaler_field_value,

    user_id_state,
    image_generation_id_state,
    canceled_images_array_state,

    *model_configuration_dropdown_fields_array,

    progress = gr.Progress()

):

    image_generation_id_state_value = image_generation_id_state.value
    user_id_state_value = user_id_state.value

    global current_image_generation_id_in_progress

    current_image_generation_id_in_progress = image_generation_id_state_value



    global cancel_image_generation_ids_object
    global cancel_image_generation_times_object

    global current_preview_image
    global current_preview_image_user_id

    current_preview_image = None
    current_preview_image_user_id = 0



    if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

        #    User likely canceled while waiting in queue

        remove_from_cancel_object(user_id_state_value, image_generation_id_state_value)

        output_text_field_update = gr.Textbox()
        return {
            output_text_field: output_text_field_update
        }



    image_width = int(image_width)
    image_height = int(image_height)
    guidance_scale = float(guidance_scale)
    base_model_steps = int(base_model_steps)
    base_model_steps_field_for_sdxl_turbo = int(base_model_steps_field_for_sdxl_turbo)
    actual_seed = int(actual_seed)
    refining_denoise_start_field_value = rounded_number(refining_denoise_start_field_value, 2)
    refining_steps_for_older_configuration_field_value = int(refining_steps_for_older_configuration_field_value)
    upscaling_steps = int(upscaling_steps)



    base_model_name_value = base_model_array[base_model_field_index]



    position_in_array = 0

    model_configuration_field_object = {}

    for model_configuration_field_index in model_configuration_dropdown_fields_array:

        this_base_model = base_model_array[position_in_array]

        model_configuration_field_object[this_base_model] = model_configuration_field_index

        position_in_array += 1

    model_configuration_field_index = model_configuration_field_object[base_model_name_value]
    model_configuration_name_value = base_model_object_of_model_configuration_arrays[base_model_name_value][model_configuration_field_index]



    if base_model_name_value == "sdxl_turbo":

        negative_prompt_text = ""
        base_model_steps = base_model_steps_field_for_sdxl_turbo
        guidance_scale = 0



    current_estimated_total_base_model_steps = base_model_steps
    current_estimated_total_refiner_steps = 0
    current_estimated_total_upscaler_steps = upscaling_steps

    global current_actual_total_base_model_steps
    global current_actual_total_refiner_steps
    global current_actual_total_upscaler_steps

    current_actual_total_base_model_steps = 0
    current_actual_total_refiner_steps = 0
    current_actual_total_upscaler_steps = 0



    scheduler_value = schedulers_array[scheduler_index]



    if not base_model_valid(base_model_name_value):
        error_function("Base model is not valid.")

    if not model_configuration_valid(base_model_name_value, model_configuration_name_value):
        error_function("Model configuration is not valid.")

    if not prompt_valid(prompt_text):
        error_function("Prompt is not valid.")

    if not negative_prompt_valid(negative_prompt_text):
        error_function("Negative prompt is not valid.")

    if not scheduler_valid(scheduler_value):
        error_function("Scheduler/sampler is not valid.")

    if not width_valid(image_width):
        error_function("Image width is not valid.")

    if not height_valid(image_height):
        error_function("Image height is not valid.")

    if base_model_name_value != "sdxl_turbo":

        if not guidance_scale_valid(guidance_scale):
            error_function("Guidance scale is not valid.")

    if not steps_valid(base_model_steps, base_model_name_value):
        error_function("Steps option is not valid.")

    if not seed_valid(actual_seed):
        error_function("Seed is not valid.")



    add_seed_into_pipe = numerical_bool(add_seed_into_pipe)
    use_torch_manual_seed_but_do_not_add_to_pipe_field = numerical_bool(use_torch_manual_seed_but_do_not_add_to_pipe_field)



    refining_selection_field_value = numerical_bool(refining_selection_field_value)



    refining_use_denoising_start_in_base_model_when_using_refiner_field_value = numerical_bool(refining_use_denoising_start_in_base_model_when_using_refiner_field_value)
    refining_base_model_output_to_refiner_is_in_latent_space_field_value = numerical_bool(refining_base_model_output_to_refiner_is_in_latent_space_field_value)



    use_upscaler = numerical_bool(upscaling_selection_field_value)



    use_refiner = 0

    num_inference_steps_in_refiner = base_model_steps

    if refining_selection_field_value:

        use_refiner = 1

        if not refiner_denoise_start_valid(refining_denoise_start_field_value):
            error_function("Refiner denoise start is not valid.")

#        if refining_steps_for_older_configuration_field_value > 0:

            #    For older configrations. Doesn't reflect actual number of steps.

#            num_inference_steps_in_refiner = refining_steps_for_older_configuration_field_value



    if use_upscaler == 1:

        if not upscaling_steps_valid(upscaling_steps):
            error_function("Upscaling steps option is not valid.")



    if (
        (last_model_configuration_name_selected_state_value == "") or
        (model_configuration_name_value != last_model_configuration_name_selected_state_value)
    ):

        if (last_model_configuration_name_selected_state_value != ""):

            if "pipe" in globals():
                del pipe

            if device == "cuda":
               torch.cuda.empty_cache()

        if show_messages_in_command_prompt == 1:

            print ("Base model is loading.");

        progress(
            progress = 0,
            desc = "Base model is loading"
        )

        (
            pipe
        ) = construct_pipe(
            base_model_name_value,
            model_configuration_name_value
        )

        last_model_configuration_name_selected_state_value = model_configuration_name_value

    else:

        pipe = stored_pipe_state



    (
        scheduler_used
    ) = configure_scheduler(
        pipe,
        scheduler_value
    )



    if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):
        use_refiner = 0
        use_upscaler = 0



    if log_generation_times == 1:

        generation_start_time = time.time()



    #    Only 77 tokens are allowed in the prompt. 2 are reserved, meaning it is
    #    truncated to 75. This happens automatically, but we want to tell people
    #    that

    tokenizer = pipe.tokenizer

    max_token_length_of_model = pipe.tokenizer.model_max_length - number_of_reserved_tokens

    token_length_of_prompt_text = len(tokenizer.tokenize(prompt_text))
    token_length_of_negative_prompt_text = len(tokenizer.tokenize(negative_prompt_text))



    prompt_truncated_field_group_update = gr.Group(
        visible = False
    )

    prompt_truncated_field_update = gr.HTML(
        value = ""
    )

    negative_prompt_truncated_field_group_update = gr.Group(
        visible = False
    )

    negative_prompt_truncated_field_update = gr.HTML(
        value = ""
    )

    prompt_text_not_used_substring = ""
    negative_prompt_text_not_used_substring = ""

    truncated_prompts = 0

    partial_prompt_or_negative_prompt_length_too_long_message = ""

    if token_length_of_prompt_text > max_token_length_of_model:

        (
            prompt_text,
            prompt_text_not_used_substring
        ) = truncate_prompt(
            pipe,
            prompt_text
        )

        prompt_truncated_field_group_update = gr.Group(
            visible = True
        )

        prompt_truncated_field_update = gr.Textbox(
            value = prompt_text_not_used_substring
        )

        truncated_prompts += 1

        partial_prompt_or_negative_prompt_length_too_long_message += "prompt"

    if token_length_of_negative_prompt_text > max_token_length_of_model:

        (
            negative_prompt_text,
            negative_prompt_text_not_used_substring
        ) = truncate_prompt(
            pipe,
            negative_prompt_text
        )

        negative_prompt_truncated_field_group_update = gr.Group(
            visible = True
        )

        negative_prompt_truncated_field_update = gr.Textbox(
            value = negative_prompt_text_not_used_substring
        )

        truncated_prompts += 1

        if truncated_prompts == 2:
            partial_prompt_or_negative_prompt_length_too_long_message += " and "

        partial_prompt_or_negative_prompt_length_too_long_message += "negative prompt"

    if len(partial_prompt_or_negative_prompt_length_too_long_message) > 0:

        partial_prompt_or_negative_prompt_length_too_long_message += " was"

        if truncated_prompts == 2:
            partial_prompt_or_negative_prompt_length_too_long_message += " were"

        prompt_or_negative_prompt_length_too_long_message = "Note: Part of your " + partial_prompt_or_negative_prompt_length_too_long_message + " truncated automatically because it was too long."

        show_message(prompt_or_negative_prompt_length_too_long_message)



    if add_seed_into_pipe == 1:

        generator = torch.manual_seed(actual_seed)

    else:

        if use_torch_manual_seed_but_do_not_add_to_pipe_field == 1:

            torch.manual_seed(actual_seed)

        generator = None



    denoising_end_applicable = 0

    if base_model_name_value not in base_models_not_supporting_denoising_end_for_base_model_object:

        denoising_end_applicable = 1



    denoising_end_in_base_model_to_use = None

    output_type_in_base_model_to_use = "pil"

    if use_refiner == 1:

        if (
            (refining_use_denoising_start_in_base_model_when_using_refiner_field_value == 1) and
            (denoising_end_applicable == 1)
        ):

            denoising_end_in_base_model_to_use = refining_denoise_start_field_value

            current_estimated_total_base_model_steps = rounded_number(base_model_steps * refining_denoise_start_field_value)

            if current_estimated_total_base_model_steps < 1:

                current_estimated_total_base_model_steps = 1

        if refining_base_model_output_to_refiner_is_in_latent_space_field_value == 1:

            output_type_in_base_model_to_use = "latent"

        current_estimated_total_refiner_steps = rounded_number(base_model_steps - (base_model_steps * refining_denoise_start_field_value))

        if current_estimated_total_refiner_steps < 1:

            current_estimated_total_refiner_steps = 1



    upscaled_image_width = 0
    upscaled_image_height = 0

    if use_upscaler == 1:

        upscaled_image_width = int(image_width * 2)
        upscaled_image_height = int(image_height * 2)



    current_base_model_generation_start_time = 0

    global saved_final_base_model_pil_image_if_using_refiner
    saved_final_base_model_pil_image_if_using_refiner = None

    global upscaled_image_canceled
    global upscaled_image_canceled_latents

    upscaled_image_canceled = 0
#    upscaled_image_canceled_latents = None



    def callback_function_for_base_model_progress(
        callback_pipe,
        callback_step_index,
        callback_timestep,
        callback_kwargs
    ):

        callback_step_number = (int(callback_step_index) + 1)

        global current_actual_total_base_model_steps

        current_actual_total_base_model_steps += 1

        global current_base_model_generation_start_time

        if callback_step_number == 1:

            current_base_model_generation_start_time = time.time()

        seconds_per_step = 0

        if callback_step_number > 1:

            seconds_per_step = ((time.time() - current_base_model_generation_start_time) / int(callback_step_index))

            nice_time_per_step = nice_elapsed_time(seconds_per_step)

            base_model_progress_text = nice_time_per_step + " per step"

        else:

            base_model_progress_text = "Base model processing started"

        cancel_process = image_processing_is_canceled(user_id_state_value, image_generation_id_state_value)

        if cancel_process:

            pipe._interrupt = True

        if (
            use_image_preview and
            (
                ((int(callback_step_index) % image_preview_step_interval) == 0) or
                (seconds_per_step >= image_preview_seconds_interval)
            ) and
            (callback_step_number < current_estimated_total_base_model_steps)
        ):

            latents = callback_kwargs["latents"]

            temporary_extra = str(user_id_state_value) + "_base_model_" + str(callback_step_number)

            model_to_use = base_model_name_value

            is_final_image = 0

            pil_image = create_image_from_latents(
                model_to_use,
                pipe,
                latents,
                generator,
                is_final_image
            )

            create_preview_image(
                model_to_use,
                user_id_state_value,
                pipe,
                latents,
                generator,
                temporary_extra
            )

        if (
            cancel_process or
            (
                (base_model_steps == callback_step_number) and
                (use_refiner == 1)
            )
        ):

            global saved_final_base_model_pil_image_if_using_refiner

            latents = callback_kwargs["latents"]

            is_final_image = 1

            saved_final_base_model_pil_image_if_using_refiner = create_image_from_latents(
                base_model_name_value,
                pipe,
                latents,
                generator,
                is_final_image
            )

        if (
            (show_image_creation_progress_log == 1) and
            (callback_step_number <= current_estimated_total_base_model_steps)
        ):

            progress(
                progress = (
                    callback_step_number,
                    current_estimated_total_base_model_steps
                ),
                desc = base_model_progress_text,
                unit = "base model steps"
            )

        return {}

    callback_to_do_for_base_model_progress = callback_function_for_base_model_progress



    if (
        (show_image_creation_progress_log == 1) or
        enable_image_generation_cancellation or
        use_image_preview
    ):

        current_refiner_generation_start_time = 0

        def callback_function_for_refiner_progress(
            callback_pipe,
            callback_step_index,
            callback_timestep,
            callback_kwargs
        ):

            callback_step_number = (int(callback_step_index) + 1)

            global current_actual_total_refiner_steps

            current_actual_total_refiner_steps += 1

            global current_refiner_generation_start_time

            if callback_step_number == 1:

                current_refiner_generation_start_time = time.time()

            seconds_per_step = 0

            if callback_step_number > 1:

                seconds_per_step = ((time.time() - current_refiner_generation_start_time) / int(callback_step_index))

                nice_time_per_step = nice_elapsed_time(seconds_per_step)

                refiner_progress_text = nice_time_per_step + " per step"

            else:

                refiner_progress_text = "Refiner processing started"

            if (
                use_image_preview and
                (
                    ((int(callback_step_index) % image_preview_step_interval) == 0) or
                    (seconds_per_step >= image_preview_seconds_interval)
                ) and
                (callback_step_number < current_estimated_total_refiner_steps)
            ):

                latents = callback_kwargs["latents"]

                temporary_extra = str(user_id_state_value) + "_refiner_" + str(callback_step_number)

                model_to_use = base_model_name_value

                create_preview_image(
                    model_to_use,
                    user_id_state_value,
                    pipe,
                    latents,
                    generator,
                    temporary_extra
                )

            if (
                (show_image_creation_progress_log == 1) and
                (callback_step_number <= current_estimated_total_refiner_steps)
            ):

                progress(
                    progress = (
                        callback_step_number,
                        current_estimated_total_refiner_steps
                    ),
                    desc = refiner_progress_text,
                    unit = "est. refiner steps"
                )

            if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

                refiner._interrupt = True

            return {}

        callback_to_do_for_refiner_progress = callback_function_for_refiner_progress



        current_upscaler_generation_start_time = 0

        def callback_function_for_upscaler_progress(
            callback_step_index,
            callback_timestep,
            callback_latents
        ):

            callback_step_number = (int(callback_step_index) + 1)

            global current_actual_total_upscaler_steps

            current_actual_total_upscaler_steps += 1

            global current_upscaler_generation_start_time

            if callback_step_number == 1:

                current_upscaler_generation_start_time = time.time()

            seconds_per_step = 0

            if callback_step_number > 1:

                seconds_per_step = ((time.time() - current_upscaler_generation_start_time) / int(callback_step_index))

                nice_time_per_step = nice_elapsed_time(seconds_per_step)

                upscaler_progress_text = nice_time_per_step + " per step"

            else:

                upscaler_progress_text = "Upscaler processing started"

            if (
                use_image_preview and
                (
                    ((int(callback_step_index) % image_preview_step_interval) == 0) or
                    (seconds_per_step >= image_preview_seconds_interval)
                ) and
                (callback_step_number < current_estimated_total_upscaler_steps)
            ):

                latents = callback_latents

                temporary_extra = str(user_id_state_value) + "_upscale_" + str(callback_step_index)

                model_to_use = "upscaler"

                create_preview_image(
                    model_to_use,
                    user_id_state_value,
                    pipe,
                    latents,
                    generator,
                    temporary_extra
                )

            if (
                (show_image_creation_progress_log == 1) and
                (callback_step_number <= current_estimated_total_upscaler_steps)
            ):

                progress(
                    progress = (
                        callback_step_number,
                        current_estimated_total_upscaler_steps
                    ),
                    desc = upscaler_progress_text,
                    unit = "upscaler steps"
                )

            if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

                #    This doesn't work here. We raise an excpetion to cancel
                #    instead.

#                upscaler._interrupt = True

                global upscaled_image_canceled
                global upscaled_image_canceled_latents

                upscaled_image_canceled = 1

                latents = callback_latents

                upscaled_image_canceled_latents = latents

                raise Exception("end_at_this_step")

            return {}

        callback_to_do_for_upscaler_progress = callback_function_for_upscaler_progress



    else:

        callback_to_do_for_base_model_progress = None
        callback_to_do_for_refiner_progress = None
        callback_to_do_for_upscaler_progress = None



    if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

        if show_messages_in_command_prompt == 1:

            print ("Image processing was canceled before any part of the image could be created.");

        remove_from_cancel_object(user_id_state_value, image_generation_id_state_value)

        output_text_field_update = gr.Textbox()
        return {
            output_text_field: output_text_field_update
        }

#        output_image_field_update = gr.Image()
#        output_image_gallery_field_update = gr.Gallery()
#        output_text_field_update = gr.Textbox()
#        prompt_truncated_field_group_update = gr.Group()
#        prompt_truncated_field_update = gr.Textbox()
#        negative_prompt_truncated_field_group_update = gr.Group()
#        negative_prompt_truncated_field_update = gr.Textbox()
#        error_text_field_update = gr.Textbox()

#        return (
#            output_image_field_update,
#            output_image_gallery_field_update,
#            output_text_field_update,
#            prompt_truncated_field_group_update,
#            prompt_truncated_field_update,
#            negative_prompt_truncated_field_group_update,
#            negative_prompt_truncated_field_update,
#            last_model_configuration_name_selected_state_value,
#            last_refiner_name_selected_state_value,
#            last_upscaler_name_selected_state_value,
#            pipe,
#            stored_refiner_state,
#            stored_upscaler_state,
#            error_text_field_update
#        )



    task_info_for_progress = "Initial image creation has begun"
    if use_refiner == 1:
        if use_upscaler == 1:
            task_info_for_command_prompt = "Will create initial image, then refine and then upscale.\nInitial image steps..."
        else:
            task_info_for_command_prompt = "Will create initial image and then refine.\nInitial image steps..."
    else:
        if use_upscaler == 1:
            task_info_for_command_prompt = "Will create initial image and then upscale.\nInitial image steps..."
        else:
            task_info_for_command_prompt = "Will create image (no refining or upscaling).\nImage steps..."
            task_info_for_progress = "Image creation has begun"

    if show_messages_in_command_prompt == 1:

        print (task_info_for_command_prompt);

    if show_image_creation_progress_log == 1:

        progress(
            progress = 0,
            desc = task_info_for_progress
        )



    base_image = pipe(
        prompt = prompt_text,
        negative_prompt = negative_prompt_text,
        width = image_width,
        height = image_height,
        num_inference_steps = base_model_steps,
        guidance_scale = guidance_scale,
        num_images_per_prompt = 1,
        generator = generator,
        denoising_end = denoising_end_in_base_model_to_use,
        output_type = output_type_in_base_model_to_use,
        callback_on_step_end = callback_to_do_for_base_model_progress
    )


    have_upscaled_image = 0
    have_refined_image = 0

    base_image_for_next_step = base_image.images
    refined_image_for_next_step = None
    upscaled_image_for_next_step = None

    error_count = 0

    error_array = []

    if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

        use_refiner = 0
        use_upscaler = 0

    refiner = {}
    upscaler = {}

    refiner_error = 0
    upscaler_error = 0

    if (
        (use_refiner == 1) or
        (use_upscaler == 1)
    ):

        image_for_next_step = base_image_for_next_step

        if use_refiner == 1:

            try:

                if (last_refiner_name_selected_state_value == ""):

                    if show_messages_in_command_prompt == 1:

                        print ("Refiner is loading.");

                        progress(
                        progress = 0,
                        desc = "Refiner is loading"
                    )

                    refiner = construct_refiner()

                    last_refiner_name_selected_state_value = "refiner"

                else:

                    refiner = stored_refiner_state

            except BaseException as error_message:

                use_refiner = 0

                refiner_error = 1

                error_count += 1

                error_array.extend([
                    "Error " + str(error_count) + ": An error occurred while trying to load the refiner:\n" + str(error_message)
                ])

            if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

                use_refiner = 0
                use_upscaler = 0

            if use_refiner == 1:

                if show_messages_in_command_prompt == 1:

                    print ("Refiner steps...");

                if show_image_creation_progress_log == 1:

                    progress(
                        progress = 0,
                        desc = "Refining is beginning"
                    )

                try:

                    refined_image = refiner(
                        prompt = prompt_text,
                        negative_prompt = negative_prompt_text,
                        image = base_image_for_next_step,
                        num_inference_steps = num_inference_steps_in_refiner,
                        denoising_start = refining_denoise_start_field_value,
                        output_type = "pil",
                        generator = generator,
                        callback_on_step_end = callback_to_do_for_refiner_progress
                    )

                except BaseException as error_message:

                    #    User chose to refine image, but something went wrong.
                    #    We won't use the refiner. Since we will be using the
                    #    base image, where the output could be latents, we'll
                    #    need to handle that later.

                    use_refiner = 0
                    use_upscaler = 0

                    refiner_error = 1

                    error_count += 1

                    error_array.extend([
                        "Error " + str(error_count) + ": An error occurred while refining:\n" + str(error_message)
                    ])

            if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

                use_upscaler = 0

            if use_refiner == 1:

                #    User chose to refine image and it succeded.

                have_refined_image = 1

                refined_image_for_next_step = refined_image.images[0]

                image_for_next_step = refined_image_for_next_step

        if use_upscaler == 1:

            try:

                if (last_upscaler_name_selected_state_value == ""):

                    if show_messages_in_command_prompt == 1:

                        print ("Upscaler is loading.");

                    progress(
                        progress = 0,
                        desc = "Upscaler is loading"
                    )

                    upscaler = construct_upscaler()

                    last_upscaler_name_selected_state_value = "upscaler"

                else:

                    upscaler = stored_upscaler_state

            except BaseException as error_message:

                use_upscaler = 0

                upscaler_error = 1

                error_count += 1

                error_array.extend([
                    "Error " + str(error_count) + ": An error occurred while trying to load the upscaler:\n" + str(error_message)
                ])

            if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

                use_upscaler = 0

            if use_upscaler == 1:

                if show_messages_in_command_prompt == 1:

                    print ("Upscaler steps...");

                if show_image_creation_progress_log == 1:

                    progress(
                        progress = 0,
                        desc = "Upscaling is beginning"
                    )

                try:

                    upscaled_image = upscaler(
                        prompt = prompt_text,
                        negative_prompt = negative_prompt_text,
                        image = image_for_next_step,
                        num_inference_steps = upscaling_steps,
                        guidance_scale = 0,
                        generator = generator,
                        callback = callback_to_do_for_upscaler_progress,
                        callback_steps = 1
                    )

                except BaseException as error_message:

                    use_upscaler = 0

                    if str(error_message) != "end_at_this_step":

                        #    User chose to upscale image, but something went
                        #    wrong. We won't use the upscaler.
                        #
                        #    If "error_message" is "end_at_this_step", then the
                        #    user chose to cancel. We have to handle canceling
                        #    when upscaling differently. We don't want to treat
                        #    this as an actual error, so we don't do the below.

                        upscaler_error = 1

                        error_count += 1

                        error_array.extend([
                            "Error " + str(error_count) + ":\nAn error occurred while upscaling:\n" + str(error_message)
                        ])

            if use_upscaler == 1:

                #    User chose to upscale image and it succeded.

                have_upscaled_image = 1

                upscaled_image_for_next_step = upscaled_image.images[0]



    if upscaled_image_canceled == 1:

        latents = upscaled_image_canceled_latents

        temporary_extra = str(user_id_state_value) + "_upscale_canceled"

        model_to_use = "upscaler"

        is_final_image = 1

        pil_image = create_image_from_latents(
            model_to_use,
            pipe,
            latents,
            generator,
            is_final_image
        )

        for key, value in enumerate(pil_image):
            image_to_return = value

    elif have_upscaled_image == 1:

        image_to_return = upscaled_image_for_next_step

    elif have_refined_image == 1:

        image_to_return = refined_image_for_next_step

    else:

        if output_type_in_base_model_to_use == "latent":

            #    Image is in latent space

            for key, value in enumerate(saved_final_base_model_pil_image_if_using_refiner):
                image_to_return = value

        else:

            image_to_return = base_image_for_next_step[0]



    image_has_been_canceled = 0

    if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value):

        image_has_been_canceled = 1



    if device == "cuda":
        torch.cuda.empty_cache()



    which_image = ""



    image_generation_information = create_image_generation_information(
        base_model_name_value,
        model_configuration_name_value,
        scheduler_used,
        scheduler_value,
        prompt_text,
        prompt_text_not_used_substring,
        negative_prompt_text,
        negative_prompt_text_not_used_substring,
        image_width,
        image_height,
        actual_seed,
        add_seed_into_pipe,
        guidance_scale,
        base_model_steps,
        display_xformers_usage_in_prompt_info,
        display_default_attn_processor_usage_in_prompt_info,
        display_diffusers_version_in_prompt_info,
        use_refiner,
        refiner_error,
        refining_denoise_start_field_value,
        denoising_end_applicable,
        refining_use_denoising_start_in_base_model_when_using_refiner_field_value,
        refining_base_model_output_to_refiner_is_in_latent_space_field_value,
        refining_steps_for_older_configuration_field_value,
        use_upscaler,
        upscaler_error,
        upscaling_steps,
        upscaled_image_width,
        upscaled_image_height,
        current_actual_total_base_model_steps,
        current_actual_total_refiner_steps,
        current_actual_total_upscaler_steps,
        generation_start_time,
        image_has_been_canceled,
        which_image
    )



#    have_refined_image,
#    have_upscaled_image,

#Image before refining
#Image before upscaling



    output_text_field_update = gr.Textbox(
        value = image_generation_information,
        lines = 12
    )



    if add_generation_information_to_image == 1:

        #    Add generation info to image

        info_to_save_in_image = "\n-----------\nImage generation information:\n" + image_generation_information + "\n-----------\n"

        image_to_return.info = {"parameters": info_to_save_in_image}







#    save_base_image_when_using_refiner_field_value,
#    save_refined_image_when_using_upscaler_field_value,






    save_canceled_images = 0

    if (
        (auto_save_imagery == 1) and
        (
            (image_has_been_canceled == 0) or
            (save_canceled_images == 1)
        )
    ):



        if not os.path.exists(saved_images_dir):
            os.makedirs(saved_images_dir)

        yy_mm_dd_date_stamp = datetime.today().strftime('%Y-%m-%d')

        saved_images_date_dir = saved_images_dir + "/" + yy_mm_dd_date_stamp + "/"

        if not os.path.exists(saved_images_date_dir):
            os.makedirs(saved_images_date_dir)

        image_count = 1

        file_name_without_extension = yy_mm_dd_date_stamp + "-" + ('%04d' % image_count)

        saved_image_path_and_file = saved_images_date_dir + file_name_without_extension + ".png"

        while os.path.exists(saved_image_path_and_file):

            file_name_without_extension = yy_mm_dd_date_stamp + "-" + ('%04d' % image_count)

            saved_image_path_and_file = saved_images_date_dir + file_name_without_extension + ".png"

            image_count += 1

        final_image_file_path_and_file = saved_images_date_dir + file_name_without_extension + ".png"

        save_image_file(
            final_image_file_path_and_file,
            image_to_return,
            add_generation_information_to_image,
            info_to_save_in_image,
        )

        final_image_text_file_path_and_file = saved_images_date_dir + file_name_without_extension + ".txt"

        prompt_info_file_handle = open(final_image_text_file_path_and_file, "w")
        prompt_info_file_handle.writelines(image_generation_information)
        prompt_info_file_handle.close()



    output_image_field_update = gr.Image(
        value = image_to_return
    )

    image_gallery_array_state_value.insert(0, image_to_return)
    prompt_information_array_state_value.insert(0, image_generation_information)

    output_image_gallery_field_update = gr.Gallery(
        value = image_gallery_array_state_value,
        selected_index = 0
    )

    image_gallery_array_state_update = image_gallery_array_state_value

    prompt_information_array_state_update = prompt_information_array_state_value



    error_text_field_update = gr.Textbox(
        visible = False
    )

    if (len(error_array) > 0):

        error_information = "\n\n".join(error_array)

        error_text_field_update = gr.Textbox(
            value = error_information,
            visible = True
        )



    if show_messages_in_command_prompt == 1:

        print ("Image created.")



    last_model_configuration_name_selected_state_update = last_model_configuration_name_selected_state_value
    last_refiner_name_selected_state_update = last_refiner_name_selected_state_value
    last_upscaler_name_selected_state_update = last_upscaler_name_selected_state_value



    current_preview_image = None
    current_preview_image_user_id = 0



    if image_has_been_canceled == 1:

        remove_from_cancel_object(user_id_state_value, image_generation_id_state_value)



    return (
        output_image_field_update,
        output_image_gallery_field_update,
        output_text_field_update,
        prompt_truncated_field_group_update,
        prompt_truncated_field_update,
        negative_prompt_truncated_field_group_update,
        negative_prompt_truncated_field_update,
        last_model_configuration_name_selected_state_update,
        last_refiner_name_selected_state_update,
        last_upscaler_name_selected_state_update,
        pipe,
        refiner,
        upscaler,
        error_text_field_update
    )







#####################
#
#    Cancel Image Function
#
#        Image generation has not yet started and we need to cancel it.
#
#####################

def cancel_image_function(
    user_id_state,
    image_generation_id_state
):

    user_id_state_value = user_id_state.value
    image_generation_id_state_value = image_generation_id_state.value



    global image_generation_ids_object
    global cancel_image_generation_times_object

    if (user_id_state_value not in cancel_image_generation_ids_object):

        cancel_image_generation_ids_object[user_id_state_value] = []

    cancel_image_generation_ids_object[user_id_state_value].append(image_generation_id_state_value) 
    cancel_image_generation_times_object[image_generation_id_state_value] = time.time()



    cancel_image_button_update = gr.Button(
        value = cancel_image_button_in_progress_text,
        interactive = False
    )

    cancel_image_message_field_row_update = gr.Row(
        visible = True
    )

    cancel_image_object = {}

    if image_generation_id_state_value == current_image_generation_id_in_progress:

        #    If they cancel the image that is currently being generated, we will
        #    stop it after the next step is complete and return the image to
        #    them. They will not need to reload the page again to create another
        #    image.

        cancel_image_message_field_html = canceled_image_in_process_of_being_generated



        show_message("Your image generation is being canceled. After the current step in the process is complete, the cancellation will be complete.")



        generate_image_button_row_update = gr.Row(
            visible = False
        )

        cancel_image_object.update({
            generate_image_button_row: generate_image_button_row_update,
            cancel_image_message_field_row: cancel_image_message_field_row_update
        })

    else:

        #    If they cancel there place in the queue, we have to make them
        #    reload the page if they want to generate another image.

        cancel_image_message_field_html = canceled_image_in_queue_message

        generate_image_button_update = gr.Button(
            value = generate_image_button_normal_text,
            interactive = False
        )

        output_text_field_update = gr.Textbox(
            visible = True
        )

        output_image_preview_field_row_update = gr.Row(
            visible = False
        )

        cancel_image_button_row_update = gr.Row(
            visible = False
        )

        cancel_image_object.update({
            generate_image_button: generate_image_button_update,
            output_text_field: output_text_field_update,
            output_image_preview_field_row: output_image_preview_field_row_update,
            cancel_image_button_row: cancel_image_button_row_update,
            cancel_image_message_field_row: cancel_image_message_field_row_update
        })



    cancel_image_message_field_update = gr.Button(
        value = cancel_image_message_field_html
    )

    cancel_image_object.update({
        cancel_image_button: cancel_image_button_update,
        cancel_image_message_field_row: cancel_image_message_field_row_update,
        cancel_image_message_field: cancel_image_message_field_update
    })



    return cancel_image_object







#####################
#
#    Cancel Image Processing
#
#        When running on Windows, this is an attempt at closing the command
#        prompt from the web display. It's really not worth having this. You can
#        just close the prompt. I would like a nice way to cancel image
#        creation, but couldn't figure that out.
#
#####################

def close_command_prompt():

    #    I simply don't know how to stop the image generation without closing
    #    the command prompt. Doing that requires the code below twice for some
    #    reason.
    #
    #    Method:
    #        https://stackoverflow.com/questions/67146623/how-to-close-the-command-prompt-from-python-script-directly

    gr.Warning("The command prompt window has been closed. Any image generation in progress has been stopped. To generate any other images, you will need to launch the command prompt again.")

    os.system('title kill_window')

    os.system(f'taskkill /f /fi "WINDOWTITLE eq kill_window"')
    os.system(f'taskkill /f /fi "WINDOWTITLE eq kill_window"')







#####################
#
#    Download Data From HuggingFace
#
#        This will download a lot of data at once rather than waiting until you
#        use each model.
#
#####################

def download_data_from_huggingface(
    download_data_option
):

    if (
        (script_being_run_on_hugging_face == 0) and
        ("HF_HUB_OFFLINE" in os.environ) and
        (int(os.environ["HF_HUB_OFFLINE"]) == 0)
    ):

        data_to_get_partial_message = "the default model configuration defined in \"base_model_model_configuration_defaults_object\" for that model will be downloaded. It accesses data that is linked in \"model_configuration_links_object\"."

        if download_data_option == "2":

            data_to_get_partial_message = "all model data, for each model configuration, will be downloaded. This is defined in \"base_model_object_of_model_configuration_arrays\" and accesses data that is linked in \"model_configuration_links_object\"."

        download_data_message = "For each model in the model dropdown (\"base_model_array\"), " + data_to_get_partial_message + " That could easily be dozens of gigabytes of data or more that is about to be downloaded. If you want to stop the download, close the command prompt."

        print (download_data_message)

        data_links_downloaded_object = {}

        for this_base_model in base_model_array:

            base_model_name_value = this_base_model

            if download_data_option == "1":

                default_model_configuration_for_this_base_model = base_model_model_configuration_defaults_object[this_base_model]

                print ("Downloading/loading \"" + this_base_model + "\" model data for \"" + default_model_configuration_for_this_base_model + "\"...")

                model_configuration_name_value = default_model_configuration_for_this_base_model

                construct_pipe (
                    base_model_name_value,
                    model_configuration_name_value
                )

            else:

                for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:

                    if (
                        (this_model_configuration in model_configuration_names_object) and
                        (this_model_configuration in model_configuration_links_object)
                    ):

                        model_configuration_name_value = this_model_configuration
                        model_configuration_link_value = model_configuration_links_object[this_model_configuration]

                        if model_configuration_link_value not in data_links_downloaded_object:

                            print ("Downloading/loading \"" + this_base_model + "\" model data from \"" + model_configuration_link_value + "\"...")

                            construct_pipe (
                                base_model_name_value,
                                model_configuration_name_value
                            )

                            data_links_downloaded_object[model_configuration_link_value] = 1

        print ("Downloading/loading refiner data...")

        construct_refiner()

        print ("Downloading/loading upscaler data...")

        construct_upscaler()

        print ("The data has been downloaded.")

    else:

        error_function("In order to download model data, \"HF_HUB_OFFLINE\" must be set to \"0\" in the Windows .bat file that launched this script.")







#####################
#
#    Get Query Params
#
#        Get variables from the url of the page and update the display to
#        reflect them.
#
#####################

def get_query_params(
    request: gr.Request
):

    raw_url_params = str(request.query_params)

    import urllib.parse
 
    unprocessed_url_object = urllib.parse.parse_qs(raw_url_params)
 
    url_object = {}

    for url_param_key in unprocessed_url_object:

        url_param_value = unprocessed_url_object[url_param_key][0]

        if len(url_param_value) > 0:

            url_param_key = str(url_param_key)

            url_param_key_lowercase = url_param_key.lower()

            url_object[url_param_key_lowercase] = str(unprocessed_url_object[url_param_key][0]).lower()



    field_object = {}



    user_id_number = int(random.randrange(0, 1000000000))

    user_id_state_update = gr.State(user_id_number)

    field_object.update({user_id_state: user_id_state_update})



    base_model_name_value = default_base_model

    base_model_field_key_in_url = "model"

    if base_model_field_key_in_url in url_object:

        base_model_field_in_url = url_object[base_model_field_key_in_url].lower()

        if base_model_valid(base_model_field_in_url):

            base_model_name_value = base_model_field_in_url

            base_model_nicely_named_value = base_model_names_object[base_model_name_value]

            field_object.update({base_model_field: base_model_nicely_named_value})



    download_data_key_in_url = "download_data"

    if download_data_key_in_url in url_object:

        download_data_option_in_url = str(url_object[download_data_key_in_url])

        if (
            (download_data_option_in_url == "1") or
            (download_data_option_in_url == "2")
        ):

            download_data_from_huggingface(download_data_option_in_url)



    model_configuration_key_in_url = "model_config"

    model_configuration_in_url = ""

    if model_configuration_key_in_url in url_object:

        model_configuration_in_url = url_object[model_configuration_key_in_url].lower()

    for this_base_model in base_model_array:

        if base_model_name_value == this_base_model:

            model_configuration_name_value = base_model_model_configuration_defaults_object[this_base_model]

        if len(model_configuration_in_url) > 0:

            for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:

                if model_configuration_valid(base_model_name_value, model_configuration_in_url):

                    model_configuration_name_value = this_model_configuration

                    field_object.update({initial_model_configuration_name_selected_state: model_configuration_in_url})



    prompt_field_key_in_url = "prompt"

    if prompt_field_key_in_url in url_object:

        prompt_field_in_url = url_object[prompt_field_key_in_url].lower()

        if prompt_valid(prompt_field_in_url):

            field_object.update({prompt_field: prompt_field_in_url})



    negative_prompt_field_key_in_url = "neg_prompt"

    if negative_prompt_field_key_in_url in url_object:

        negative_prompt_field_in_url = url_object[negative_prompt_field_key_in_url].lower()

        if prompt_valid(negative_prompt_valid):

            field_object.update({negative_prompt_field: negative_prompt_field_in_url})



    scheduler_field_key_in_url = "scheduler"

    if scheduler_field_key_in_url in url_object:

        scheduler_field_in_url = url_object[scheduler_field_key_in_url].lower()

        if scheduler_valid(scheduler_field_in_url):

            scheduler_name_value = scheduler_field_in_url

            scheduler_nicely_named_value = scheduler_long_names_object[scheduler_name_value]

            field_object.update({scheduler_field: scheduler_nicely_named_value})



    image_width_field_key_in_url = "width"

    if image_width_field_key_in_url in url_object:

        image_width_field_in_url = str(url_object[image_width_field_key_in_url])

        if width_valid(image_width_field_in_url):

            field_object.update({image_width_field: image_width_field_in_url})



    image_height_field_key_in_url = "height"

    if image_height_field_key_in_url in url_object:

        image_height_field_in_url = str(url_object[image_height_field_key_in_url])

        if height_valid(image_height_field_in_url):

            field_object.update({image_height_field: image_height_field_in_url})



    guidance_scale_field_key_in_url = "guidance"

    if guidance_scale_field_key_in_url in url_object:

        guidance_scale_field_in_url = str(url_object[guidance_scale_field_key_in_url])

        if guidance_scale_valid(guidance_scale_field_in_url):

            field_object.update({guidance_scale_field: guidance_scale_field_in_url})



    steps_key_in_url = "steps"

    if steps_key_in_url in url_object:

        steps_in_url = str(url_object[steps_key_in_url])

        if steps_valid(steps_in_url, base_model_name_value):

            if base_model_name_value == "sdxl_turbo":

                field_object.update({base_model_steps_field_for_sdxl_turbo_field: steps_in_url})

            else:

                field_object.update({base_model_steps_field: steps_in_url})



    seed_field_key_in_url = "seed"

    if seed_field_key_in_url in url_object:

        seed_field_in_url = url_object[seed_field_key_in_url]

        if seed_valid(seed_field_in_url):

            field_object.update({seed_field: seed_field_in_url})



    add_seed_key_in_url = "add_seed"

    add_seed_to_generation = None

    if add_seed_key_in_url in url_object:

        add_seed_in_url = url_object[add_seed_key_in_url].lower()

        add_seed_to_generation = True

        if (
            (add_seed_in_url == "0") or
            (add_seed_in_url == "n") or
            (add_seed_in_url == "no") or
            (add_seed_in_url == "false")
        ):

            add_seed_to_generation = False

        field_object.update({add_seed_into_pipe_field: add_seed_to_generation})



    use_torch_manual_seed_but_not_in_generator_key_in_url = "use_torch_manual_seed_but_not_in_generator"

    if use_torch_manual_seed_but_not_in_generator_key_in_url in url_object:

        use_torch_manual_seed_but_not_in_generator_in_url = url_object[use_torch_manual_seed_but_not_in_generator_key_in_url].lower()

        use_torch_manual_seed_but_not_in_generator_to_generation = False

        if (
            (use_torch_manual_seed_but_not_in_generator_in_url == "1") or
            (use_torch_manual_seed_but_not_in_generator_in_url == "y") or
            (use_torch_manual_seed_but_not_in_generator_in_url == "yes") or
            (use_torch_manual_seed_but_not_in_generator_in_url == "true")
        ):

            use_torch_manual_seed_but_not_in_generator_to_generation = True

        if (
            add_seed_to_generation and
            add_seed_to_generation == True
        ):

            use_torch_manual_seed_but_not_in_generator_to_generation = False

        field_object.update({use_torch_manual_seed_but_do_not_add_to_pipe_field: use_torch_manual_seed_but_not_in_generator_to_generation})



    refiner_key_in_url = "refiner"

    if refiner_key_in_url in url_object:

        refiner_in_url = url_object[refiner_key_in_url].lower()

        refiner_in_url_formatted = "No"

        if (
            (refiner_in_url == "1") or
            (refiner_in_url == "y") or
            (refiner_in_url == "yes") or
            (refiner_in_url == "true")
        ):

            refiner_in_url_formatted = "Yes"

        field_object.update({refining_selection_field: refiner_in_url_formatted})



    refiner_denoise_start_key_in_url = "denoise_start"

    if refiner_denoise_start_key_in_url in url_object:

        refiner_denoise_start_in_url = str(url_object[refiner_denoise_start_key_in_url])

        if refiner_denoise_start_valid(refiner_denoise_start_in_url):

            field_object.update({refining_denoise_start_field: refiner_denoise_start_in_url})



    refining_steps_key_in_url = "refiner_steps"

    if refining_steps_key_in_url in url_object:

        refining_steps_in_url = str(url_object[refining_steps_key_in_url])

        if refining_steps_valid(refining_steps_in_url):

            field_object.update({refining_steps_for_older_configuration_field: refining_steps_in_url})



    use_denoising_start_in_base_model_when_using_refiner_key_in_url = "use_denoise_end"

    if use_denoising_start_in_base_model_when_using_refiner_key_in_url in url_object:

        use_denoising_start_in_base_model_when_using_refiner_in_url = url_object[use_denoising_start_in_base_model_when_using_refiner_key_in_url].lower()

        use_denoising_start_in_base_model_when_using_refiner_bool = True

        if (
            (use_denoising_start_in_base_model_when_using_refiner_in_url == "0") or
            (use_denoising_start_in_base_model_when_using_refiner_in_url == "n") or
            (use_denoising_start_in_base_model_when_using_refiner_in_url == "no") or
            (use_denoising_start_in_base_model_when_using_refiner_in_url == "false")
        ):

            use_denoising_start_in_base_model_when_using_refiner_bool = False

        field_object.update({refining_use_denoising_start_in_base_model_when_using_refiner_field: use_denoising_start_in_base_model_when_using_refiner_bool})



    base_model_output_to_refiner_is_in_latent_space_key_in_url = "latent_space_before_refiner"

    if base_model_output_to_refiner_is_in_latent_space_key_in_url in url_object:

        base_model_output_to_refiner_is_in_latent_space_in_url = url_object[base_model_output_to_refiner_is_in_latent_space_key_in_url].lower()

        base_model_output_to_refiner_is_in_latent_space_bool = True

        if (
            (base_model_output_to_refiner_is_in_latent_space_in_url == "0") or
            (base_model_output_to_refiner_is_in_latent_space_in_url == "n") or
            (base_model_output_to_refiner_is_in_latent_space_in_url == "no") or
            (base_model_output_to_refiner_is_in_latent_space_in_url == "false")
        ):

            base_model_output_to_refiner_is_in_latent_space_bool = False

        field_object.update({refining_base_model_output_to_refiner_is_in_latent_space_field: base_model_output_to_refiner_is_in_latent_space_bool})



    upscaler_key_in_url = "upscaler"

    if upscaler_key_in_url in url_object:

        upscaler_in_url = url_object[upscaler_key_in_url].lower()

        upscaler_in_url_formatted = "No"

        if (
            (upscaler_in_url == "1") or
            (upscaler_in_url == "y") or
            (upscaler_in_url == "yes") or
            (upscaler_in_url == "true")
        ):

            upscaler_in_url_formatted = "Yes"

        field_object.update({upscaling_selection_field: upscaler_in_url_formatted})



    upscaling_steps_key_in_url = "upscaler_steps"

    if upscaling_steps_key_in_url in url_object:

        upscaling_steps_in_url = str(url_object[upscaling_steps_key_in_url])

        if upscaling_steps_valid(upscaling_steps_in_url):

            field_object.update({upscaling_num_inference_steps_field: upscaling_steps_in_url})



    generate_image_button_update = gr.Button(
        interactive = True
    )

    field_object.update({generate_image_button: generate_image_button_update})



    return field_object







#####################
#
#    Set Base Model and Model Configuration from query_params
#
#        We need to handle this separate because the model configuration needs
#        to return those field components in order. They are not named as they
#        are dynamic.
#
#####################

def set_base_model_and_model_configuration_from_query_params(
    base_model_field_index,
    initial_model_configuration_name_selected_state_value,
    *model_configuration_dropdown_fields_array
):

    base_model_name_value = base_model_array[base_model_field_index]

    model_configuration_name_value_for_selected_base_model = initial_model_configuration_name_selected_state_value

    model_configuration_dropdown_fields_array = []

    for this_base_model in base_model_array:

        model_configuration_name_default_value_for_this_base_model = base_model_model_configuration_defaults_object[this_base_model]

        for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:

            if (
                (base_model_name_value == this_base_model) and
                (model_configuration_name_value_for_selected_base_model == this_model_configuration)
            ):

                model_configuration_name_default_value_for_this_base_model = model_configuration_name_value_for_selected_base_model

            this_configuration_field_default_value = model_configuration_names_object[model_configuration_name_default_value_for_this_base_model]

        this_configuration_field = gr.Dropdown(
            value = this_configuration_field_default_value
        )

        model_configuration_dropdown_fields_array.append(this_configuration_field)

    base_model_and_model_configuration_return_outputs = []

    for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array:

        base_model_and_model_configuration_return_outputs.append(
            this_model_configuration_dropdown_field
        )

    return base_model_and_model_configuration_return_outputs







###############################################################################
###############################################################################
#
#
#
#
#
#
#    Create Web Display
#
#
#
#
#
#
###############################################################################
###############################################################################



css_to_use = """

/* Hide border on image preview */

.generating {border: none !important;}

/* Hide footer */

footer
{
    display: none !important;
}

/* Dropdowns */

.sp_dropdown ul
{
    min-width: auto;
    max-width: fit-content !important;
    max-height: 250px !important;
}

/* Checkboxes */

.sp_checkbox label
{
    align-items: start;
}

.sp_checkbox label input
{
    margin-top: 3px;
}
/* Scrollbar on prompt information */

.textbox_vertical_scroll textarea
{
    resize: none;
    overflow-y: auto !important;
}
     
.textbox_vertical_scroll textarea::-webkit-scrollbar
{
    width: 15px;
}

.textbox_vertical_scroll textarea::-webkit-scrollbar-track
{
    background-color: rgb(245, 245, 245);
}

.textbox_vertical_scroll textarea::-webkit-scrollbar-track:hover
{
    background-color: rgb(242, 242, 242);
}

.textbox_vertical_scroll textarea::-webkit-scrollbar-thumb
{
    background-color: rgb(214, 214, 214);
    width: 100%;
    height: 60px;
    max-height: 80%;
    border: 1px solid rgb(224, 224, 224);
}

.textbox_vertical_scroll textarea::-webkit-scrollbar-thumb:hover
{
    background-color: rgb(184, 184, 184);
}

.image_scaling button.image-button img
{
    object-fit: scale-down;
}

/* Size of image for image preview */

#image_preview_id div.image-container
{
    width: 100%;
}

#image_preview_id img
{
    max-height: 300px;
    max-width: 300px;
#    max-height: 100%;
#    max-width: 100%;
    object-fit: scale-down;
    margin: 0 auto;
}

"""

with gr.Blocks(
    title = "Spaghetti AI",
    css = css_to_use,
    theme = gr.themes.Default(
        spacing_size = "sm", # sm, md, lg
        radius_size = "sm" # none, sm, md, lg
    ),
    analytics_enabled = False
) as sd_interface:

    #    Variables to store for user session

    image_gallery_array_state = gr.State([])

    prompt_information_array_state = gr.State([])

    initial_model_configuration_name_selected_state = gr.State("")

    last_model_configuration_name_selected_state = gr.State("")
    last_refiner_name_selected_state = gr.State("")
    last_upscaler_name_selected_state = gr.State("")

    stored_pipe_state = gr.State({})
    stored_refiner_state = gr.State({})
    stored_upscaler_state = gr.State({})

    user_id_state = gr.State(0)
    image_generation_id_state = gr.State("")
    canceled_images_array_state = gr.State([])

    gr.Markdown(opening_html)

    with gr.Row():

        with gr.Column(
            scale = 3,
            min_width = 200
        ):

            with gr.Row(
                visible = True
            ) as generate_image_button_row:

                generate_image_button = gr.Button(
                    value = generate_image_button_normal_text,
                    variant = "primary",
                    interactive = False
                )

            with gr.Row(
                visible = False
            ) as cancel_image_button_row:

                cancel_image_button = gr.Button(
                    value = cancel_image_button_text,
                    variant = "stop",
                    interactive = False
                )

            with gr.Row(
                visible = False
            ) as cancel_image_message_field_row:

                cancel_image_message_field = gr.HTML(
                    value = ""
                )

            with gr.Group():

                with gr.Row():

                    prompt_field = gr.Textbox(
                        label = "Prompt (77 token limit):",
#                        label = "Prompt:",
#                        info = "77 token limit",
                        value = default_prompt
                    )

                with gr.Row(
                    elem_id = "negative_prompt_field_row_id",
                    visible = default_negative_prompt_field_row_visibility
                ):

                    negative_prompt_field = gr.Textbox(
                        label = "Negative Prompt (77 token limit):",
#                        label = "Negative Prompt:",
#                        info = "77 token limit",
                        value = default_negative_prompt
                    )

                with gr.Row(
                    elem_id = "negative_prompt_for_sdxl_turbo_field_row_id",
                    visible = default_negative_prompt_for_sdxl_turbo_field_row_visibility
                ):

                    negative_prompt_for_sdxl_turbo_field = gr.HTML(
                        value = "<div style=\"padding: 10px; text-align: center; background: #fff;\">Negative prompt is not used for SDXL Turbo.</div>"
                    )

            with gr.Group(
                visible = refiner_group_visible
            ):

                refiner_label_text = "Refiner"

                with gr.Accordion(
                    label = refiner_label_text,
                    open = refiner_accordion_open,
                    visible = refiner_accordion_visible
                ) as refiner_accordion:

                    with gr.Row():

                        refining_selection_field = gr.Radio(
                            label = "Use refiner?",
                            choices = ["Yes", "No"],
                            value = default_refine_option,
                            container = True
                        )

                    with gr.Row():

                        refining_denoise_start_field = gr.Slider(
                            label = "Refiner denoise start %",
                            minimum = minimum_refiner_denoise_start,
                            maximum = maximum_refiner_denoise_start,
                            value = default_refiner_denoise_start,
                            step = refiner_denoise_start_input_slider_steps
                        )

                    with gr.Row(
                        elem_id = "refining_use_denoising_start_in_base_model_when_using_refiner_field_row_id",
                        visible = default_refining_use_denoising_start_in_base_model_when_using_refiner_field_row_visibility
                    ):

                        refining_use_denoising_start_in_base_model_when_using_refiner_field = gr.Checkbox(
                            label = "Use \"denoising_start\" value as \"denoising_end\" value in base model generation when using refiner",
                            value = default_use_denoising_start_in_base_model_when_using_refiner_is_selected,
                            interactive = True,
                            container = True,
                            elem_classes = "sp_checkbox"
                        )

                    with gr.Row():

                        refining_base_model_output_to_refiner_is_in_latent_space_field = gr.Checkbox(
                            label = "Base model output in latent space instead of PIL image when using refiner",
                            value = default_base_model_output_to_refiner_is_in_latent_space_is_selected,
                            interactive = True,
                            container = True,
                            elem_classes = "sp_checkbox"
                        )









# Might add this again eventually (not needed, but to match older configuration)

                    model_configuration_include_refiner_number_of_steps_object = {}

                    refining_steps_for_older_configuration_field_visible = False

                    if default_model_configuration in model_configuration_include_refiner_number_of_steps_object:

                        refining_steps_for_older_configuration_field_visible = True

                    with gr.Row(
                        elem_id = "refining_steps_for_older_configuration_field_row_id",
                        visible = refining_steps_for_older_configuration_field_visible
                    ):

                        refining_steps_for_older_configuration_field = gr.Slider(
                            label = "Refining steps:",
                            minimum = 1,
                            maximum = maximum_refining_steps_for_online_config_field,
                            value = default_refining_steps_for_online_config_field,
                            step = 1
                        )









            with gr.Group(
                visible = upscaler_group_visible
            ):

                with gr.Accordion(
                    label = "Upscaler",
                    open = upscaler_accordion_open,
                    visible = upscaler_group_visible
                ):

                    #
                    #
                    #
                    #    Upscaler
                    #
                    #
                    #

                    with gr.Row():

                        upscaling_selection_field = gr.Radio(
                            label = "Upscale by 2x?",
                            choices = ["Yes", "No"],
                            value = default_upscale_option,
                            container = True
                        )

                    with gr.Row():

                        upscaling_num_inference_steps_field = gr.Slider(
                            label = "Upscaler number of steps",
                            minimum = 1,
                            maximum = maximum_upscaler_steps,
                            value = default_upscaler_steps,
                            step = 1
                        )

            if (
                (enable_refiner == 1) or
                (enable_upscaler == 1)
            ):

                refiner_and_upscaler_text_field = gr.HTML(
                    value = "<div id=\"refiner_and_upscaler_info_message_div_id\" style=\"text-align: center;\">" + default_refiner_and_upscaler_status_text + "</div>"
                )

        with gr.Column(
            scale = 2,
            min_width = 200
        ):

            with gr.Group():

                with gr.Row():

                    base_model_field = gr.Dropdown(
                        label = "Model:",
                        choices = default_base_model_choices_array,
                        value = default_base_model_nicely_named_value,
                        type = "index",
                        filterable = False,
                        interactive = True,
                        elem_classes = "sp_dropdown"
                    )

                model_configuration_dropdown_field_values_for_js = ""

                model_configuration_dropdown_fields_array = []

                for this_base_model in base_model_array:

                    this_model_configuration_choices_array = []

                    for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]:

                        this_model_configuration_choices_array.append(
                            model_configuration_names_object[this_model_configuration]
                        )

                    this_configuration_field_row_visibility = False

                    if (
                        (this_base_model == default_base_model) and
                        (allow_other_model_versions == 1)
                    ):

                        this_configuration_field_row_visibility = True

                    this_configuration_field_default_value = model_configuration_names_object[base_model_model_configuration_defaults_object[this_base_model]]

                    this_configuration_field_default_value_for_js = this_configuration_field_default_value
                    this_configuration_field_default_value_for_js = this_configuration_field_default_value_for_js.replace("\"", "\\\"")

                    model_configuration_dropdown_field_values_for_js += "\"" + this_base_model + "\": \"" + this_configuration_field_default_value_for_js + "\","

                    with gr.Row(
                        elem_id = "model_configuration_field_" + this_base_model + "_row_id",
                        visible = this_configuration_field_row_visibility
                    ):

                        this_configuration_field = gr.Dropdown(
                            label = "Version:",
                            choices = this_model_configuration_choices_array,
                            value = this_configuration_field_default_value,
                            type = "index",
                            filterable = False,
                            interactive = True,
                            elem_classes = "sp_dropdown"
                        )

                        model_configuration_dropdown_fields_array.append(this_configuration_field)

                with gr.Row():

                    scheduler_field = gr.Dropdown(
                        label = "Scheduler / Sampler:",
                        choices = default_scheduler_choices_array,
                        value = default_scheduler_nicely_named_value,
                        type = "index",
                        filterable = False,
                        interactive = True,
                        elem_classes = "sp_dropdown"
                    )

                with gr.Row():

                    image_width_field = gr.Slider(
                        label = "Width:",
                        minimum = minimum_width,
                        maximum = maximum_width,
                        value = default_width,
                        step = width_and_height_input_slider_steps,
                        interactive = True
                    )

                    image_height_field = gr.Slider(
                        label = "Height:",
                        minimum = minimum_height,
                        maximum = maximum_height,
                        value = default_height,
                        step = width_and_height_input_slider_steps,
                        interactive = True
                    )

                with gr.Row(
                    elem_id = "base_model_steps_field_row_id",
                    visible = default_base_model_steps_field_row_visibility
                ):

                    base_model_steps_field = gr.Slider(
                        label = "Steps:",
                        minimum = 1,
                        maximum = maximum_base_model_steps,
                        value = default_base_model_steps,
                        step = 1,
                        interactive = True
                    )

                with gr.Row(
                    elem_id = "base_model_steps_field_for_sdxl_turbo_field_row_id",
                    visible = default_base_model_steps_field_for_sdxl_turbo_field_row_visibility
                ):

                    base_model_steps_field_for_sdxl_turbo_field = gr.Slider(
                        label = "Steps:",
                        info = "Try using only 1 or a couple of steps.",
                        minimum = 1,
                        maximum = maximum_base_model_steps_for_sdxl_turbo,
                        value = default_base_model_steps_for_sdxl_turbo,
                        step = 1,
                        interactive = True
                    )

                with gr.Row(
                    elem_id = "guidance_scale_field_row_id",
                    visible = default_guidance_scale_field_row_visibility
                ):

                    guidance_scale_field = gr.Slider(
                        label = "Guidance Scale:",
                        minimum = minimum_guidance_scale,
                        maximum = maximum_guidance_scale,
                        value = default_guidance_scale,
                        step = guidance_scale_input_slider_steps,
                        interactive = True
                    )

                with gr.Row(
                    elem_id = "guidance_scale_for_sdxl_turbo_field_row_id",
                    visible = default_guidance_scale_for_sdxl_turbo_field_row_visibility
                ):

                    guidance_scale_for_sdxl_turbo_field = gr.HTML(
                        value = "<div style=\"padding: 10px; text-align: center; background: #fff;\">Guidance scale is not used for SDXL Turbo.</div>"
                    )

                with gr.Row():

                    if default_seed_value == "random":

                        default_seed_value = generate_random_seed()

                    #    If you use a slider or number field for the seed, some
                    #    seeds can't be duplicated using those fields. If you
                    #    enter a number greater than 9007199254740992, the seed
                    #    won't reliably be used. This is a technical limitation
                    #    as of writing this. See the bug report here:
                    #        https://github.com/gradio-app/gradio/issues/5354
                    #
                    #    Until this is fixed, we use a textbox if the max seed
                    #    allowed is greater than that number. Using the slider,
                    #    and not entering a number, might be the way to get
                    #    reliable numbers above that number, if you just don't
                    #    then use the up and down arrows in the field to go up
                    #    or down a number.
                    #
                    #    For now, I do this, but I might eventually have a
                    #    setting on the page to allow the slider.

                    if make_seed_selection_a_textbox == 1:

                        seed_field = gr.Textbox(
                            label = "Seed:",
                            value = default_seed_value,
                            interactive = True
                        )

                    else:

                        seed_field = gr.Slider(
                            label = "Seed:",
                            minimum = 0,
                            maximum = maximum_seed,
                            value = default_seed_value,
                            step = 1,
                            interactive = True
                        )



        with gr.Column(
            scale = 3,
            min_width = 200
        ):

            output_image_field_visibility = True
            output_image_gallery_field_visibility = False

            if use_image_gallery == 1:

                output_image_field_visibility = False
                output_image_gallery_field_visibility = True

            with gr.Row(
                visible = output_image_field_visibility
            ):

                output_image_field = gr.Image(
                    label = "Generated Image",
                    type = "pil",
                    height = gradio_image_component_height,
                    elem_classes = "image_scaling"
                )

            with gr.Row(
                visible = output_image_gallery_field_visibility
            ):

                output_image_gallery_field = gr.Gallery(
                    label = "Generated Images",
                    value = [],
                    selected_index = 0,
                    allow_preview = True,
                    preview = True,
#                    columns = "2",
#                    rows = None,
#                    columns = "2",
#                    rows = None,
                    height = gradio_image_gallery_component_height,
                    object_fit = "scale-down",
                    show_download_button = True,
                    elem_classes = "image_scaling"
                )

            with gr.Row(
                visible = False
            ) as output_image_preview_field_row:

                output_image_preview_field_every_value = None

                if enable_image_preview == 1:

                    output_image_preview_field_every_value = load_image_preview_frequency_in_seconds

                output_image_preview_field = gr.Image(
                    elem_id = "image_preview_id",
#                    value = load_image_preview(user_id_state),
#                    every = output_image_preview_field_every_value,
                    label = "Preview",
                    type = "pil",
                    interactive = False,
                    show_download_button = True,
                    height = gradio_image_gallery_component_height,
                    elem_classes = "image_scaling"
                )

            with gr.Row():

                error_text_field = gr.Textbox(
                    label = "Error Information:",
                    value = "",
                    show_copy_button = True,
                    lines = 5,
                    max_lines = 8,
                    autoscroll = False,
                    interactive = False,
                    container = True,
                    elem_classes = "textbox_vertical_scroll",
                    visible = False
                )

            with gr.Row():

                output_text_field = gr.Textbox(
                    label = "Prompt Information:",
                    value = "After an image is generated, its generation information will appear here. All of this information is also embedded in the image itself. If you open the image in a text program, it will appear at the top." + additional_prompt_info_html,
                    show_copy_button = True,
                    lines = 5,
                    max_lines = 8,
                    autoscroll = False,
                    interactive = False,
                    container = True,
                    elem_classes = "textbox_vertical_scroll"
                )

            with gr.Group(
                    visible = False
            ) as prompt_truncated_field_group:

                with gr.Row():

                    prompt_truncated_field = gr.Textbox(
                        label = "Prompt Truncated:",
                        info = "",
                        show_copy_button = True,
                        lines = 3,
                        max_lines = 5,
                        autoscroll = False,
                        interactive = False,
                        container = True,
                        elem_classes = "textbox_vertical_scroll"
                    )

            with gr.Group(
                    visible = False
            ) as negative_prompt_truncated_field_group:

                with gr.Row():

                    negative_prompt_truncated_field = gr.Textbox(
                        label = "Negative Prompt Truncated:",
                        info = "Your negative prompt was been truncated because it was too long. The part below was removed.",
                        value = "",
                        show_copy_button = True,
                        lines = 3,
                        max_lines = 5,
                        autoscroll = False,
                        interactive = False,
                        container = True,
                        elem_classes = "textbox_vertical_scroll"
                    )

            if enable_close_command_prompt_button == 1:

                with gr.Row():

                    close_command_prompt_button = gr.Button(
                        value = "Close Command Prompt",
                        variant = "stop"
                    )

                gr.Markdown("Closing the command prompt will cancel any images in the process of being created. You will need to launch it again, and then likely refresh the page, to create more images.")



    with gr.Accordion(
        label = "Other Settings",
        open = True
    ):

        with gr.Row():

            add_seed_into_pipe_field = gr.Checkbox(
                label = "Add seed to generation (to make it deterministic)",
                value = default_add_seed_into_pipe_is_selected,
                interactive = True,
                container = False,
                elem_classes = "sp_checkbox"
            )

        with gr.Row():

            use_torch_manual_seed_but_do_not_add_to_pipe_field = gr.Checkbox(
                label = "Use torch.manual_seed, but don't actually add seed to generation (ignored if \"Add seed to generation\" is checked. This is to maintain compatibility with PhotoReal site)",
                value = default_use_torch_manual_seed_but_do_not_add_to_pipe_is_selected,
                interactive = True,
                container = False,
                elem_classes = "sp_checkbox"
            )

        with gr.Row():

            save_or_display_word_text_for_save_base_image = "Display"
            if auto_save_imagery == 1:
                save_or_display_word_text_for_save_base_image = "Save"

            save_base_image_when_using_refiner_field = gr.Checkbox(
                label = save_or_display_word_text_for_save_base_image + " base image as well when using refiner (doesn't work yet)",
                value = default_save_base_image_when_using_refiner_is_selected,
                interactive = True,
                container = False,
                elem_classes = "sp_checkbox"
            )

        with gr.Row():

            save_or_display_word_text_for_save_refined_image = "display"
            if auto_save_imagery == 1:
                save_or_display_word_text_for_save_refined_image = "save"

            save_refined_image_when_using_upscaler_field = gr.Checkbox(
                label = "If applicable, " + save_or_display_word_text_for_save_refined_image + " refined image as well when using upscaler (doesn't work yet)",
                value = default_save_refined_image_when_using_upscaler_is_selected,
                interactive = True,
                container = False,
                elem_classes = "sp_checkbox"
            )



    if len(ending_html) > 0:

        with gr.Accordion(
            label = "Information",
            open = True
        ):

            gr.Markdown(ending_html)



    #####################
    #
    #    Update Refiner and Upscaler Status Function for Javascript
    #
    #        When the refiner or upscaler is turned on or off, a text message is
    #        printed on the page. That needs to be updated.
    #
    #####################

    update_refiner_and_upscaler_status_function_js = """

async (
    refiningSelectionFieldValue,
    upscalingSelectionFieldValue
) => {{
    "use strict";

    var refinerOnText = "{0}";
    var refinerOffText = "{1}";
    var upscalerOnText = "{2}";
    var upscalerOffText = "{3}";

    var refinerAndUpscalerInfoMessageHtml = "";

    if (refiningSelectionFieldValue === "Yes") {{

        refinerAndUpscalerInfoMessageHtml += refinerOnText;

    }}
    else {{

        refinerAndUpscalerInfoMessageHtml += refinerOffText;

    }}

    if (upscalingSelectionFieldValue === "Yes") {{

        refinerAndUpscalerInfoMessageHtml += upscalerOnText;

    }}
    else {{

        refinerAndUpscalerInfoMessageHtml += upscalerOffText;

    }}

    document.getElementById("refiner_and_upscaler_info_message_div_id").innerHTML = refinerAndUpscalerInfoMessageHtml;

}}

""".format(
    refiner_on_text,
    refiner_off_text,
    upscaler_on_text,
    upscaler_off_text
)



    #####################
    #
    #    Model Change Function for Javascript
    #
    #        When the base model or model configuration is changed, we may need
    #        to show and hide certain fields.
    #
    #####################

    model_change_function_js = """

async (
    baseModelFieldFullNameValue,
    possiblyModelConfigurationFullNameValue
) => {{
    "use strict";

    var baseModelNamesObject = {0};
    var modelConfigurationNamesObject = {1};
    var baseModelArray = {2};
    var modelConfigurationIncludeRefinerNumberOfStepsObject = {3};
    var baseModelsNotSupportingDenoisingEndForBaseModelObject = {4}
    var allowOtherModelVersions = {5};

    var baseModelFullNamesToBaseModelIdConversion = {{}};
    Object.keys(baseModelNamesObject).forEach(key => {{
        baseModelFullNamesToBaseModelIdConversion[baseModelNamesObject[key]] = key;
    }});
    var baseModelFieldValue = "";
    if (baseModelFullNamesToBaseModelIdConversion.hasOwnProperty(baseModelFieldFullNameValue)) {{
        baseModelFieldValue = baseModelFullNamesToBaseModelIdConversion[baseModelFieldFullNameValue];
    }}

    var modelConfigurationFullNameValue = ""

    var isBaseModelDropdownChange = 0

    if (baseModelFieldFullNameValue === possiblyModelConfigurationFullNameValue) {{

        isBaseModelDropdownChange = 1;

        modelConfigurationFullNameValue = window.modelConfigurationDropdownFieldValuesObject[baseModelFieldValue];

    }}
    else {{

        modelConfigurationFullNameValue = possiblyModelConfigurationFullNameValue;

        window.modelConfigurationDropdownFieldValuesObject[baseModelFieldValue] = modelConfigurationFullNameValue;

    }}

    var modelConfigurationFullNamesToModelConfigurationIdConversion = {{}};
    Object.keys(modelConfigurationNamesObject).forEach(key => {{
        modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationNamesObject[key]] = key;
    }});
    var modelConfigurationNameValue = "";
    if (modelConfigurationFullNamesToModelConfigurationIdConversion.hasOwnProperty(modelConfigurationFullNameValue)) {{
        modelConfigurationNameValue = modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationFullNameValue];
    }}

    for (var thisBaseModel of baseModelArray) {{

        var thisModelConfigurationElementId = "model_configuration_field_" + thisBaseModel + "_row_id";

        var thisModelConfigurationElementDisplay = "none";

        if (
            (thisBaseModel === baseModelFieldValue) &&
            (allowOtherModelVersions === 1)
        ) {{

            thisModelConfigurationElementDisplay = "block";

        }}

        document.getElementById(thisModelConfigurationElementId).style.display = thisModelConfigurationElementDisplay;

    }}

    var modelConfigurationFullNamesToModelConfigurationIdConversion = {{}};
    Object.keys(modelConfigurationNamesObject).forEach(key => {{
        modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationNamesObject[key]] = key;
    }});
    var modelConfigurationNameValue = "";
    if (modelConfigurationFullNamesToModelConfigurationIdConversion.hasOwnProperty(modelConfigurationFullNameValue)) {{
        modelConfigurationNameValue = modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationFullNameValue];
    }}

    if (
        baseModelFieldValue &&
        modelConfigurationNameValue
    ) {{

        var negativePromptFieldDisplay = "block";
        var negativePromptForSdxlTurboFieldDisplay = "none";
        var baseModelNumInferenceStepsFieldDisplay = "block";
        var baseModelNumInferenceStepsFieldForSdxlTurboFieldDisplay = "none";
        var guidanceScaleFieldDisplay = "block";
        var guidanceScaleForSdxlTurboFieldDisplay = "none";

        if (baseModelFieldValue === "sdxl_turbo") {{

            negativePromptFieldDisplay = "none";
            negativePromptForSdxlTurboFieldDisplay = "block";
            baseModelNumInferenceStepsFieldDisplay = "none";
            baseModelNumInferenceStepsFieldForSdxlTurboFieldDisplay = "block";
            guidanceScaleFieldDisplay = "none";
            guidanceScaleForSdxlTurboFieldDisplay = "block";

        }}

        document.getElementById("negative_prompt_field_row_id").style.display = negativePromptFieldDisplay;
        document.getElementById("negative_prompt_for_sdxl_turbo_field_row_id").style.display = negativePromptForSdxlTurboFieldDisplay;
        document.getElementById("base_model_steps_field_row_id").style.display = baseModelNumInferenceStepsFieldDisplay;
        document.getElementById("base_model_steps_field_for_sdxl_turbo_field_row_id").style.display = baseModelNumInferenceStepsFieldForSdxlTurboFieldDisplay;
        document.getElementById("guidance_scale_field_row_id").style.display = guidanceScaleFieldDisplay;
        document.getElementById("guidance_scale_for_sdxl_turbo_field_row_id").style.display = guidanceScaleForSdxlTurboFieldDisplay;



















        var refiningStepsForOlderConfiguration = "none";

        if (Object.keys(modelConfigurationIncludeRefinerNumberOfStepsObject).includes(modelConfigurationNameValue)) {{

             refiningStepsForOlderConfiguration = "block";

        }}

        var refiningUseDenoisingStartInBaseModelWhenUsingRefinerFieldDisplay = "block";

        if (Object.keys(baseModelsNotSupportingDenoisingEndForBaseModelObject).includes(baseModelFieldValue)) {{

             refiningUseDenoisingStartInBaseModelWhenUsingRefinerFieldDisplay = "none";

        }}

        document.getElementById("refining_use_denoising_start_in_base_model_when_using_refiner_field_row_id").style.display = refiningUseDenoisingStartInBaseModelWhenUsingRefinerFieldDisplay;
        document.getElementById("refining_steps_for_older_configuration_field_row_id").style.display = refiningStepsForOlderConfiguration;

    }}

}}

""".format(
    base_model_names_object,
    model_configuration_names_object,
    base_model_array,
    model_configuration_include_refiner_number_of_steps_object,
    base_models_not_supporting_denoising_end_for_base_model_object,
    allow_other_model_versions
)



    base_model_field.change(
        fn = None,
        inputs = [
            base_model_field
        ],
        outputs = None,
        js = model_change_function_js
    )



    for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array:

        this_model_configuration_dropdown_field.change(
            fn = None,
            inputs = [
                base_model_field,
                this_model_configuration_dropdown_field
            ],
            outputs = None,
            js = model_change_function_js
        )



    output_image_gallery_field.select(
        fn = update_prompt_info_from_gallery,
        inputs = [
            prompt_information_array_state
        ],
        outputs = [
            output_image_gallery_field,
            output_text_field
        ],
        show_progress = "hidden"
    )



    if (
        (enable_refiner == 1) or
        (enable_upscaler == 1)
    ):

        triggers_array = []

        if enable_refiner == 1:

            triggers_array.extend([
                base_model_field.change,
                refining_selection_field.change
            ])

            for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array:

                triggers_array.extend([
                    this_model_configuration_dropdown_field.change
                ])

        if enable_upscaler == 1:

            triggers_array.extend([
                upscaling_selection_field.change
            ])

        gr.on(
            triggers = triggers_array,
            fn = None,
            inputs = [
                refining_selection_field,
                upscaling_selection_field
            ],
            outputs = None,
            show_progress = "hidden",
            queue = False,
            js = update_refiner_and_upscaler_status_function_js
        )



    create_image_function_inputs = [

        base_model_field,

        prompt_field,
        negative_prompt_field,

        scheduler_field,

        image_width_field,
        image_height_field,
        guidance_scale_field,
        base_model_steps_field,
        base_model_steps_field_for_sdxl_turbo_field,
        seed_field,
        add_seed_into_pipe_field,
        use_torch_manual_seed_but_do_not_add_to_pipe_field,

        refining_selection_field,
        refining_denoise_start_field,
        refining_use_denoising_start_in_base_model_when_using_refiner_field,
        refining_base_model_output_to_refiner_is_in_latent_space_field,

        refining_steps_for_older_configuration_field,

        upscaling_selection_field,
        upscaling_num_inference_steps_field,

        image_gallery_array_state,
        prompt_information_array_state,
        last_model_configuration_name_selected_state,
        last_refiner_name_selected_state,
        last_upscaler_name_selected_state,
        stored_pipe_state,
        stored_refiner_state,
        stored_upscaler_state,

        save_base_image_when_using_refiner_field,
        save_refined_image_when_using_upscaler_field,

        user_id_state,
        image_generation_id_state,
        canceled_images_array_state

    ]

    for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array:

        create_image_function_inputs.append(
            this_model_configuration_dropdown_field
        )



    generate_image_button_click_event = generate_image_button.click(
        fn = before_create_image_function,
        inputs = None,
        outputs = [
            generate_image_button,
            output_image_field,
            output_image_gallery_field,
            output_text_field,
            output_image_preview_field_row,
            prompt_truncated_field_group,
            prompt_truncated_field,
            negative_prompt_truncated_field_group,
            negative_prompt_truncated_field,
            cancel_image_button_row,
            cancel_image_button,
            cancel_image_message_field_row,
            cancel_image_message_field,
            error_text_field,
            image_generation_id_state
        ],
        show_progress = "hidden",
        queue = True
    ).then(
        fn = create_image_function,
        inputs = create_image_function_inputs,
        outputs = [
            output_image_field,
            output_image_gallery_field,
            output_text_field,
            prompt_truncated_field_group,
            prompt_truncated_field,
            negative_prompt_truncated_field_group,
            negative_prompt_truncated_field,
            last_model_configuration_name_selected_state,
            last_refiner_name_selected_state,
            last_upscaler_name_selected_state,
            stored_pipe_state,
            stored_refiner_state,
            stored_upscaler_state,
            error_text_field
        ],
        show_progress = "full",
        queue = True
    ).then(
        fn = after_create_image_function,
        inputs = None,
        outputs = [
            generate_image_button,
            output_image_field,
            output_image_gallery_field,
            output_text_field,
            output_image_preview_field_row,
            generate_image_button_row,
            cancel_image_button_row,
            cancel_image_button,
            cancel_image_message_field_row,
            cancel_image_message_field
        ],
        show_progress = "hidden",
        queue = True
    )



    verify_seed_field_textbox_function_js = """

async (
    seedFieldTextboxValue
) => {{
    "use strict";

    var defaultSeedMaximum = parseInt({0});

    seedFieldTextboxValue = parseInt(seedFieldTextboxValue);

    if (isNaN(seedFieldTextboxValue)) {{

        seedFieldTextboxValue = "";

    }}
    else if (seedFieldTextboxValue > defaultSeedMaximum) {{

        seedFieldTextboxValue = defaultSeedMaximum;

    }}

    return [
        seedFieldTextboxValue
    ];

}}

""".format(
    maximum_seed
)



    if make_seed_selection_a_textbox == 1:

        seed_field.change(
            fn = None,
            inputs = [
                seed_field
            ],
            outputs = [
                seed_field
            ],
            show_progress = "hidden",
            queue = False,
            js = verify_seed_field_textbox_function_js
        )



    if enable_image_generation_cancellation == 1:



        cancel_image_click_event = cancel_image_button.click(
            fn = cancel_image_function,
            inputs = [
                user_id_state,
                image_generation_id_state
            ],
            outputs = [
                generate_image_button_row,
                generate_image_button,
                output_text_field,
                output_image_preview_field_row,
                cancel_image_button_row,
                cancel_image_button,
                cancel_image_message_field_row,
                cancel_image_message_field
            ],
            show_progress = "hidden",
            cancels = [generate_image_button_click_event],
            queue = True
        )



    if enable_close_command_prompt_button == 1:

        #    https://github.com/gradio-app/gradio/pull/2433/files

        close_command_prompt_button_click_event = close_command_prompt_button.click(
            fn = close_command_prompt,
            inputs = None,
            outputs = None,
            cancels = [generate_image_button_click_event],
            queue = True
        )



    #    Remove last comma

    model_configuration_dropdown_field_values_for_js = model_configuration_dropdown_field_values_for_js[:-1]



    script_on_load_js = """

async () => {{
    "use strict";

    window.modelConfigurationDropdownFieldValuesObject = {{{0}}};

    document.querySelectorAll(".textbox_vertical_scroll textarea").forEach(e => e.classList.remove("scroll-hide"));

}}

""".format(
    model_configuration_dropdown_field_values_for_js
)



    model_base_model_and_model_configuration_inputs = [
        base_model_field,
        initial_model_configuration_name_selected_state
    ]

    model_base_model_and_model_configuration_outputs = []

    for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array:

        model_base_model_and_model_configuration_inputs.append(
            this_model_configuration_dropdown_field
        )

        model_base_model_and_model_configuration_outputs.append(
            this_model_configuration_dropdown_field
        )



    sd_interface_load_outputs = create_image_function_inputs + [
        initial_model_configuration_name_selected_state,
        generate_image_button
    ]



    every_value_in_seconds_for_image_preview = None

    if enable_image_preview == 1:

        every_value_in_seconds_for_image_preview = load_image_preview_frequency_in_seconds



    sd_interface_continuous = sd_interface.load(
        fn = get_query_params,
        inputs = None,
        outputs = sd_interface_load_outputs,
        show_progress = "hidden",
        queue = False,
        scroll_to_output = False,
        js = script_on_load_js
    ).then(
        fn = set_base_model_and_model_configuration_from_query_params,
        inputs = model_base_model_and_model_configuration_inputs,
        outputs = model_base_model_and_model_configuration_outputs,
        show_progress = "hidden",
        queue = False
    ).then(
        fn = set_base_model_and_model_configuration_from_query_params,
        inputs = model_base_model_and_model_configuration_inputs,
        outputs = model_base_model_and_model_configuration_outputs,
        show_progress = "hidden",
        queue = False,
    ).then(
        fn = load_image_preview,
        inputs = [
            user_id_state
        ],
        outputs = [
            output_image_preview_field
        ],
        show_progress = "hidden",
        every = every_value_in_seconds_for_image_preview,
        queue = True
    )



sd_interface.queue(
    max_size = max_queue_size
)



inbrowser = False

if auto_open_browser == 1:

    inbrowser = True

sd_interface.queue().launch(
#sd_interface.launch(
    inbrowser = inbrowser,
    share = None,
    show_api = False,
    quiet = True,
    show_error = True,
    state_session_capacity = 10000,
    max_threads = 40
)