|
import gradio as gr |
|
import os |
|
import requests |
|
import random |
|
import time |
|
r = requests.get(f'https://huggingface.co/spaces/xp3857/text-to-image/raw/main/css.css') |
|
css = r.text |
|
name2 = "runwayml/stable-diffusion-v1-5" |
|
|
|
model = gr.Interface.load(f"models/{name2}") |
|
models=[ |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
gr.Interface.load(f"models/{name2}"), |
|
] |
|
|
|
o = "V" |
|
|
|
t_out = (""" |
|
|
|
<link href="https://fonts.googleapis.com/css2?family=Poppins&display=swap" rel="stylesheet"> |
|
<!--Stylesheets--> |
|
<style media="screen"> |
|
*, |
|
*:before, |
|
*:after{ |
|
padding: 0; |
|
margin: 0; |
|
box-sizing: border-box; |
|
} |
|
|
|
#container{ |
|
position: fixed; |
|
width: 100%; |
|
height: 100%; |
|
top: 50%; |
|
left: 50%; |
|
z-index:19999; |
|
background: #ff000059; |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
.popup{ |
|
background-color: #ffffff; |
|
width: 420px; |
|
padding: 30px 40px; |
|
transform: translate(-50%,-50%); |
|
left: 50%; |
|
top: 50%; |
|
border-radius: 8px; |
|
font-family: "Poppins",sans-serif; |
|
text-align: center; |
|
position: fixed; |
|
width: 200px; |
|
height: 200px; |
|
top: 0px; |
|
left: 0px; |
|
z-index:199999; |
|
} |
|
.popup button{ |
|
display: block; |
|
margin: 0 0 20px auto; |
|
background-color: transparent; |
|
font-size: 30px; |
|
color: #ffffff; |
|
background: #03549a; |
|
border-radius: 100%; |
|
width: 40px; |
|
height: 40px; |
|
border: none; |
|
outline: none; |
|
cursor: pointer; |
|
} |
|
.popup h2{ |
|
margin-top: -20px; |
|
} |
|
.popup p{ |
|
font-size: 14px; |
|
text-align: justify; |
|
margin: 20px 0; |
|
line-height: 25px; |
|
} |
|
|
|
</style> |
|
|
|
<div id="container"> |
|
<div class="popup"> |
|
<button id="close">×</button> |
|
<h2>Max Loading Time Reached</h2> |
|
<p><center> |
|
Click to Continue, or use a simplier Prompt |
|
</center> |
|
</p> |
|
<a href="#">Let's Go</a> |
|
</div> |
|
</div> |
|
<!--Script--> |
|
<script type="text/javascript"> |
|
window.addEventListener("load", function(){ |
|
setTimeout( |
|
function open(event){ |
|
document.querySelector(".popup").style.display = "block"; |
|
}, |
|
2000 |
|
) |
|
}); |
|
|
|
|
|
document.querySelector("#close").addEventListener("click", function(){ |
|
document.querySelector(".popup").style.display = "none"; |
|
}); |
|
</script> |
|
|
|
""") |
|
m_out = (""" |
|
|
|
|
|
<div id="restart"> |
|
<h3 id="head">Loading Time Limit Reached.</h3><br> |
|
<h4 id="cont">Please Click Launch again to Continue, or choose a Simpler Prompt</h4> |
|
</div> |
|
|
|
""") |
|
|
|
def ac(): |
|
def clear(): |
|
return gr.update(value=0),gr.update(value=0) |
|
def start(): |
|
print ("start") |
|
stamp = time.time() |
|
return gr.update(value=stamp),gr.update(value=0) |
|
def end(stamp): |
|
ts = stamp + 10 |
|
ti = time.time() |
|
if ti > ts and stamp != 0: |
|
print("Cancelling") |
|
return gr.update(value=1),gr.HTML.update(f"{m_out}", visible=True) |
|
|
|
|
|
else: |
|
print("Passing") |
|
return gr.update(value=0),None |
|
|
|
def im_fn(put,fac="",h=None): |
|
if h == o: |
|
put = f"{put}{fac}" |
|
fac = f"{fac} " |
|
rn = random.randint(0, 19) |
|
model=models[rn] |
|
return model(put),fac |
|
elif h != o: |
|
return(None,None) |
|
def cl_fac(): |
|
return "",gr.HTML.update("", visible=False) |
|
with gr.Blocks(css=css) as b: |
|
|
|
with gr.Row(): |
|
put = gr.Textbox() |
|
btn1 = gr.Button() |
|
message=gr.HTML("", visible=False) |
|
with gr.Row(): |
|
out1 = gr.Image() |
|
out2 = gr.Image() |
|
with gr.Row(): |
|
out3 = gr.Image() |
|
out4 = gr.Image() |
|
|
|
with gr.Row(visible=False): |
|
h=gr.Textbox(value="V") |
|
t_state=gr.Number() |
|
t_switch=gr.Textbox(value=0) |
|
|
|
fac_b = gr.Textbox(value="",visible=False) |
|
btn1.click(cl_fac,None,[fac_b,message]) |
|
|
|
b1=btn1.click(start,None,[t_state,t_switch]) |
|
sta = t_state.change(end,t_state,[t_switch,message],every=1,show_progress=False) |
|
|
|
b2=btn1.click(im_fn,[put,fac_b,h],[out1,fac_b]) |
|
b3=out1.change(im_fn,[put,fac_b,h],[out2,fac_b]) |
|
b4=out2.change(im_fn,[put,fac_b,h],[out3,fac_b]) |
|
b5=out3.change(im_fn,[put,fac_b,h],[out4,fac_b]) |
|
|
|
t_switch.change(clear,None,[t_switch,fac_b], cancels=[sta,b2,b3,b4,b5]) |
|
|
|
b.queue(concurrency_count=100).launch(show_api=False) |
|
ac() |