refactor gui.py

app.py

@@ -123,12 +123,12 @@ for url_embed in download_embeds:
         )
 
 # Build list models
-embed_list = get_model_list(directory_embeds)
-model_list = get_model_list(directory_models)
-model_list = load_diffusers_format_model + model_list
-lora_model_list = get_model_list(directory_loras)
+embed_list: list = get_model_list(directory_embeds)
+model_list: list = get_model_list(directory_models)
+model_list: list = load_diffusers_format_model + model_list
+lora_model_list: list = get_model_list(directory_loras)
 lora_model_list.insert(0, "None")
-vae_model_list = get_model_list(directory_vaes)
+vae_model_list: list = get_model_list(directory_vaes)
 vae_model_list.insert(0, "None")
 
 
@@ -163,12 +163,7 @@ print('\033[33m🏁 Download and listing of valid models completed.\033[0m')
 #######################
 # GUI
 #######################
-import spaces
 import gradio as gr
-from PIL import Image
-import IPython.display
-import time, json
-from IPython.utils import capture
 import logging
 from utils.string_utils import extract_parameters
 from stablepy import logger
@@ -213,7 +208,14 @@ 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: str, task_name: str):
+def update_task_options(
+        model_name: str,
+        task_name: str):
+    """
+    :param model_name:
+    :param task_name:
+    :return:
+    """
     if model_name in model_list:
         if "xl" in model_name.lower():
             new_choices = sdxl_task
@@ -845,7 +847,6 @@ with gr.Blocks(css=CSS) as app:
                         "Detailfix",
                         open=False,
                         visible=True):
-
                     # Adetailer Inpaint Only
                     adetailer_inpaint_only_gui = gr.Checkbox(label="Inpaint only", value=True)
 
@@ -918,7 +919,10 @@ with gr.Blocks(css=CSS) as app:
                             open=False,
                             visible=True):
                         # Adetailer B
-                        adetailer_active_b_gui = gr.Checkbox(label="Enable Adetailer B", value=False)
+                        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",
@@ -1048,6 +1052,7 @@ with gr.Blocks(css=CSS) as app:
                 try using sampler DPM++ 1s or DPM2 with Compel or Classic prompt weights.
                 """
             )
+
             gr.Markdown(
                 """### The following examples perform specific tasks:
                 1. Generation with SDXL and upscale
@@ -1061,6 +1066,7 @@ with gr.Blocks(css=CSS) as app:
                 to preserve a person's appearance or a specific style based on an image.
                 """
             )
+
             gr.Examples(
                 examples=example_prompts,
                 fn=sd_gen.generate_pipeline,

gui.py

@@ -4,7 +4,6 @@ from stablepy import Model_Diffusers
 import torch
 import logging
 import random
-import spaces
 import gradio as gr
 
 from models.upscaler import upscaler_dict_gui