import gradio as gr
from random import randint
from all_models import models
from externalmod import gr_Interface_load
def load_fn(models):
global models_load
models_load = {}
for model in models:
if model not in models_load.keys():
try:
m = gr_Interface_load(f'models/{model}')
except Exception as error:
print(error)
m = gr.Interface(lambda txt: None, ['text'], ['image'])
models_load.update({model: m})
load_fn(models)
num_models = 6
default_models = models[:num_models]
def extend_choices(choices):
return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
def update_imgbox(choices):
choices_plus = extend_choices(choices[:num_models])
return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
def update_imgbox_gallery(choices):
choices_plus = extend_choices(choices[:num_models])
return [gr.Gallery(None, label = m, visible = (m != 'NA')) for m in choices_plus]
def gen_fn(model_str, prompt):
if model_str == 'NA':
return None
noise = ""
rand = randint(1, 500)
for i in range(rand):
noise += " "
return models_load[model_str](f'{prompt} {noise}')
def gen_fn_gallery(model_str, prompt, gallery):
if gallery is None: gallery = []
if model_str == 'NA':
return gallery
noise = ""
rand = randint(1, 500)
for i in range(rand):
noise += " "
gallery.append(models_load[model_str](f'{prompt} {noise}'))
return gallery
CSS="""
.output { width=480px; height=480px !important; }
"""
with gr.Blocks(css=CSS) as demo:
with gr.Tab('The Dream'):
txt_input = gr.Textbox(label = 'Your prompt:', lines=4) #.style(container=False,min_width=1200)
gen_button = gr.Button('Generate up to 6 images in up to 3 minutes total')
stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
gr.HTML(
"""
Scroll down to see more images and select models.
"""
)
with gr.Row():
#output = [gr.Image(label = m, show_download_button=True, elem_classes="output", show_share_button=True) for m in default_models]
output = [gr.Gallery(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True, container=True, format="png", object_fit="contain") for m in default_models]
current_models = [gr.Textbox(m, visible = False) for m in default_models]
for m, o in zip(current_models, output):
#gen_event = gen_button.click(gen_fn, [m, txt_input], o)
gen_event = gen_button.click(gen_fn_gallery, [m, txt_input, o], o)
stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
with gr.Accordion('Model selection'):
model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 866 available!', value = default_models, interactive = True)
#model_choice.change(update_imgbox, model_choice, output)
model_choice.change(update_imgbox_gallery, model_choice, output)
model_choice.change(extend_choices, model_choice, current_models)
with gr.Row():
gr.HTML(
"""