Spaces:
Running
Running
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(""" | |
<div style="text-align: center; max-width: 1200px; margin: 0 auto;"> | |
<div> | |
<style> | |
h1 { | |
font-size: 6em; | |
color: #ffc99f; | |
margin-top: 30px; | |
margin-bottom: 30px; | |
text-shadow: 3px 3px 0 rgba(0, 0, 0, 1) !important; | |
} | |
h3 { | |
color: #ffc99f; !important; | |
} | |
h4 { | |
display: inline-block; | |
color: #ffffff !important; | |
} | |
.wrapper img { | |
font-size: 98% !important; | |
white-space: nowrap !important; | |
text-align: center !important; | |
display: inline-block !important; | |
color: #ffffff !important; | |
} | |
.wrapper { | |
color: #ffffff !important; | |
} | |
.gradio-container { | |
background-image: linear-gradient(#254150, #1e2f40, #182634) !important; | |
color: #ffaa66 !important; | |
font-family: 'IBM Plex Sans', sans-serif !important; | |
} | |
.text-gray-500 { | |
color: #ffc99f !important; | |
} | |
.gr-box { | |
background-image: linear-gradient(#182634, #1e2f40, #254150) !important; | |
border-top-color: #000000 !important; | |
border-right-color: #ffffff !important; | |
border-bottom-color: #ffffff !important; | |
border-left-color: #000000 !important; | |
} | |
.gr-input { | |
color: #ffc99f; !important; | |
background-color: #254150 !important; | |
} | |
:root { | |
--neutral-100: #000000 !important; | |
} | |
</style> | |
<body> | |
<div class="center"><h1>Blitz Diffusion</h1> | |
</div> | |
</body> | |
</div> | |
<p style="margin-bottom: 1px; color: #ffaa66;"> | |
<h3>899 Stable Diffusion models, but why? For your enjoyment!</h3></p> | |
<br><div class="wrapper">9.3 <img src="https://huggingface.co/Yntec/DucHaitenLofi/resolve/main/NEW.webp" alt="NEW!" style="width:32px;height:16px;">This has become a legacy backup copy of old <u><a href="https://huggingface.co/spaces/Yntec/ToyWorld">ToyWorld</a></u>'s UI! Newer models added dailty over there! 25 new models since last update!</div> | |
<p style="margin-bottom: 1px; font-size: 98%"> | |
<br><h4>If a model is already loaded each new image takes less than <b>10</b> seconds to generate!</h4></p> | |
<p style="margin-bottom: 1px; color: #ffffff;"> | |
<br><div class="wrapper">Generate 6 images from 1 prompt at the <u><a href="https://huggingface.co/spaces/Yntec/PrintingPress">PrintingPress</a></u>, and use 6 different models at <u><a href="https://huggingface.co/spaces/Yntec/diffusion80xx">Huggingface Diffusion!</a></u>! | |
</p></p> | |
</div> | |
""") | |
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("""<style> .gr-button { | |
color: #ffffff !important; | |
text-shadow: 1px 1px 0 rgba(0, 0, 0, 1) !important; | |
background-image: linear-gradient(#76635a, #d2a489) !important; | |
border-radius: 24px !important; | |
border: solid 1px !important; | |
border-top-color: #ffc99f !important; | |
border-right-color: #000000 !important; | |
border-bottom-color: #000000 !important; | |
border-left-color: #ffc99f !important; | |
padding: 6px 30px; | |
} | |
.gr-button:active { | |
color: #ffc99f !important; | |
font-size: 98% !important; | |
text-shadow: 0px 0px 0 rgba(0, 0, 0, 1) !important; | |
background-image: linear-gradient(#d2a489, #76635a) !important; | |
border-top-color: #000000 !important; | |
border-right-color: #ffffff !important; | |
border-bottom-color: #ffffff !important; | |
border-left-color: #000000 !important; | |
} | |
.gr-button:hover { | |
filter: brightness(130%); | |
} | |
</style>""") | |
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) | |