Update app.py

app.py

@@ -73,7 +73,7 @@ def get_screenshot(chat: list,height=5000,width=600,chatblock=[],theme="light",w
 
 
 def clear_fn():
-    return None,None,None
+    return None
 rand_val=random.randint(1,1111111111111111)
 def check_rand(inp,val):
     if inp==True:
@@ -122,10 +122,8 @@ with gr.Blocks() as app:
                     chatblock=gr.Dropdown(label="Chatblocks",info="Choose specific blocks of chat",choices=[c for c in range(1,40)],multiselect=True)
             
 
-
-    im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)
-    
+    chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,client_choice,seed,temp,tokens,top_p,rep_p],chat_b)
     go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,client_choice,seed,temp,tokens,top_p,rep_p],chat_b)
-    stop_btn.click(None,None,None,cancels=go)
-    clear_btn.click(clear_fn,None,[inp,sys_inp,chat_b])
+    stop_btn.click(None,None,None,cancels=[go,im_go,chat_sub])
+    clear_btn.click(clear_fn,None,[chat_b])
 app.queue(default_concurrency_limit=10).launch()