Update app.py

app.py

@@ -50,8 +50,8 @@ t_out = ("""
     position: fixed;
     width: 100%;
     height: 100%;
-    top: 0px;
-    left: 0px;
+    top: 50%;
+    left: 50%;
     z-index:19999;
     background: #ff000059;
 
@@ -59,12 +59,13 @@ t_out = ("""
 
 #restart_btn{
     position: fixed;
-    width: 100%;
-    height: 100%;
     top: 0px;
     left: 0px;
     z-index:19999;
     background: #ff000059;
+    width: 420px;
+    padding: 30px 40px;
+    transform: translate(-50%,-50%);
 }
 
 
@@ -142,11 +143,17 @@ document.querySelector("#close").addEventListener("click", function(){
     </script>
 
 """)
+m_out = ("""
+<div id="restart_btn">
+<h1><center>Loading Time Limit Reached.<br>
+Please Click Launch again to Continue, or choose a Simpler Prompt</h1></center>
+<div>
 
+""")
 
 def ac():
     def clear():
-        return gr.update(value=0),gr.update(value=0),gr.update(label="Timeout",open=True,visible=True)
+        return gr.update(value=0),gr.update(value=0)
     def start():
         print ("start")
         stamp = time.time()
@@ -156,8 +163,8 @@ def ac():
         ti = time.time()
         if ti > ts and stamp != 0:
             print("Cancelling")
-            #return gr.update(value=1),gr.HTML.update(f"{t_out}", visible=True)
-            return gr.update(value=1)
+            return gr.update(value=1),gr.HTML.update(f"{m_out}", visible=True)
+            #return gr.update(value=1)
             
         else:
             print("Passing")
@@ -173,7 +180,7 @@ def ac():
         elif h != o:
             return(None,None)
     def cl_fac():
-        return ""
+        return "",gr.HTML.update("", visible=False)
     with gr.Blocks(css=css) as b:
         with gr.Row():
             put = gr.Textbox()            
@@ -184,27 +191,18 @@ def ac():
         with gr.Row():
             out3 = gr.Image()
             out4 = gr.Image()
-        message=gr.HTML("", visible=True)
+        message=gr.HTML("", visible=False)
 
         with gr.Row(visible=False):
             h=gr.Textbox(value="V")
             t_state=gr.Number()
             t_switch=gr.Textbox(value=0)
-        with gr.Accordion(elem_id="restart_btn",label="Timeout",open=False, visible=False) as acc:
-            restart = gr.Button()
-
-        def un_btn():
-            return gr.Button.update(elem_id="blank", visible=False)
-
-        
-        restart.click(un_btn, None, restart)
-
         
         fac_b = gr.Textbox(value="",visible=False)
-        btn1.click(cl_fac,None,fac_b)
+        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],every=1)
+        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])