init

README.md

@@ -1,6 +1,6 @@
 ---
-title: 🧩 DiffuseCraft
-emoji: 🧩🖼️
+title: 😈️ Ivan's DiffuseCraft
+emoji: 😈️
 colorFrom: red
 colorTo: pink
 sdk: gradio
@@ -8,7 +8,7 @@ sdk_version: 4.31.3
 app_file: app.py
 pinned: true
 license: mit
-short_description: Stunning images using stable diffusion.
+short_description: (ivan) Stunning images using stable diffusion.
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -25,55 +25,55 @@ from stablepy import (
 import urllib.parse
 
 preprocessor_controlnet = {
-  "openpose": [
-    "Openpose",
-    "None",
-  ],
-  "scribble": [
-    "HED",
-    "Pidinet",
-    "None",
-  ],
-  "softedge": [
-    "Pidinet",
-    "HED",
-    "HED safe",
-    "Pidinet safe",
-    "None",
-  ],
-  "segmentation": [
-    "UPerNet",
-    "None",
-  ],
-  "depth": [
-    "DPT",
-    "Midas",
-    "None",
-  ],
-  "normalbae": [
-    "NormalBae",
-    "None",
-  ],
-  "lineart": [
-    "Lineart",
-    "Lineart coarse",
-    "Lineart (anime)",
-    "None",
-    "None (anime)",
-  ],
-  "shuffle": [
-    "ContentShuffle",
-    "None",
-  ],
-  "canny": [
-    "Canny"
-  ],
-  "mlsd": [
-    "MLSD"
-  ],
-  "ip2p": [
-    "ip2p"
-  ]
+    "openpose": [
+        "Openpose",
+        "None",
+    ],
+    "scribble": [
+        "HED",
+        "Pidinet",
+        "None",
+    ],
+    "softedge": [
+        "Pidinet",
+        "HED",
+        "HED safe",
+        "Pidinet safe",
+        "None",
+    ],
+    "segmentation": [
+        "UPerNet",
+        "None",
+    ],
+    "depth": [
+        "DPT",
+        "Midas",
+        "None",
+    ],
+    "normalbae": [
+        "NormalBae",
+        "None",
+    ],
+    "lineart": [
+        "Lineart",
+        "Lineart coarse",
+        "Lineart (anime)",
+        "None",
+        "None (anime)",
+    ],
+    "shuffle": [
+        "ContentShuffle",
+        "None",
+    ],
+    "canny": [
+        "Canny"
+    ],
+    "mlsd": [
+        "MLSD"
+    ],
+    "ip2p": [
+        "ip2p"
+    ]
 }
 
 task_stablepy = {
@@ -106,7 +106,7 @@ task_model_list = list(task_stablepy.keys())
 
 def download_things(directory, url, hf_token="", civitai_api_key=""):
     url = url.strip()
-    
+
     if "drive.google.com" in url:
         original_dir = os.getcwd()
         os.chdir(directory)
@@ -119,15 +119,18 @@ def download_things(directory, url, hf_token="", civitai_api_key=""):
             url = url.replace("/blob/", "/resolve/")
         user_header = f'"Authorization: Bearer {hf_token}"'
         if hf_token:
-            os.system(f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 {url} -d {directory}  -o {url.split('/')[-1]}")
+            os.system(
+                f"aria2c --console-log-level=error --summary-interval=10 --header={user_header} -c -x 16 -k 1M -s 16 {url} -d {directory}  -o {url.split('/')[-1]}")
         else:
-            os.system (f"aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 {url} -d {directory}  -o {url.split('/')[-1]}")
+            os.system(
+                f"aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 {url} -d {directory}  -o {url.split('/')[-1]}")
     elif "civitai.com" in url:
         if "?" in url:
             url = url.split("?")[0]
         if civitai_api_key:
             url = url + f"?token={civitai_api_key}"
-            os.system(f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
+            os.system(
+                f"aria2c --console-log-level=error --summary-interval=10 -c -x 16 -k 1M -s 16 -d {directory} {url}")
         else:
             print("\033[91mYou need an API key to download Civitai models.\033[0m")
     else:
@@ -136,7 +139,7 @@ def download_things(directory, url, hf_token="", civitai_api_key=""):
 
 def get_model_list(directory_path):
     model_list = []
-    valid_extensions = {'.ckpt' , '.pt', '.pth', '.safetensors', '.bin'}
+    valid_extensions = {'.ckpt', '.pt', '.pth', '.safetensors', '.bin'}
 
     for filename in os.listdir(directory_path):
         if os.path.splitext(filename)[1] in valid_extensions:
@@ -228,7 +231,7 @@ download_embeds = [
     'https://huggingface.co/datasets/Nerfgun3/bad_prompt/blob/main/bad_prompt_version2.pt',
     'https://huggingface.co/embed/negative/resolve/main/EasyNegativeV2.safetensors',
     'https://huggingface.co/embed/negative/resolve/main/bad-hands-5.pt',
-    ]
+]
 
 for url_embed in download_embeds:
     if not os.path.exists(f"./embedings/{url_embed.split('/')[-1]}"):
@@ -243,13 +246,14 @@ lora_model_list.insert(0, "None")
 vae_model_list = get_model_list(directory_vaes)
 vae_model_list.insert(0, "None")
 
+
 def get_my_lora(link_url):
     for url in [url.strip() for url in link_url.split(',')]:
         if not os.path.exists(f"./loras/{url.split('/')[-1]}"):
             download_things(directory_loras, url, hf_token, CIVITAI_API_KEY)
     new_lora_model_list = get_model_list(directory_loras)
     new_lora_model_list.insert(0, "None")
-    
+
     return gr.update(
         choices=new_lora_model_list
     ), gr.update(
@@ -262,26 +266,27 @@ def get_my_lora(link_url):
         choices=new_lora_model_list
     ),
 
+
 print('\033[33m🏁 Download and listing of valid models completed.\033[0m')
 
 upscaler_dict_gui = {
-    None : None,
-    "Lanczos" : "Lanczos",
-    "Nearest" : "Nearest",
-    "RealESRGAN_x4plus" : "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth",
-    "RealESRNet_x4plus" : "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth",
+    None: None,
+    "Lanczos": "Lanczos",
+    "Nearest": "Nearest",
+    "RealESRGAN_x4plus": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth",
+    "RealESRNet_x4plus": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth",
     "RealESRGAN_x4plus_anime_6B": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth",
     "RealESRGAN_x2plus": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth",
     "realesr-animevideov3": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-animevideov3.pth",
     "realesr-general-x4v3": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth",
-    "realesr-general-wdn-x4v3" : "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-wdn-x4v3.pth",
-    "4x-UltraSharp" : "https://huggingface.co/Shandypur/ESRGAN-4x-UltraSharp/resolve/main/4x-UltraSharp.pth",
-    "4x_foolhardy_Remacri" : "https://huggingface.co/FacehugmanIII/4x_foolhardy_Remacri/resolve/main/4x_foolhardy_Remacri.pth",
-    "Remacri4xExtraSmoother" : "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/Remacri%204x%20ExtraSmoother.pth",
-    "AnimeSharp4x" : "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/AnimeSharp%204x.pth",
-    "lollypop" : "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/lollypop.pth",
-    "RealisticRescaler4x" : "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/RealisticRescaler%204x.pth",
-    "NickelbackFS4x" : "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/NickelbackFS%204x.pth"
+    "realesr-general-wdn-x4v3": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-wdn-x4v3.pth",
+    "4x-UltraSharp": "https://huggingface.co/Shandypur/ESRGAN-4x-UltraSharp/resolve/main/4x-UltraSharp.pth",
+    "4x_foolhardy_Remacri": "https://huggingface.co/FacehugmanIII/4x_foolhardy_Remacri/resolve/main/4x_foolhardy_Remacri.pth",
+    "Remacri4xExtraSmoother": "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/Remacri%204x%20ExtraSmoother.pth",
+    "AnimeSharp4x": "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/AnimeSharp%204x.pth",
+    "lollypop": "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/lollypop.pth",
+    "RealisticRescaler4x": "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/RealisticRescaler%204x.pth",
+    "NickelbackFS4x": "https://huggingface.co/hollowstrawberry/upscalers-backup/resolve/main/ESRGAN/NickelbackFS%204x.pth"
 }
 
 
@@ -335,16 +340,21 @@ import IPython.display
 import time, json
 from IPython.utils import capture
 import logging
+
 logging.getLogger("diffusers").setLevel(logging.ERROR)
 import diffusers
+
 diffusers.utils.logging.set_verbosity(40)
 import warnings
+
 warnings.filterwarnings(action="ignore", category=FutureWarning, module="diffusers")
 warnings.filterwarnings(action="ignore", category=UserWarning, module="diffusers")
 warnings.filterwarnings(action="ignore", category=FutureWarning, module="transformers")
 from stablepy import logger
+
 logger.setLevel(logging.DEBUG)
 
+
 def info_html(json_data, title, subtitle):
     return f"""
         <div style='padding: 0; border-radius: 10px;'>
@@ -356,10 +366,11 @@ def info_html(json_data, title, subtitle):
         </div>
         """
 
+
 class GuiSD:
     def __init__(self, stream=True):
         self.model = None
-    
+
         print("Loading model...")
         self.model = Model_Diffusers(
             base_model_id="cagliostrolab/animagine-xl-3.1",
@@ -372,7 +383,7 @@ class GuiSD:
     def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
 
         yield f"Loading model: {model_name}"
-        
+
         vae_model = vae_model if vae_model != "None" else None
 
         if model_name in model_list:
@@ -384,7 +395,6 @@ class GuiSD:
             if incompatible_vae:
                 vae_model = None
 
-        
         self.model.load_pipe(
             model_name,
             task_name=task_stablepy[task],
@@ -393,119 +403,118 @@ class GuiSD:
             retain_task_model_in_cache=False,
         )
         yield f"Model loaded: {model_name}"
-        
+
     @spaces.GPU
     def generate_pipeline(
-        self,
-        prompt,
-        neg_prompt,
-        num_images,
-        steps,
-        cfg,
-        clip_skip,
-        seed,
-        lora1,
-        lora_scale1,
-        lora2,
-        lora_scale2,
-        lora3,
-        lora_scale3,
-        lora4,
-        lora_scale4,
-        lora5,
-        lora_scale5,
-        sampler,
-        img_height,
-        img_width,
-        model_name,
-        vae_model,
-        task,
-        image_control,
-        preprocessor_name,
-        preprocess_resolution,
-        image_resolution,
-        style_prompt,  # list []
-        style_json_file,
-        image_mask,
-        strength,
-        low_threshold,
-        high_threshold,
-        value_threshold,
-        distance_threshold,
-        controlnet_output_scaling_in_unet,
-        controlnet_start_threshold,
-        controlnet_stop_threshold,
-        textual_inversion,
-        syntax_weights,
-        upscaler_model_path,
-        upscaler_increases_size,
-        esrgan_tile,
-        esrgan_tile_overlap,
-        hires_steps,
-        hires_denoising_strength,
-        hires_sampler,
-        hires_prompt,
-        hires_negative_prompt,
-        hires_before_adetailer,
-        hires_after_adetailer,
-        loop_generation,
-        leave_progress_bar,
-        disable_progress_bar,
-        image_previews,
-        display_images,
-        save_generated_images,
-        image_storage_location,
-        retain_compel_previous_load,
-        retain_detailfix_model_previous_load,
-        retain_hires_model_previous_load,
-        t2i_adapter_preprocessor,
-        t2i_adapter_conditioning_scale,
-        t2i_adapter_conditioning_factor,
-        xformers_memory_efficient_attention,
-        freeu,
-        generator_in_cpu,
-        adetailer_inpaint_only,
-        adetailer_verbose,
-        adetailer_sampler,
-        adetailer_active_a,
-        prompt_ad_a,
-        negative_prompt_ad_a,
-        strength_ad_a,
-        face_detector_ad_a,
-        person_detector_ad_a,
-        hand_detector_ad_a,
-        mask_dilation_a,
-        mask_blur_a,
-        mask_padding_a,
-        adetailer_active_b,
-        prompt_ad_b,
-        negative_prompt_ad_b,
-        strength_ad_b,
-        face_detector_ad_b,
-        person_detector_ad_b,
-        hand_detector_ad_b,
-        mask_dilation_b,
-        mask_blur_b,
-        mask_padding_b,
-        retain_task_cache_gui,
-        image_ip1,
-        mask_ip1,
-        model_ip1,
-        mode_ip1,
-        scale_ip1,
-        image_ip2,
-        mask_ip2,
-        model_ip2,
-        mode_ip2,
-        scale_ip2,
+            self,
+            prompt,
+            neg_prompt,
+            num_images,
+            steps,
+            cfg,
+            clip_skip,
+            seed,
+            lora1,
+            lora_scale1,
+            lora2,
+            lora_scale2,
+            lora3,
+            lora_scale3,
+            lora4,
+            lora_scale4,
+            lora5,
+            lora_scale5,
+            sampler,
+            img_height,
+            img_width,
+            model_name,
+            vae_model,
+            task,
+            image_control,
+            preprocessor_name,
+            preprocess_resolution,
+            image_resolution,
+            style_prompt,  # list []
+            style_json_file,
+            image_mask,
+            strength,
+            low_threshold,
+            high_threshold,
+            value_threshold,
+            distance_threshold,
+            controlnet_output_scaling_in_unet,
+            controlnet_start_threshold,
+            controlnet_stop_threshold,
+            textual_inversion,
+            syntax_weights,
+            upscaler_model_path,
+            upscaler_increases_size,
+            esrgan_tile,
+            esrgan_tile_overlap,
+            hires_steps,
+            hires_denoising_strength,
+            hires_sampler,
+            hires_prompt,
+            hires_negative_prompt,
+            hires_before_adetailer,
+            hires_after_adetailer,
+            loop_generation,
+            leave_progress_bar,
+            disable_progress_bar,
+            image_previews,
+            display_images,
+            save_generated_images,
+            image_storage_location,
+            retain_compel_previous_load,
+            retain_detailfix_model_previous_load,
+            retain_hires_model_previous_load,
+            t2i_adapter_preprocessor,
+            t2i_adapter_conditioning_scale,
+            t2i_adapter_conditioning_factor,
+            xformers_memory_efficient_attention,
+            freeu,
+            generator_in_cpu,
+            adetailer_inpaint_only,
+            adetailer_verbose,
+            adetailer_sampler,
+            adetailer_active_a,
+            prompt_ad_a,
+            negative_prompt_ad_a,
+            strength_ad_a,
+            face_detector_ad_a,
+            person_detector_ad_a,
+            hand_detector_ad_a,
+            mask_dilation_a,
+            mask_blur_a,
+            mask_padding_a,
+            adetailer_active_b,
+            prompt_ad_b,
+            negative_prompt_ad_b,
+            strength_ad_b,
+            face_detector_ad_b,
+            person_detector_ad_b,
+            hand_detector_ad_b,
+            mask_dilation_b,
+            mask_blur_b,
+            mask_padding_b,
+            retain_task_cache_gui,
+            image_ip1,
+            mask_ip1,
+            model_ip1,
+            mode_ip1,
+            scale_ip1,
+            image_ip2,
+            mask_ip2,
+            model_ip2,
+            mode_ip2,
+            scale_ip2,
     ):
-        
+
         vae_model = vae_model if vae_model != "None" else None
         loras_list = [lora1, lora2, lora3, lora4, lora5]
         vae_msg = f"VAE: {vae_model}" if vae_model else ""
         msg_lora = []
 
-        
         if model_name in model_list:
             model_is_xl = "xl" in model_name.lower()
             sdxl_in_vae = vae_model and "sdxl" in vae_model.lower()
@@ -514,8 +523,8 @@ class GuiSD:
 
             if incompatible_vae:
                 msg_inc_vae = (
-                    f"The selected VAE is for a { 'SD 1.5' if model_is_xl else 'SDXL' } model, but you"
-                    f" are using a { model_type } model. The default VAE "
+                    f"The selected VAE is for a {'SD 1.5' if model_is_xl else 'SDXL'} model, but you"
+                    f" are using a {model_type} model. The default VAE "
                     "will be used."
                 )
                 gr.Info(msg_inc_vae)
@@ -527,7 +536,7 @@ class GuiSD:
                     print(la)
                     lora_type = ("animetarot" in la.lower() or "Hyper-SD15-8steps".lower() in la.lower())
                     if (model_is_xl and lora_type) or (not model_is_xl and not lora_type):
-                        msg_inc_lora = f"The LoRA {la} is for { 'SD 1.5' if model_is_xl else 'SDXL' }, but you are using { model_type }."
+                        msg_inc_lora = f"The LoRA {la} is for {'SD 1.5' if model_is_xl else 'SDXL'}, but you are using {model_type}."
                         gr.Info(msg_inc_lora)
                         msg_lora.append(msg_inc_lora)
 
@@ -568,7 +577,10 @@ class GuiSD:
             )
 
         if task != "txt2img" and not image_control:
-            raise ValueError("No control image found: To use this function, you have to upload an image in 'Image ControlNet/Inpaint/Img2img'")
+            raise ValueError(
+                "No control image found: To use this function, "
+                "you have to upload an image in 'Image ControlNet/Inpaint/Img2img'"
+            )
 
         if task == "inpaint" and not image_mask:
             raise ValueError("No mask image found: Specify one in 'Image Mask'")
@@ -602,31 +614,31 @@ class GuiSD:
             print("No Textual inversion for SDXL")
 
         adetailer_params_A = {
-            "face_detector_ad" : face_detector_ad_a,
-            "person_detector_ad" : person_detector_ad_a,
-            "hand_detector_ad" : hand_detector_ad_a,
+            "face_detector_ad": face_detector_ad_a,
+            "person_detector_ad": person_detector_ad_a,
+            "hand_detector_ad": hand_detector_ad_a,
             "prompt": prompt_ad_a,
-            "negative_prompt" : negative_prompt_ad_a,
-            "strength" : strength_ad_a,
+            "negative_prompt": negative_prompt_ad_a,
+            "strength": strength_ad_a,
             # "image_list_task" : None,
-            "mask_dilation" : mask_dilation_a,
-            "mask_blur" : mask_blur_a,
-            "mask_padding" : mask_padding_a,
-            "inpaint_only" : adetailer_inpaint_only,
-            "sampler" : adetailer_sampler,
+            "mask_dilation": mask_dilation_a,
+            "mask_blur": mask_blur_a,
+            "mask_padding": mask_padding_a,
+            "inpaint_only": adetailer_inpaint_only,
+            "sampler": adetailer_sampler,
         }
 
         adetailer_params_B = {
-            "face_detector_ad" : face_detector_ad_b,
-            "person_detector_ad" : person_detector_ad_b,
-            "hand_detector_ad" : hand_detector_ad_b,
+            "face_detector_ad": face_detector_ad_b,
+            "person_detector_ad": person_detector_ad_b,
+            "hand_detector_ad": hand_detector_ad_b,
             "prompt": prompt_ad_b,
-            "negative_prompt" : negative_prompt_ad_b,
-            "strength" : strength_ad_b,
+            "negative_prompt": negative_prompt_ad_b,
+            "strength": strength_ad_b,
             # "image_list_task" : None,
-            "mask_dilation" : mask_dilation_b,
-            "mask_blur" : mask_blur_b,
-            "mask_padding" : mask_padding_b,
+            "mask_dilation": mask_dilation_b,
+            "mask_blur": mask_blur_b,
+            "mask_padding": mask_padding_b,
         }
         pipe_params = {
             "prompt": prompt,
@@ -709,8 +721,9 @@ class GuiSD:
 
         random_number = random.randint(1, 100)
         if random_number < 25 and num_images < 3:
-            if not upscaler_model and steps < 45 and task in ["txt2img", "img2img"] and not adetailer_active_a and not adetailer_active_b:
-                num_images *=2
+            if not upscaler_model and steps < 45 and task in ["txt2img",
+                                                              "img2img"] and not adetailer_active_a and not adetailer_active_b:
+                num_images *= 2
                 pipe_params["num_images"] = num_images
                 gr.Info("Num images x 2 🎉")
 
@@ -731,13 +744,15 @@ class GuiSD:
 
 sd_gen = GuiSD()
 
-CSS ="""
+CSS = """
 .contain { display: flex; flex-direction: column; }
 #component-0 { height: 100%; }
 #gallery { flex-grow: 1; }
 """
-sdxl_task = [k for k, v in task_stablepy.items() if v in SDXL_TASKS ]
-sd_task = [k for k, v in task_stablepy.items() if v in SD15_TASKS ]
+sdxl_task = [k for k, v in task_stablepy.items() if v in SDXL_TASKS]
+sd_task = [k for k, v in task_stablepy.items() if v in SD15_TASKS]
+
+
 def update_task_options(model_name, task_name):
     if model_name in model_list:
         if "xl" in model_name.lower():
@@ -754,7 +769,7 @@ def update_task_options(model_name, task_name):
 
 
 with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
-    gr.Markdown("# 🧩 DiffuseCraft")
+    gr.Markdown("# 🧩 (Ivan) DiffuseCraft")
     gr.Markdown(
         f"""
             ### This demo uses [diffusers](https://github.com/huggingface/diffusers) to perform different tasks in image generation.
@@ -766,7 +781,8 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
             with gr.Column(scale=2):
 
                 task_gui = gr.Dropdown(label="Task", choices=sdxl_task, value=task_model_list[0])
-                model_name_gui = gr.Dropdown(label="Model", choices=model_list, value=model_list[0], allow_custom_value=True)
+                model_name_gui = gr.Dropdown(label="Model", choices=model_list, value=model_list[0],
+                                             allow_custom_value=True)
                 prompt_gui = gr.Textbox(lines=5, placeholder="Enter prompt", label="Prompt")
                 neg_prompt_gui = gr.Textbox(lines=3, placeholder="Enter Neg prompt", label="Negative prompt")
                 with gr.Row(equal_height=False):
@@ -774,7 +790,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     clear_prompt_gui = gr.Button(value="🗑️")
                     set_random_seed = gr.Button(value="🎲")
                 generate_button = gr.Button(value="GENERATE", variant="primary")
-                
+
                 model_name_gui.change(
                     update_task_options,
                     [model_name_gui, task_gui],
@@ -782,7 +798,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                 )
 
                 load_model_gui = gr.HTML()
-                
+
                 result_images = gr.Gallery(
                     label="Generated images",
                     show_label=False,
@@ -797,7 +813,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                 )
 
                 actual_task_info = gr.HTML()
-            
+
             with gr.Column(scale=1):
                 steps_gui = gr.Slider(minimum=1, maximum=100, step=1, value=30, label="Steps")
                 cfg_gui = gr.Slider(minimum=0, maximum=30, step=0.5, value=7.5, label="CFG")
@@ -810,8 +826,6 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     free_u_gui = gr.Checkbox(value=True, label="FreeU")
 
                 with gr.Row(equal_height=False):
-                    
-
 
                     def run_set_params_gui(base_prompt):
                         valid_receptors = {  # default values
@@ -822,7 +836,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                             "height": gr.update(value=1024),
                             "Seed": gr.update(value=-1),
                             "Sampler": gr.update(value="Euler a"),
-                            "scale": gr.update(value=7.5), # cfg
+                            "scale": gr.update(value=7.5),  # cfg
                             "skip": gr.update(value=True),
                         }
                         valid_keys = list(valid_receptors.keys())
@@ -832,15 +846,15 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                             # print(val)
                             if key in valid_keys:
                                 if key == "Sampler":
-                                  if val not in scheduler_names:
-                                      continue
+                                    if val not in scheduler_names:
+                                        continue
                                 elif key == "skip":
-                                  if int(val) >= 2:
-                                    val = True
+                                    if int(val) >= 2:
+                                        val = True
                                 if key == "prompt":
-                                  if ">" in val and "<" in val:
-                                    val = re.sub(r'<[^>]+>', '', val)
-                                    print("Removed LoRA written in the prompt")
+                                    if ">" in val and "<" in val:
+                                        val = re.sub(r'<[^>]+>', '', val)
+                                        print("Removed LoRA written in the prompt")
                                 if key in ["prompt", "neg_prompt"]:
                                     val = val.strip()
                                 if key in ["Steps", "width", "height", "Seed"]:
@@ -854,8 +868,9 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                                 # print(valid_receptors)
                         return [value for value in valid_receptors.values()]
 
+
                     set_params_gui.click(
-                        run_set_params_gui, [prompt_gui],[
+                        run_set_params_gui, [prompt_gui], [
                             prompt_gui,
                             neg_prompt_gui,
                             steps_gui,
@@ -867,16 +882,21 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                             clip_skip_gui,
                         ],
                     )
-                    
-                    
+
+
                     def run_clear_prompt_gui():
                         return gr.update(value=""), gr.update(value="")
+
+
                     clear_prompt_gui.click(
                         run_clear_prompt_gui, [], [prompt_gui, neg_prompt_gui]
                     )
 
+
                     def run_set_random_seed():
                         return -1
+
+
                     set_random_seed.click(
                         run_set_random_seed, [], seed_gui
                     )
@@ -890,22 +910,30 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     ("Classic-ignore", "Classic-ignore"),
                     ("None", "None"),
                 ]
-                prompt_syntax_gui = gr.Dropdown(label="Prompt Syntax", choices=prompt_s_options, value=prompt_s_options[0][1])
+                prompt_syntax_gui = gr.Dropdown(label="Prompt Syntax", choices=prompt_s_options,
+                                                value=prompt_s_options[0][1])
                 vae_model_gui = gr.Dropdown(label="VAE Model", choices=vae_model_list)
 
                 with gr.Accordion("Hires fix", open=False, visible=True):
 
                     upscaler_keys = list(upscaler_dict_gui.keys())
 
-                    upscaler_model_path_gui = gr.Dropdown(label="Upscaler", choices=upscaler_keys, value=upscaler_keys[0])
-                    upscaler_increases_size_gui = gr.Slider(minimum=1.1, maximum=6., step=0.1, value=1.4, label="Upscale by")
+                    upscaler_model_path_gui = gr.Dropdown(label="Upscaler", choices=upscaler_keys,
+                                                          value=upscaler_keys[0])
+                    upscaler_increases_size_gui = gr.Slider(minimum=1.1, maximum=6., step=0.1, value=1.4,
+                                                            label="Upscale by")
                     esrgan_tile_gui = gr.Slider(minimum=0, value=100, maximum=500, step=1, label="ESRGAN Tile")
-                    esrgan_tile_overlap_gui = gr.Slider(minimum=1, maximum=200, step=1, value=10, label="ESRGAN Tile Overlap")
+                    esrgan_tile_overlap_gui = gr.Slider(minimum=1, maximum=200, step=1, value=10,
+                                                        label="ESRGAN Tile Overlap")
                     hires_steps_gui = gr.Slider(minimum=0, value=30, maximum=100, step=1, label="Hires Steps")
-                    hires_denoising_strength_gui = gr.Slider(minimum=0.1, maximum=1.0, step=0.01, value=0.55, label="Hires Denoising Strength")
-                    hires_sampler_gui = gr.Dropdown(label="Hires Sampler", choices=["Use same sampler"] + scheduler_names[:-1], value="Use same sampler")
+                    hires_denoising_strength_gui = gr.Slider(minimum=0.1, maximum=1.0, step=0.01, value=0.55,
+                                                             label="Hires Denoising Strength")
+                    hires_sampler_gui = gr.Dropdown(label="Hires Sampler",
+                                                    choices=["Use same sampler"] + scheduler_names[:-1],
+                                                    value="Use same sampler")
                     hires_prompt_gui = gr.Textbox(label="Hires Prompt", placeholder="Main prompt will be use", lines=3)
-                    hires_negative_prompt_gui = gr.Textbox(label="Hires Negative Prompt", placeholder="Main negative prompt will be use", lines=3)
+                    hires_negative_prompt_gui = gr.Textbox(label="Hires Negative Prompt",
+                                                           placeholder="Main negative prompt will be use", lines=3)
 
                 with gr.Accordion("LoRA", open=False, visible=True):
                     lora1_gui = gr.Dropdown(label="Lora1", choices=lora_model_list)
@@ -928,7 +956,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                             [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui]
                         )
 
-                with gr.Accordion("IP-Adapter", open=False, visible=True):##############
+                with gr.Accordion("IP-Adapter", open=False, visible=True):  ##############
 
                     IP_MODELS = sorted(list(set(IP_ADAPTERS_SD + IP_ADAPTERS_SDXL)))
                     MODE_IP_OPTIONS = ["original", "style", "layout", "style+layout"]
@@ -953,8 +981,11 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         minimum=0.01, maximum=1.0, step=0.01, value=0.55, label="Strength",
                         info="This option adjusts the level of changes for img2img and inpainting."
                     )
-                    image_resolution_gui = gr.Slider(minimum=64, maximum=2048, step=64, value=1024, label="Image Resolution")
-                    preprocessor_name_gui = gr.Dropdown(label="Preprocessor Name", choices=preprocessor_controlnet["canny"])
+                    image_resolution_gui = gr.Slider(minimum=64, maximum=2048, step=64, value=1024,
+                                                     label="Image Resolution")
+                    preprocessor_name_gui = gr.Dropdown(label="Preprocessor Name",
+                                                        choices=preprocessor_controlnet["canny"])
+
 
                     def change_preprocessor_choices(task):
                         task = task_stablepy[task]
@@ -964,24 +995,35 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                             choices_task = preprocessor_controlnet["canny"]
                         return gr.update(choices=choices_task, value=choices_task[0])
 
+
                     task_gui.change(
                         change_preprocessor_choices,
                         [task_gui],
                         [preprocessor_name_gui],
                     )
-                    preprocess_resolution_gui = gr.Slider(minimum=64, maximum=2048, step=64, value=512, label="Preprocess Resolution")
-                    low_threshold_gui = gr.Slider(minimum=1, maximum=255, step=1, value=100, label="Canny low threshold")
-                    high_threshold_gui = gr.Slider(minimum=1, maximum=255, step=1, value=200, label="Canny high threshold")
-                    value_threshold_gui = gr.Slider(minimum=1, maximum=2.0, step=0.01, value=0.1, label="Hough value threshold (MLSD)")
-                    distance_threshold_gui = gr.Slider(minimum=1, maximum=20.0, step=0.01, value=0.1, label="Hough distance threshold (MLSD)")
-                    control_net_output_scaling_gui = gr.Slider(minimum=0, maximum=5.0, step=0.1, value=1, label="ControlNet Output Scaling in UNet")
-                    control_net_start_threshold_gui = gr.Slider(minimum=0, maximum=1, step=0.01, value=0, label="ControlNet Start Threshold (%)")
-                    control_net_stop_threshold_gui = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="ControlNet Stop Threshold (%)")
+                    preprocess_resolution_gui = gr.Slider(minimum=64, maximum=2048, step=64, value=512,
+                                                          label="Preprocess Resolution")
+                    low_threshold_gui = gr.Slider(minimum=1, maximum=255, step=1, value=100,
+                                                  label="Canny low threshold")
+                    high_threshold_gui = gr.Slider(minimum=1, maximum=255, step=1, value=200,
+                                                   label="Canny high threshold")
+                    value_threshold_gui = gr.Slider(minimum=1, maximum=2.0, step=0.01, value=0.1,
+                                                    label="Hough value threshold (MLSD)")
+                    distance_threshold_gui = gr.Slider(minimum=1, maximum=20.0, step=0.01, value=0.1,
+                                                       label="Hough distance threshold (MLSD)")
+                    control_net_output_scaling_gui = gr.Slider(minimum=0, maximum=5.0, step=0.1, value=1,
+                                                               label="ControlNet Output Scaling in UNet")
+                    control_net_start_threshold_gui = gr.Slider(minimum=0, maximum=1, step=0.01, value=0,
+                                                                label="ControlNet Start Threshold (%)")
+                    control_net_stop_threshold_gui = gr.Slider(minimum=0, maximum=1, step=0.01, value=1,
+                                                               label="ControlNet Stop Threshold (%)")
 
                 with gr.Accordion("T2I adapter", open=False, visible=True):
                     t2i_adapter_preprocessor_gui = gr.Checkbox(value=True, label="T2i Adapter Preprocessor")
-                    adapter_conditioning_scale_gui = gr.Slider(minimum=0, maximum=5., step=0.1, value=1, label="Adapter Conditioning Scale")
-                    adapter_conditioning_factor_gui = gr.Slider(minimum=0, maximum=1., step=0.01, value=0.55, label="Adapter Conditioning Factor (%)")
+                    adapter_conditioning_scale_gui = gr.Slider(minimum=0, maximum=5., step=0.1, value=1,
+                                                               label="Adapter Conditioning Scale")
+                    adapter_conditioning_factor_gui = gr.Slider(minimum=0, maximum=1., step=0.01, value=0.55,
+                                                                label="Adapter Conditioning Factor (%)")
 
                 with gr.Accordion("Styles", open=False, visible=True):
 
@@ -1000,6 +1042,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     style_json_gui = gr.File(label="Style JSON File")
                     style_button = gr.Button("Load styles")
 
+
                     def load_json_style_file(json):
                         if not sd_gen.model:
                             gr.Info("First load the model")
@@ -1009,7 +1052,8 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         gr.Info(f"{len(sd_gen.model.STYLE_NAMES)} styles loaded")
                         return gr.update(value=None, choices=sd_gen.model.STYLE_NAMES)
 
-                    style_button.click(load_json_style_file, [style_json_gui], [style_prompt_gui])                        
+
+                    style_button.click(load_json_style_file, [style_json_gui], [style_prompt_gui])
 
                 with gr.Accordion("Textual inversion", open=False, visible=False):
                     active_textual_inversion_gui = gr.Checkbox(value=False, label="Active Textual Inversion in prompt")
@@ -1024,14 +1068,18 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
 
                     # Adetailer Sampler
                     adetailer_sampler_options = ["Use same sampler"] + scheduler_names[:-1]
-                    adetailer_sampler_gui = gr.Dropdown(label="Adetailer sampler:", choices=adetailer_sampler_options, value="Use same sampler")
+                    adetailer_sampler_gui = gr.Dropdown(label="Adetailer sampler:", choices=adetailer_sampler_options,
+                                                        value="Use same sampler")
 
                     with gr.Accordion("Detailfix A", open=False, visible=True):
                         # Adetailer A
                         adetailer_active_a_gui = gr.Checkbox(label="Enable Adetailer A", value=False)
-                        prompt_ad_a_gui = gr.Textbox(label="Main prompt", placeholder="Main prompt will be use", lines=3)
-                        negative_prompt_ad_a_gui = gr.Textbox(label="Negative prompt", placeholder="Main negative prompt will be use", lines=3)
-                        strength_ad_a_gui = gr.Number(label="Strength:", value=0.35, step=0.01, minimum=0.01, maximum=1.0)
+                        prompt_ad_a_gui = gr.Textbox(label="Main prompt", placeholder="Main prompt will be use",
+                                                     lines=3)
+                        negative_prompt_ad_a_gui = gr.Textbox(label="Negative prompt",
+                                                              placeholder="Main negative prompt will be use", lines=3)
+                        strength_ad_a_gui = gr.Number(label="Strength:", value=0.35, step=0.01, minimum=0.01,
+                                                      maximum=1.0)
                         face_detector_ad_a_gui = gr.Checkbox(label="Face detector", value=True)
                         person_detector_ad_a_gui = gr.Checkbox(label="Person detector", value=True)
                         hand_detector_ad_a_gui = gr.Checkbox(label="Hand detector", value=False)
@@ -1042,9 +1090,12 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     with gr.Accordion("Detailfix B", open=False, visible=True):
                         # Adetailer B
                         adetailer_active_b_gui = gr.Checkbox(label="Enable Adetailer B", value=False)
-                        prompt_ad_b_gui = gr.Textbox(label="Main prompt", placeholder="Main prompt will be use", lines=3)
-                        negative_prompt_ad_b_gui = gr.Textbox(label="Negative prompt", placeholder="Main negative prompt will be use", lines=3)
-                        strength_ad_b_gui = gr.Number(label="Strength:", value=0.35, step=0.01, minimum=0.01, maximum=1.0)
+                        prompt_ad_b_gui = gr.Textbox(label="Main prompt", placeholder="Main prompt will be use",
+                                                     lines=3)
+                        negative_prompt_ad_b_gui = gr.Textbox(label="Negative prompt",
+                                                              placeholder="Main negative prompt will be use", lines=3)
+                        strength_ad_b_gui = gr.Number(label="Strength:", value=0.35, step=0.01, minimum=0.01,
+                                                      maximum=1.0)
                         face_detector_ad_b_gui = gr.Checkbox(label="Face detector", value=True)
                         person_detector_ad_b_gui = gr.Checkbox(label="Person detector", value=True)
                         hand_detector_ad_b_gui = gr.Checkbox(label="Hand detector", value=False)
@@ -1067,9 +1118,12 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     save_generated_images_gui = gr.Checkbox(value=False, label="Save Generated Images")
                     image_storage_location_gui = gr.Textbox(value="./images", label="Image Storage Location")
                     retain_compel_previous_load_gui = gr.Checkbox(value=False, label="Retain Compel Previous Load")
-                    retain_detailfix_model_previous_load_gui = gr.Checkbox(value=False, label="Retain Detailfix Model Previous Load")
-                    retain_hires_model_previous_load_gui = gr.Checkbox(value=False, label="Retain Hires Model Previous Load")
-                    xformers_memory_efficient_attention_gui = gr.Checkbox(value=False, label="Xformers Memory Efficient Attention")
+                    retain_detailfix_model_previous_load_gui = gr.Checkbox(value=False,
+                                                                           label="Retain Detailfix Model Previous Load")
+                    retain_hires_model_previous_load_gui = gr.Checkbox(value=False,
+                                                                       label="Retain Hires Model Previous Load")
+                    xformers_memory_efficient_attention_gui = gr.Checkbox(value=False,
+                                                                          label="Xformers Memory Efficient Attention")
 
         with gr.Accordion("Examples and help", open=False, visible=True):
             gr.Markdown(
@@ -1106,7 +1160,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         None,
                         1.0,
                         None,
-                        1.0,                        
+                        1.0,
                         None,
                         1.0,
                         None,
@@ -1115,24 +1169,24 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         1152,
                         896,
                         "cagliostrolab/animagine-xl-3.1",
-                        None, # vae
+                        None,  # vae
                         "txt2img",
-                        None, # img conttol
-                        "Canny", # preprocessor
-                        512, # preproc resolution
-                        1024, # img resolution
-                        None, # Style prompt
-                        None, # Style json
-                        None, # img Mask
-                        0.35, # strength
-                        100, # low th canny
-                        200, # high th canny
-                        0.1, # value mstd
-                        0.1, # distance mstd
-                        1.0, # cn scale
-                        0., # cn start
-                        1., # cn end
-                        False, # ti
+                        None,  # img conttol
+                        "Canny",  # preprocessor
+                        512,  # preproc resolution
+                        1024,  # img resolution
+                        None,  # Style prompt
+                        None,  # Style json
+                        None,  # img Mask
+                        0.35,  # strength
+                        100,  # low th canny
+                        200,  # high th canny
+                        0.1,  # value mstd
+                        0.1,  # distance mstd
+                        1.0,  # cn scale
+                        0.,  # cn start
+                        1.,  # cn end
+                        False,  # ti
                         "Classic",
                         "Nearest",
                     ],
@@ -1149,7 +1203,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         None,
                         1.0,
                         None,
-                        1.0,                        
+                        1.0,
                         None,
                         1.0,
                         None,
@@ -1158,24 +1212,24 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         1024,
                         1024,
                         "kitty7779/ponyDiffusionV6XL",
-                        None, # vae
+                        None,  # vae
                         "txt2img",
-                        None, # img conttol
-                        "Canny", # preprocessor
-                        512, # preproc resolution
-                        1024, # img resolution
-                        None, # Style prompt
-                        None, # Style json
-                        None, # img Mask
-                        0.35, # strength
-                        100, # low th canny
-                        200, # high th canny
-                        0.1, # value mstd
-                        0.1, # distance mstd
-                        1.0, # cn scale
-                        0., # cn start
-                        1., # cn end
-                        False, # ti
+                        None,  # img conttol
+                        "Canny",  # preprocessor
+                        512,  # preproc resolution
+                        1024,  # img resolution
+                        None,  # Style prompt
+                        None,  # Style json
+                        None,  # img Mask
+                        0.35,  # strength
+                        100,  # low th canny
+                        200,  # high th canny
+                        0.1,  # value mstd
+                        0.1,  # distance mstd
+                        1.0,  # cn scale
+                        0.,  # cn start
+                        1.,  # cn end
+                        False,  # ti
                         "Classic",
                         "Nearest",
                     ],
@@ -1192,7 +1246,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         None,
                         1.0,
                         None,
-                        1.0,                        
+                        1.0,
                         None,
                         1.0,
                         None,
@@ -1201,24 +1255,24 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         1024,
                         1024,
                         "misri/epicrealismXL_v7FinalDestination",
-                        None, # vae
+                        None,  # vae
                         "canny ControlNet",
-                        "image.webp", # img conttol
-                        "Canny", # preprocessor
-                        1024, # preproc resolution
-                        1024, # img resolution
-                        None, # Style prompt
-                        None, # Style json
-                        None, # img Mask
-                        0.35, # strength
-                        100, # low th canny
-                        200, # high th canny
-                        0.1, # value mstd
-                        0.1, # distance mstd
-                        1.0, # cn scale
-                        0., # cn start
-                        1., # cn end
-                        False, # ti
+                        "image.webp",  # img conttol
+                        "Canny",  # preprocessor
+                        1024,  # preproc resolution
+                        1024,  # img resolution
+                        None,  # Style prompt
+                        None,  # Style json
+                        None,  # img Mask
+                        0.35,  # strength
+                        100,  # low th canny
+                        200,  # high th canny
+                        0.1,  # value mstd
+                        0.1,  # distance mstd
+                        1.0,  # cn scale
+                        0.,  # cn start
+                        1.,  # cn end
+                        False,  # ti
                         "Classic",
                         None,
                     ],
@@ -1235,7 +1289,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         None,
                         1.0,
                         None,
-                        1.0,                        
+                        1.0,
                         None,
                         1.0,
                         None,
@@ -1244,24 +1298,24 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         1024,
                         1024,
                         "misri/juggernautXL_juggernautX",
-                        None, # vae
+                        None,  # vae
                         "optical pattern ControlNet",
-                        "spiral_no_transparent.png", # img conttol
-                        "Canny", # preprocessor
-                        512, # preproc resolution
-                        1024, # img resolution
-                        None, # Style prompt
-                        None, # Style json
-                        None, # img Mask
-                        0.35, # strength
-                        100, # low th canny
-                        200, # high th canny
-                        0.1, # value mstd
-                        0.1, # distance mstd
-                        1.0, # cn scale
-                        0.05, # cn start
-                        0.75, # cn end
-                        False, # ti
+                        "spiral_no_transparent.png",  # img conttol
+                        "Canny",  # preprocessor
+                        512,  # preproc resolution
+                        1024,  # img resolution
+                        None,  # Style prompt
+                        None,  # Style json
+                        None,  # img Mask
+                        0.35,  # strength
+                        100,  # low th canny
+                        200,  # high th canny
+                        0.1,  # value mstd
+                        0.1,  # distance mstd
+                        1.0,  # cn scale
+                        0.05,  # cn start
+                        0.75,  # cn end
+                        False,  # ti
                         "Classic",
                         None,
                     ],
@@ -1278,7 +1332,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         None,
                         1.0,
                         None,
-                        1.0,                        
+                        1.0,
                         None,
                         1.0,
                         None,
@@ -1287,24 +1341,24 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         1024,
                         1024,
                         "cagliostrolab/animagine-xl-3.1",
-                        None, # vae
+                        None,  # vae
                         "lineart ControlNet",
-                        "color_image.png", # img conttol
-                        "Lineart", # preprocessor
-                        512, # preproc resolution
-                        896, # img resolution
-                        None, # Style prompt
-                        None, # Style json
-                        None, # img Mask
-                        0.35, # strength
-                        100, # low th canny
-                        200, # high th canny
-                        0.1, # value mstd
-                        0.1, # distance mstd
-                        1.0, # cn scale
-                        0., # cn start
-                        1., # cn end
-                        False, # ti
+                        "color_image.png",  # img conttol
+                        "Lineart",  # preprocessor
+                        512,  # preproc resolution
+                        896,  # img resolution
+                        None,  # Style prompt
+                        None,  # Style json
+                        None,  # img Mask
+                        0.35,  # strength
+                        100,  # low th canny
+                        200,  # high th canny
+                        0.1,  # value mstd
+                        0.1,  # distance mstd
+                        1.0,  # cn scale
+                        0.,  # cn start
+                        1.,  # cn end
+                        False,  # ti
                         "Compel",
                         None,
                     ],
@@ -1321,7 +1375,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         None,
                         1.0,
                         None,
-                        1.0,                        
+                        1.0,
                         None,
                         1.0,
                         None,
@@ -1330,24 +1384,24 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                         512,
                         512,
                         "digiplay/majicMIX_realistic_v7",
-                        None, # vae
+                        None,  # vae
                         "openpose ControlNet",
-                        "image.webp", # img conttol
-                        "Canny", # preprocessor
-                        512, # preproc resolution
-                        1024, # img resolution
-                        None, # Style prompt
-                        None, # Style json
-                        None, # img Mask
-                        0.35, # strength
-                        100, # low th canny
-                        200, # high th canny
-                        0.1, # value mstd
-                        0.1, # distance mstd
-                        1.0, # cn scale
-                        0., # cn start
-                        0.9, # cn end
-                        False, # ti
+                        "image.webp",  # img conttol
+                        "Canny",  # preprocessor
+                        512,  # preproc resolution
+                        1024,  # img resolution
+                        None,  # Style prompt
+                        None,  # Style json
+                        None,  # img Mask
+                        0.35,  # strength
+                        100,  # low th canny
+                        200,  # high th canny
+                        0.1,  # value mstd
+                        0.1,  # distance mstd
+                        1.0,  # cn scale
+                        0.,  # cn start
+                        0.9,  # cn end
+                        False,  # ti
                         "Compel",
                         "Nearest",
                     ],
@@ -1402,7 +1456,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
 
     with gr.Tab("Inpaint mask maker", render=True):
 
-        def create_mask_now(img, invert):            
+        def create_mask_now(img, invert):
             import numpy as np
             import time
 
@@ -1429,6 +1483,7 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
 
             return img["background"], rgb_mask
 
+
         with gr.Row():
             with gr.Column(scale=2):
                 # image_base = gr.ImageEditor(label="Base image", show_label=True, brush=gr.Brush(colors=["#000000"]))
@@ -1438,15 +1493,15 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
                     # enable crop (or disable it)
                     # transforms=["crop"],
                     brush=gr.Brush(
-                      default_size="16", # or leave it as 'auto'
-                      color_mode="fixed", # 'fixed' hides the user swatches and colorpicker, 'defaults' shows it
-                      # default_color="black", # html names are supported
-                      colors=[
-                        "rgba(0, 0, 0, 1)", # rgb(a)
-                        "rgba(0, 0, 0, 0.1)",
-                        "rgba(255, 255, 255, 0.1)",
-                        # "hsl(360, 120, 120)" # in fact any valid colorstring
-                      ]
+                        default_size="16",  # or leave it as 'auto'
+                        color_mode="fixed",  # 'fixed' hides the user swatches and colorpicker, 'defaults' shows it
+                        # default_color="black", # html names are supported
+                        colors=[
+                            "rgba(0, 0, 0, 1)",  # rgb(a)
+                            "rgba(0, 0, 0, 0.1)",
+                            "rgba(255, 255, 255, 0.1)",
+                            # "hsl(360, 120, 120)" # in fact any valid colorstring
+                        ]
                     ),
                     eraser=gr.Eraser(default_size="16")
                 )
@@ -1459,10 +1514,13 @@ with gr.Blocks(theme="NoCrypt/miku", css=CSS) as app:
 
             btn.click(create_mask_now, [image_base, invert_mask], [img_source, img_result])
 
+
             def send_img(img_source, img_result):
                 return img_source, img_result
+
+
             btn_send.click(send_img, [img_source, img_result], [image_control, image_mask_gui])
-            
+
     generate_button.click(
         fn=sd_gen.load_new_model,
         inputs=[