8720e435f5
Re-write UI to use async codes: (1) for faster start, and (2) for better live preview. Removed opencv dependency Plan to support Linux soon
97 lines
5.3 KiB
Python
97 lines
5.3 KiB
Python
import gradio as gr
|
||
import sys
|
||
import time
|
||
import modules.path
|
||
import fooocus_version
|
||
import modules.html
|
||
import modules.async_worker as worker
|
||
|
||
from modules.sdxl_styles import style_keys, aspect_ratios
|
||
|
||
|
||
def generate_clicked(*args):
|
||
yield gr.update(interactive=False), \
|
||
gr.update(visible=True, value=modules.html.make_progress_html(1, 'Processing text encoding ...')), \
|
||
gr.update(visible=True, value=None), \
|
||
gr.update(visible=False)
|
||
|
||
worker.buffer.append(list(args))
|
||
finished = False
|
||
|
||
while not finished:
|
||
time.sleep(0.01)
|
||
if len(worker.outputs) > 0:
|
||
flag, product = worker.outputs.pop(0)
|
||
if flag == 'preview':
|
||
percentage, title, image = product
|
||
yield gr.update(interactive=False), \
|
||
gr.update(visible=True, value=modules.html.make_progress_html(percentage, title)), \
|
||
gr.update(visible=True, value=image) if image is not None else gr.update(), \
|
||
gr.update(visible=False)
|
||
if flag == 'results':
|
||
yield gr.update(interactive=True), \
|
||
gr.update(visible=False), \
|
||
gr.update(visible=False), \
|
||
gr.update(visible=True, value=product)
|
||
finished = True
|
||
return
|
||
|
||
|
||
block = gr.Blocks(title='Fooocus ' + fooocus_version.version, css=modules.html.css).queue()
|
||
with block:
|
||
with gr.Row():
|
||
with gr.Column():
|
||
progress_window = gr.Image(label='Preview', show_label=True, height=640, visible=False)
|
||
progress_html = gr.HTML(value=modules.html.make_progress_html(32, 'Progress 32%'), visible=False, elem_id='progress-bar', elem_classes='progress-bar')
|
||
gallery = gr.Gallery(label='Gallery', show_label=False, object_fit='contain', height=720, visible=True)
|
||
with gr.Row():
|
||
with gr.Column(scale=0.85):
|
||
prompt = gr.Textbox(show_label=False, placeholder="Type prompt here.", container=False, autofocus=True)
|
||
with gr.Column(scale=0.15, min_width=0):
|
||
run_button = gr.Button(label="Generate", value="Generate")
|
||
with gr.Row():
|
||
advanced_checkbox = gr.Checkbox(label='Advanced', value=False, container=False)
|
||
with gr.Column(scale=0.5, visible=False) as right_col:
|
||
with gr.Tab(label='Setting'):
|
||
performance_selction = gr.Radio(label='Performance', choices=['Speed', 'Quality'], value='Speed')
|
||
aspect_ratios_selction = gr.Radio(label='Aspect Ratios (width × height)', choices=list(aspect_ratios.keys()),
|
||
value='1152×896')
|
||
image_number = gr.Slider(label='Image Number', minimum=1, maximum=32, step=1, value=2)
|
||
image_seed = gr.Number(label='Random Seed', value=-1, precision=0)
|
||
negative_prompt = gr.Textbox(label='Negative Prompt', show_label=True, placeholder="Type prompt here.")
|
||
with gr.Tab(label='Style'):
|
||
style_selction = gr.Radio(show_label=False, container=True,
|
||
choices=style_keys, value='cinematic-default')
|
||
with gr.Tab(label='Advanced'):
|
||
with gr.Row():
|
||
base_model = gr.Dropdown(label='SDXL Base Model', choices=modules.path.model_filenames, value=modules.path.default_base_model_name, show_label=True)
|
||
refiner_model = gr.Dropdown(label='SDXL Refiner', choices=['None'] + modules.path.model_filenames, value=modules.path.default_refiner_model_name, show_label=True)
|
||
with gr.Accordion(label='LoRAs', open=True):
|
||
lora_ctrls = []
|
||
for i in range(5):
|
||
with gr.Row():
|
||
lora_model = gr.Dropdown(label=f'SDXL LoRA {i+1}', choices=['None'] + modules.path.lora_filenames, value=modules.path.default_lora_name if i == 0 else 'None')
|
||
lora_weight = gr.Slider(label='Weight', minimum=-2, maximum=2, step=0.01, value=modules.path.default_lora_weight)
|
||
lora_ctrls += [lora_model, lora_weight]
|
||
model_refresh = gr.Button(label='Refresh', value='Refresh All Files', variant='secondary')
|
||
|
||
def model_refresh_clicked():
|
||
modules.path.update_all_model_names()
|
||
results = []
|
||
results += [gr.update(choices=modules.path.model_filenames), gr.update(choices=['None'] + modules.path.model_filenames)]
|
||
for i in range(5):
|
||
results += [gr.update(choices=['None'] + modules.path.lora_filenames), gr.update()]
|
||
return results
|
||
|
||
model_refresh.click(model_refresh_clicked, [], [base_model, refiner_model] + lora_ctrls)
|
||
|
||
advanced_checkbox.change(lambda x: gr.update(visible=x), advanced_checkbox, right_col)
|
||
ctrls = [
|
||
prompt, negative_prompt, style_selction,
|
||
performance_selction, aspect_ratios_selction, image_number, image_seed
|
||
]
|
||
ctrls += [base_model, refiner_model] + lora_ctrls
|
||
run_button.click(fn=generate_clicked, inputs=ctrls, outputs=[run_button, progress_html, progress_window, gallery])
|
||
|
||
block.launch(inbrowser=True, server_name='0.0.0.0' if '--listen' in sys.argv else None)
|