Update app.py

app.py

@@ -3,30 +3,11 @@ import os
 import requests
 import random
 import time
+
 name2 = "dreamlike-art/dreamlike-photoreal-2.0"
-#name2 = "runwayml/stable-diffusion-v1-5"
 
 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}"),
+	gr.Interface.load(f"models/{name2}"),
 ]
 #o = os.getenv("P")
 o = "V"
@@ -39,72 +20,78 @@ m_out = ("""
 """)
 loading=("""
 <div class="lds-ellipsis"><div></div><div></div><div></div><div></div></div>""")
-def ac():
-    def clear():
-        return gr.update(value=0),gr.update(value=0)
-    def start():
-        stamp = time.time()
-        return gr.update(value=stamp),gr.update(value=0)
-    def end(stamp):
-        ts = stamp + 120
-        ti = time.time()
-        if ti > ts and stamp != 0:
-            return gr.update(value=1),gr.HTML.update(f"{m_out}",visible=True)
-        else:
-            return gr.update(value=0),None
-    def im_fn(put,fac="",h=None):
-        try:
-            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)
-        except Exception:
-            return None, None 
-    def cl_fac():
-        return "",gr.HTML.update(f"{loading}")
-    with gr.Blocks() as b:
-        with gr.Row():
-            with gr.Column():
-                put = gr.Textbox()     
-            with gr.Column():
-                with gr.Row():
-                    btn1 = gr.Button("Run")
-                    btn2 = gr.Button("Clear")
-        message=gr.HTML("<div></div>")
-        message2=gr.HTML("",visible=False)
+def build():
+	def clear():
+		return gr.update(value=0),gr.update(value=0)
+	def start():
+		stamp = time.time()
+		return gr.update(value=stamp),gr.update(value=0)
+	def end(stamp):
+		ts = stamp + 360
+		ti = time.time()
+		if ti > ts and stamp != 0:
+			return gr.update(value=1),gr.HTML.update(f"{m_out}",visible=True)
+		else:
+			return gr.update(value=0),None
+	def im_fn(put,fac="",h=None):
+		try:
+			if h == o:
+				put = f"{put}{fac}"
+				fac = f"{fac} "
+				rn = 0
+				# rn = random.randint(0, 19)
+				model=models[rn]
+				return model(put),fac
+			elif h != o:
+				return(None,None)
+		except Exception as E:
+			return None, None 
+	def cl_fac():
+		return "",gr.HTML.update(f"{loading}")
+	with gr.Blocks() as b:
+		with gr.Row():
+			with gr.Column():
+				put = gr.Textbox()     
+			with gr.Column():
+				with gr.Row():
+					btn1 = gr.Button("Run")
+					btn2 = gr.Button("Clear")
+		message=gr.HTML("<div></div>")
+		message2=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():
+			out1 = gr.Image()
+			out2 = gr.Image()
+		with gr.Row():
+			out3 = gr.Image()
+			out4 = gr.Image()
+		with gr.Row():
+			out5 = gr.Image()
+			out6 = gr.Image()
 
-        with gr.Row(visible=False):
-            h=gr.Textbox(value="V")
-            t_state=gr.Number()
-            t_switch=gr.Textbox(value=0)
-        def clear_all():
-            return "",None,None,None,None,None,None,1,gr.HTML.update("<div></div>")
-        fac_b = gr.Textbox(value="",visible=False)
+		with gr.Row(visible=False):
+			h_variavel=gr.Textbox(value="V")
+			t_state=gr.Number()
+			t_switch=gr.Textbox(value=0)
+		def clear_all():
+			return "",None,None,None,None,None,None,1,gr.HTML.update("<div></div>")
+		fac_b = gr.Textbox(value="",visible=False)
 
-        def noth():
-            return gr.HTML.update("<div></div>")
-        #a1=btn1.click(noth,None,btn1,every=1)
-        btn1.click(cl_fac,None,[fac_b,message],show_progress=False)
-        b1=btn1.click(start,None,[t_state,t_switch],show_progress=True)
-        sta = t_state.change(end,t_state,[t_switch,message2],every=1,show_progress=True)
-        b2=btn1.click(im_fn,[put,fac_b,h],[out1,fac_b], show_progress=True)
-        b3=out1.change(im_fn,[put,fac_b,h],[out2,fac_b], show_progress=True)        
-        b4=out2.change(im_fn,[put,fac_b,h],[out3,fac_b], show_progress=True)        
-        b5=out3.change(im_fn,[put,fac_b,h],[out4,fac_b], show_progress=True)
-        b6=out4.change(noth,None,message, show_progress=False)
-        swi=t_switch.change(clear,None,[t_switch,fac_b], cancels=[sta,b2,b3,b4,b5],show_progress=False)
-        #btn2.click(noth,None,message,cancels=[b1,sta,b2,b3,b4,b5,swi],show_progress=False)
-        btn2.click(clear_all, None,[fac_b,put,out1,out2,out3,out4,t_state,t_switch,message],cancels=[b1,sta,b2,b3,b4,b5,swi],show_progress=False)
-    b.queue(concurrency_count=100).launch(show_api=False)
-ac()
+		def noth():
+			return gr.HTML.update("<div></div>")
+		#a1=btn1.click(noth,None,btn1,every=1)
+		btn1.click(cl_fac,None,[fac_b,message],show_progress=False)
+		b1=btn1.click(start,None,[t_state,t_switch],show_progress=True)
+		sta = t_state.change(end,t_state,[t_switch,message2],every=1,show_progress=True)
+		b2=btn1.click(im_fn,[put,fac_b,h_variavel],[out1,fac_b], show_progress=True)
+		b3=out1.change(im_fn,[put,fac_b,h_variavel],[out2,fac_b], show_progress=True)
+		b4=out2.change(im_fn,[put,fac_b,h_variavel],[out3,fac_b], show_progress=True)
+		b5=out3.change(im_fn,[put,fac_b,h_variavel],[out4,fac_b], show_progress=True)
+		b6=out4.change(im_fn,[put,fac_b,h_variavel],[out5,fac_b], show_progress=True)
+		b7=out5.change(im_fn,[put,fac_b,h_variavel],[out6,fac_b], show_progress=True)
+		b8=out6.change(noth,None,message, show_progress=False)
+		swi=t_switch.change(clear,None,[t_switch,fac_b], cancels=[sta,b2,b3,b4,b5,b6,b7],show_progress=False)
+		#btn2.click(noth,None,message,cancels=[b1,sta,b2,b3,b4,b5,swi],show_progress=False)
+		btn2.click(clear_all, None,[fac_b,put,out1,out2,out3,out4,t_state,t_switch,message],cancels=[b1,sta,b2,b3,b4,b5,swi],show_progress=False)
+	b.queue(concurrency_count=100).launch(show_api=False)
+build()