Update app.py

app.py

@@ -54,18 +54,21 @@ def ac():
             print("Passing")
             pass
         return t
-    def im_fn(put,fac="",h=None):
+    def im_fn(put,fac="",val,h=None):
         if h == o:
-            rn = random.randint(0, 19)
-            model=models[rn]
-            put = f"{put}{fac}"
-            fac = f"{fac} "
-            print (rn)
-            return model(put),fac
+            if int(val) != 0:
+                rn = random.randint(0, 19)
+                model=models[rn]
+                put = f"{put}{fac}"
+                fac = f"{fac} "
+                print (rn)
+                return model(put),fac,1
+            elif int(val)==0:
+                return None,"",0
         elif h != o:
-            return(None,None)
+            return(None,"")
     def cl_fac():
-        return "",None,None,None,None
+        return "",0,None,None,None,None
     with gr.Blocks() as b:
         with gr.Row():
             put = gr.Textbox()            
@@ -81,15 +84,17 @@ def ac():
             t_state=gr.Number()
             t_switch=gr.Textbox(value=0)
         fac_b = gr.Textbox(value="",visible=False)
-        
-        b0=btn1.click(cl_fac,None,[fac_b,out1,out2,out3,out4],show_progress=False)
+        fac_c = gr.Textbox(value=0)
+        fac_d = gr.Textbox(value=1)
 
+        
+        b0=btn1.click(cl_fac,None,[fac_b,fac_c,out1,out2,out3,out4],show_progress=False)
         b1=btn1.click(start,None,[t_state,t_switch])
         sta = t_state.change(end,t_state,t_switch,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])        
-        b4=out2.change(im_fn,[put,fac_b,h],[out3,fac_b])        
-        b5=out3.change(im_fn,[put,fac_b,h],[out4,fac_b])
+        b2=btn1.click(im_fn,[put,fac_b,fac_d,h],[out1,fac_b,fac_c])
+        b3=out1.change(im_fn,[put,fac_b,fac_c,h],[out2,fac_b,fac_c])        
+        b4=out2.change(im_fn,[put,fac_b,fac_c,h],[out3,fac_b,fac_c])        
+        b5=out3.change(im_fn,[put,fac_b,fac_c,h],[out4,fac_b,fac_c])
         t_switch.change(clear,t_switch,t_switch, cancels=[sta,b1,b2,b3,b4,b5])
 
     b.queue(concurrency_count=100).launch(show_api=False)