|
import gradio as gr |
|
import os |
|
import sys |
|
from pathlib import Path |
|
|
|
models = [ |
|
"Yntec/OpenLexica", |
|
"Yntec/MapleSyrup", |
|
] |
|
current_model = models[0] |
|
|
|
text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link") |
|
|
|
models2=[ |
|
gr.Interface.load(f"models/{models[0]}",live=True,preprocess=False), |
|
gr.Interface.load(f"models/{models[1]}",live=True,preprocess=False), |
|
] |
|
|
|
|
|
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): |
|
proc1=models2[model_choice] |
|
output1=proc1(inputs) |
|
return(output1) |
|
css="""""" |
|
|
|
|
|
with gr.Blocks(css='style.css') as demo: |
|
gr.HTML( |
|
""" |
|
<div style="text-align: center; max-width: 1200px; margin: 0 auto;"> |
|
<div> |
|
<style> |
|
h1 { |
|
font-size: 6em; |
|
color: #ffffff; |
|
margin-top: 30px; |
|
margin-bottom: 30px; |
|
text-shadow: 3px 3px 0 rgba(0, 0, 0, 1) !important; |
|
} |
|
</style> |
|
<body> |
|
<div class="center"><h1>ToyWorldXL 401</h1> |
|
</div> |
|
</body> |
|
</div> |
|
<p style="margin-bottom: 10px; color: #ffffff;"> |
|
PlaceHolder |
|
</p> |
|
<p style="margin-bottom: 10px; color: #ffaa66; font-size: 98%"> |
|
❤️ Press the Like Button if you enjoy my space! ❤️</a> |
|
</p> |
|
</div> |
|
""" |
|
) |
|
with gr.Blocks(css=css) as myface: |
|
gr.HTML() |
|
with gr.Row(): |
|
with gr.Tab("Title"): |
|
gr.HTML(""" <title>Minimum Multiplier</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;"> |
|
<h1>Fill the Textbox at the top and click Generate Image</h1> |
|
<br><h4>The first time you load a model it takes 200 seconds</h4> |
|
<br><h4>But after it loads each image takes 20 seconds to generate!</h4> |
|
|
|
""") |
|
|
|
with gr.Tab("Description"): |
|
gr.HTML("""<div style="text-align:center;"> |
|
<h4>As many Text-to-Image Models as I can fit here</h4><br> |
|
<h4>Suggest more up in the "Community" button</h4> |
|
|
|
</div>""") |
|
|
|
with gr.Tab("Tools"): |
|
with gr.Tab("View"): |
|
with gr.Row(): |
|
with gr.Column(style="width=50%, height=70%"): |
|
gr.Pil(label="Crop") |
|
with gr.Column(style="width=50%, height=70%"): |
|
gr.Pil(label="Crop") |
|
|
|
|
|
with gr.Tab("Draw"): |
|
with gr.Column(style="width=50%, height=70%"): |
|
gr.Pil(label="Crop") |
|
with gr.Column(style="width=50%, height=70%"): |
|
gr.Pil(label="Draw") |
|
|
|
|
|
gr.ImagePaint(label="Draw") |
|
|
|
with gr.Tab("Text"): |
|
with gr.Row(): |
|
|
|
with gr.Column(scale=50): |
|
gr.Textbox(label="", lines=8, interactive=True) |
|
|
|
|
|
with gr.Column(scale=50): |
|
gr.Textbox(label="", lines=8, interactive=True) |
|
|
|
with gr.Tab("Color Picker"): |
|
with gr.Row(): |
|
|
|
with gr.Column(scale=50): |
|
gr.ColorPicker(label="Color", interactive=True) |
|
|
|
|
|
with gr.Column(scale=50): |
|
gr.ImagePaint(label="Draw", interactive=True) |
|
with gr.Row(): |
|
with gr.Column(scale=100): |
|
magic1=gr.Textbox(lines=4) |
|
gr.HTML("""<style> .gr-button { |
|
color: white !important; |
|
border-color: #000000 !important; |
|
background: #006699 !important; |
|
}</style>""") |
|
run=gr.Button("Generate Image") |
|
with gr.Row(): |
|
with gr.Column(scale=100): |
|
|
|
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(style="width=800px"): |
|
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) |
|
use_short=gr.Button("Use Short Prompt") |
|
see_prompts=gr.Button("Extend Idea") |
|
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(concurrency_count=200) |
|
myface.launch(inline=True, show_api=False, max_threads=400) |
