|
|
import gradio as gr |
|
|
from constants import APP_VERSION |
|
|
from frontend.webui.text_to_image_ui import get_text_to_image_ui |
|
|
from frontend.webui.image_to_image_ui import get_image_to_image_ui |
|
|
from frontend.webui.generation_settings_ui import get_generation_settings_ui |
|
|
from frontend.webui.models_ui import get_models_ui |
|
|
from frontend.webui.image_variations_ui import get_image_variations_ui |
|
|
from paths import FastStableDiffusionPaths |
|
|
from state import get_settings |
|
|
|
|
|
app_settings = get_settings() |
|
|
|
|
|
|
|
|
def _get_footer_message() -> str: |
|
|
version = f"<center><p> {APP_VERSION} " |
|
|
footer_msg = version + ( |
|
|
' © 2025 <a href="https://github.com/rupeshs">' |
|
|
" Rupesh Sreeraman</a> | AEGIS Bio-Digital Lab 10</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 |
|
|
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 |
|
|
|
|
|
with gr.Blocks( |
|
|
css=FastStableDiffusionPaths.get_css_path(), |
|
|
title="AEGIS Bio-Digital Lab 10", |
|
|
) as fastsd_web_ui: |
|
|
|
|
|
gr.HTML(""" |
|
|
<div style="text-align: center; margin-bottom: 20px;"> |
|
|
<img src="file/logo/aegis.jpeg" alt="AEGIS Logo" style="max-width: 200px; height: auto; border-radius: 10px; box-shadow: 0 4px 15px rgba(30, 60, 114, 0.3);"> |
|
|
</div> |
|
|
""") |
|
|
|
|
|
gr.HTML("<center><H1>🧬 AEGIS Bio-Digital Lab 10</H1></center>") |
|
|
gr.HTML("<center><p><strong>Artificially Expanded Genetic Information System</strong></p></center>") |
|
|
gr.HTML("<center><p><em>Visualizing the Fabric of Life and Technology Through Time</em></p></center>") |
|
|
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" |
|
|
|
|
|
mode = gr.Radio( |
|
|
["LCM", "LCM-LoRA", "LCM-OpenVINO"], |
|
|
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("Generation Settings"): |
|
|
get_generation_settings_ui() |
|
|
with gr.TabItem("Models"): |
|
|
get_models_ui() |
|
|
|
|
|
gr.HTML(_get_footer_message()) |
|
|
|
|
|
return fastsd_web_ui |
|
|
|
|
|
|
|
|
def start_webui( |
|
|
share: bool = False, |
|
|
): |
|
|
webui = get_web_ui() |
|
|
webui.launch(share=share) |
|
|
|
