import gradio as gr
import os
import sys
from pathlib import Path
from all_models import models
from externalmod import gr_Interface_load
from prompt_extend import extend_prompt
text_gen1 = extend_prompt
current_model = models[0]
#text_gen1=gr.Interface.load("spaces/phenomenon1981/MagicPrompt-Stable-Diffusion")
#text_gen1=gr.Interface.load("spaces/Yntec/prompt-extend")
#text_gen1=gr.Interface.load("spaces/daspartho/prompt-extend")
#text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
models2 = []
for model in models:
try:
models2.append(gr_Interface_load(f'models/{model}'))
except Exception as error:
print(error)
models2.append(gr.Interface(lambda: None, ['text'], ['image']))
def text_it1(inputs,text_gen1=text_gen1):
go_t1=text_gen1(inputs)
return(go_t1)
def set_model(current_model):
current_model = models[current_model]
return gr.update(label=(f"{current_model}"))
def send_it1(inputs, model_choice): #negative_prompt,
proc1=models2[model_choice]
output1=proc1(inputs)
#negative_prompt=negative_prompt
return(output1)
css=""""""
with gr.Blocks(css=css) as myface:
gr.HTML("""
899 Stable Diffusion models, but why? For your enjoyment!
9.3
This has become a legacy backup copy of old
ToyWorld's UI! Newer models added dailty over there! 25 new models since last update!
If a model is already loaded each new image takes less than 10 seconds to generate!
""")
with gr.Row():
with gr.Column(scale=100):
#Model selection dropdown
model_name1 = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model, interactive=True)
with gr.Row():
with gr.Column(scale=100):
magic1 = gr.Textbox(label="Your Prompt", lines=4) #Positive
#with gr.Column(scale=100):
#negative_prompt=gr.Textbox(label="Negative Prompt", lines=1)
gr.HTML("""""")
run = gr.Button("Generate Image")
with gr.Row():
with gr.Column():
output1 = gr.Image(label=(f"{current_model}"))
with gr.Row():
with gr.Column(scale=50):
input_text=gr.Textbox(label="Use this box to extend an idea automagically, by typing some words and clicking Extend Idea", lines=2)
see_prompts=gr.Button("Extend Idea -> overwrite the contents of the `Your PromptĀ“ box above")
use_short=gr.Button("Copy the contents of this box to the `Your PromptĀ“ box above")
def short_prompt(inputs):
return(inputs)
model_name1.change(set_model, inputs=model_name1, outputs=[output1])
run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1])
use_short.click(short_prompt,inputs=[input_text], outputs=magic1)
see_prompts.click(text_it1,inputs=[input_text], outputs=magic1)
myface.queue()
myface.launch(show_api=False)