Update app.py

app.py

@@ -57,6 +57,19 @@ t_out = ("""
 
 }
 
+#restart_btn{
+    position: fixed;
+    width: 100%;
+    height: 100%;
+    top: 0px;
+    left: 0px;
+    z-index:19999;
+    background: #ff000059;
+}
+
+
+
+
 .popup{
     background-color: #ffffff;
     width: 420px;
@@ -143,7 +156,9 @@ 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),gr.HTML.update(f"{t_out}", visible=True)
+            return gr.update(value=1),gr.Button.update(elem_id="restart_btn", visible=True)
+            
         else:
             print("Passing")
             return gr.update(value=0),None
@@ -175,12 +190,20 @@ def ac():
             h=gr.Textbox(value="V")
             t_state=gr.Number()
             t_switch=gr.Textbox(value=0)
-            
+        restart = gr.Button(visible=False)
+
+        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)
 
         b1=btn1.click(start,None,[t_state,t_switch])
-        sta = t_state.change(end,t_state,[t_switch,message],every=1)
+        sta = t_state.change(end,t_state,[t_switch,restart],every=1)
 
         b2=btn1.click(im_fn,[put,fac_b,h],[out1,fac_b])
         b3=out1.change(im_fn,[put,fac_b,h],[out2,fac_b])