Update app.py

app.py

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