| | from datetime import datetime
|
| |
|
| | import gradio as gr
|
| | from backend.device import get_device_name
|
| | from constants import APP_VERSION
|
| | from frontend.webui.controlnet_ui import get_controlnet_ui
|
| | from frontend.webui.generation_settings_ui import get_generation_settings_ui
|
| | from frontend.webui.image_to_image_ui import get_image_to_image_ui
|
| | from frontend.webui.image_variations_ui import get_image_variations_ui
|
| | from frontend.webui.lora_models_ui import get_lora_models_ui
|
| | from frontend.webui.models_ui import get_models_ui
|
| | from frontend.webui.text_to_image_ui import get_text_to_image_ui
|
| | from frontend.webui.upscaler_ui import get_upscaler_ui
|
| | from state import get_settings
|
| |
|
| | app_settings = get_settings()
|
| |
|
| |
|
| | def _get_footer_message() -> str:
|
| | version = f"<center><p> {APP_VERSION} "
|
| | current_year = datetime.now().year
|
| | footer_msg = version + (
|
| | f' © 2023 - {current_year} <a href="https://github.com/rupeshs">'
|
| | " Rupesh Sreeraman</a></p></center>"
|
| | )
|
| | return footer_msg
|
| |
|
| |
|
| | def get_web_ui() -> gr.Blocks:
|
| | def change_mode(mode):
|
| | global app_settings
|
| | app_settings.settings.lcm_diffusion_setting.use_lcm_lora = False
|
| | app_settings.settings.lcm_diffusion_setting.use_openvino = False
|
| | app_settings.settings.lcm_diffusion_setting.use_gguf_model = False
|
| | if mode == "LCM-LoRA":
|
| | app_settings.settings.lcm_diffusion_setting.use_lcm_lora = True
|
| | elif mode == "LCM-OpenVINO":
|
| | app_settings.settings.lcm_diffusion_setting.use_openvino = True
|
| | elif mode == "GGUF":
|
| | app_settings.settings.lcm_diffusion_setting.use_gguf_model = True
|
| |
|
| |
|
| |
|
| | if app_settings.settings.lcm_diffusion_setting.lora:
|
| | app_settings.settings.lcm_diffusion_setting.lora.enabled = False
|
| | if app_settings.settings.lcm_diffusion_setting.controlnet:
|
| | app_settings.settings.lcm_diffusion_setting.controlnet.enabled = False
|
| | theme = gr.themes.Default(
|
| | primary_hue="blue",
|
| | )
|
| | with gr.Blocks(
|
| | title="FastSD CPU",
|
| | theme=theme,
|
| | css="footer {visibility: hidden}",
|
| | ) as fastsd_web_ui:
|
| | gr.HTML("<center><H1>FastSD CPU</H1></center>")
|
| | gr.Markdown(
|
| | f"**Processor : {get_device_name()}**",
|
| | elem_id="processor",
|
| | )
|
| | current_mode = "LCM"
|
| | if app_settings.settings.lcm_diffusion_setting.use_openvino:
|
| | current_mode = "LCM-OpenVINO"
|
| | elif app_settings.settings.lcm_diffusion_setting.use_lcm_lora:
|
| | current_mode = "LCM-LoRA"
|
| | elif app_settings.settings.lcm_diffusion_setting.use_gguf_model:
|
| | current_mode = "GGUF"
|
| |
|
| | mode = gr.Radio(
|
| | ["LCM", "LCM-LoRA", "LCM-OpenVINO", "GGUF"],
|
| | label="Mode",
|
| | info="Current working mode",
|
| | value=current_mode,
|
| | )
|
| | mode.change(change_mode, inputs=mode)
|
| |
|
| | with gr.Tabs():
|
| | with gr.TabItem("Text to Image"):
|
| | get_text_to_image_ui()
|
| | with gr.TabItem("Image to Image"):
|
| | get_image_to_image_ui()
|
| | with gr.TabItem("Image Variations"):
|
| | get_image_variations_ui()
|
| | with gr.TabItem("Upscaler"):
|
| | get_upscaler_ui()
|
| | with gr.TabItem("Generation Settings"):
|
| | get_generation_settings_ui()
|
| | with gr.TabItem("Models"):
|
| | get_models_ui()
|
| | with gr.TabItem("Lora Models"):
|
| | get_lora_models_ui()
|
| | with gr.TabItem("ControlNet"):
|
| | get_controlnet_ui()
|
| |
|
| | gr.HTML(_get_footer_message())
|
| |
|
| | return fastsd_web_ui
|
| |
|
| |
|
| | def start_webui(
|
| | share: bool = False,
|
| | ):
|
| | webui = get_web_ui()
|
| | webui.queue()
|
| | webui.launch(share=share)
|
| |
|
