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