File size: 3,356 Bytes
d0dbbdd c21ae5c 5e8b132 b4cd781 305b06e d0dbbdd 5e8b132 d0dbbdd 5e8b132 d0dbbdd 5e8b132 d0dbbdd 5e8b132 d0dbbdd 5e8b132 20698a7 5e8b132 c21ae5c 5e8b132 d0dbbdd 5e8b132 20698a7 e0b7ec8 088ae2c b4cd781 26d09bf 81d2307 5e8b132 305b06e d0dbbdd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | import gradio as gr
from core.settings import MODEL_MAP_CHECKPOINT
from .ui_components import (
create_base_parameter_ui, create_lora_settings_ui,
create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_krea2_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
create_conditioning_ui, create_vae_override_ui,
create_model_architecture_filter_ui, create_category_filter_ui,
create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui,
create_pid_ui
)
def create_ui():
"""Creates the UI components for the Txt2Img tab."""
prefix = "txt2img"
components = {}
with gr.Column():
components.update(create_model_architecture_filter_ui(prefix))
with gr.Row():
components.update(create_category_filter_ui(prefix))
components[f'base_model_{prefix}'] = gr.Dropdown(
label="Base Model",
choices=list(MODEL_MAP_CHECKPOINT.keys()),
value=list(MODEL_MAP_CHECKPOINT.keys())[0],
scale=3,
allow_custom_value=True
)
with gr.Column(scale=1):
components[f'run_{prefix}'] = gr.Button("Run", variant="primary")
components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3)
components[f'neg_prompt_{prefix}'] = gr.Text(label="Negative prompt", lines=3)
with gr.Row():
with gr.Column(scale=1):
param_defaults = {'w': 1024, 'h': 1024, 'cs_vis': False, 'cs_val': 1}
components.update(create_base_parameter_ui(prefix, param_defaults))
with gr.Column(scale=1):
components[f'result_{prefix}'] = gr.Gallery(label="Result", show_label=False, columns=2, object_fit="contain", height=627)
components.update(create_lora_settings_ui(prefix))
components.update(create_controlnet_ui(prefix))
components.update(create_anima_controlnet_lllite_ui(prefix))
components.update(create_diffsynth_controlnet_ui(prefix))
components.update(create_krea2_controlnet_ui(prefix))
components.update(create_ipadapter_ui(prefix))
components.update(create_flux1_ipadapter_ui(prefix))
components.update(create_sd3_ipadapter_ui(prefix))
components.update(create_embedding_ui(prefix))
components.update(create_style_ui(prefix))
components.update(create_conditioning_ui(prefix))
components.update(create_reference_latent_ui(prefix))
components.update(create_hidream_o1_reference_ui(prefix))
components.update(create_joyai_reference_ui(prefix))
components.update(create_krea2_identity_edit_ui(prefix))
components.update(create_krea2_reference_edit_ui(prefix))
components.update(create_qwen_image_edit_ui(prefix))
components.update(create_reference_image_ui(prefix))
components.update(create_vae_override_ui(prefix))
components.update(create_pid_ui(prefix))
return components |