|
|
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)
|
|
|
|
