Update app.py

app.py

@@ -378,12 +378,10 @@ 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(
+    chat_event = prompt_txtbox.submit(
         before_chat_begin, None, [close, reset, regen],
         concurrency_limit=20,
-    )
-    
-    chat_event = prompt_txtbox.submit(
+    ).then(
         chat_stream, 
         [cur_arxiv_id, local_data, prompt_txtbox, chat_state],
         [prompt_txtbox, chatbot, local_data, close, reset, regen],