refactor gui.py

app.py

@@ -1,12 +1,7 @@
 import spaces
 import os
-from stablepy import Model_Diffusers
-from stablepy.diffusers_vanilla.model import scheduler_names
 from stablepy.diffusers_vanilla.style_prompt_config import STYLE_NAMES
-import torch
 import re
-import shutil
-import random
 from stablepy import (
     CONTROLNET_MODEL_IDS,
     VALID_TASKS,
@@ -22,7 +17,6 @@ from stablepy import (
     SD15_TASKS,
     SDXL_TASKS,
 )
-import urllib.parse
 from config import (
     MINIMUM_IMAGE_NUMBER,
     MAXIMUM_IMAGE_NUMBER,
@@ -204,7 +198,6 @@ warnings.filterwarnings(
 
 logger.setLevel(logging.DEBUG)
 
-
 # init GuiSD
 sd_gen: object = GuiSD(
     model_list=model_list,
@@ -220,7 +213,7 @@ 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):
+def update_task_options(model_name: str, task_name: str):
     if model_name in model_list:
         if "xl" in model_name.lower():
             new_choices = sdxl_task

gui.py

@@ -1,21 +1,11 @@
 import spaces
 import os
 from stablepy import Model_Diffusers
-from stablepy.diffusers_vanilla.model import scheduler_names
-from stablepy.diffusers_vanilla.style_prompt_config import STYLE_NAMES
 import torch
-import re
-import shutil
+import logging
 import random
 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
 
 from models.upscaler import upscaler_dict_gui
 
@@ -23,7 +13,6 @@ logging.getLogger("diffusers").setLevel(logging.ERROR)
 import diffusers
 
 diffusers.utils.logging.set_verbosity(40)
-import warnings
 from utils.download_utils import download_things
 
 hf_token: str = os.environ.get("HF_TOKEN")
@@ -411,8 +400,6 @@ class GuiSD:
             "ip_adapter_scale": params_ip_scale,
         }
 
-        # print(pipe_params)
-
         random_number = random.randint(1, 100)
         if random_number < 25 and num_images < 3:
             if (not upscaler_model and