Update app.py

app.py

@@ -378,28 +378,34 @@ with gr.Blocks(css="constants/styles.css", theme=gr.themes.Soft()) as demo:
     
     chat_button.click(None, [cur_arxiv_id], [local_data, chatbot], js=OPEN_CHAT_IF)
 
-    prompt_txtbox.submit(
-        before_chat_begin, None, [close, reset, regen],
+    chat_event1 = prompt_txtbox.submit(
+        before_chat_begin, None, [reset, regen],
         concurrency_limit=20,
-    ).then(
+    )    
+    chat_event2 = chat_event1.then(
         chat_stream, 
         [cur_arxiv_id, local_data, prompt_txtbox, chat_state],
-        [prompt_txtbox, chatbot, local_data, close, reset, regen],
+        [prompt_txtbox, chatbot, local_data, reset, regen],
         concurrency_limit=20, queue=True
-    ).then(
+    )    
+    chat_event2.then(
         None, [cur_arxiv_id, local_data], None, 
         js=UPDATE_CHAT_HISTORY
     )
 
-    close.click(None, None, None,js=CLOSE_CHAT_IF)
+    close.click(
+        None, None, None, cancels=[chat_event1, chat_event2]
+    ).then(
+        None, None, None,js=CLOSE_CHAT_IF
+    )
     
     reset.click(
-        before_chat_begin, None, [close, reset, regen],
+        before_chat_begin, None, [reset, regen],
         concurrency_limit=20,
     ).then(
         chat_reset,
         [local_data, chat_state],
-        [prompt_txtbox, chatbot, local_data, close, reset, regen],
+        [prompt_txtbox, chatbot, local_data, reset, regen],
         concurrency_limit=20,
     ).then(
         None, [cur_arxiv_id, local_data], None,