Update app.py

app.py

@@ -377,21 +377,27 @@ 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)
-    close.click(None, None, None,js=CLOSE_CHAT_IF) 
+    close.click(None, None, None,js=CLOSE_CHAT_IF)
 
-    prompt_txtbox.submit(
+    chat_event = prompt_txtbox.submit(
         before_chat_begin, None, [close, reset, regen],
         concurrency_limit=20,
-    ).then(
+    )
+    
+    chat_event2 = chat_event.then(
         chat_stream, 
         [cur_arxiv_id, local_data, prompt_txtbox, chat_state],
         [prompt_txtbox, chatbot, local_data, close, reset, regen],
         concurrency_limit=20,
-    ).then(
+    )
+    
+    chat_event3 = chat_event2.then(
         None, [cur_arxiv_id, local_data], None, 
         js=UPDATE_CHAT_HISTORY
     )
 
+    close.click(None, None, None, cancels=[chat_event, chat_event2, chat_event3])
+    
     reset.click(
         before_chat_begin, None, [close, reset, regen],
         concurrency_limit=20,