From 77e1bf7af1d9274ef3a04159f0bce208477821f1 Mon Sep 17 00:00:00 2001 From: Weitian Leung Date: Sat, 22 Jul 2023 23:01:35 +0800 Subject: [PATCH] Fix the issue where it requires two clicks on the send button to submit a chat. Uses chat_input instead the text_area and button, and only show the empty user/assistant placeholder when a new chat is made --- web_demo2.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/web_demo2.py b/web_demo2.py index 203cbdc..2c91ed3 100644 --- a/web_demo2.py +++ b/web_demo2.py @@ -45,19 +45,14 @@ for i, (query, response) in enumerate(st.session_state.history): st.markdown(query) with st.chat_message(name="assistant", avatar="assistant"): st.markdown(response) -with st.chat_message(name="user", avatar="user"): - input_placeholder = st.empty() -with st.chat_message(name="assistant", avatar="assistant"): - message_placeholder = st.empty() -prompt_text = st.text_area(label="用户命令输入", - height=100, - placeholder="请在这儿输入您的命令") +prompt_text = st.chat_input(placeholder="请在这儿输入您的命令") +if prompt_text: + with st.chat_message(name="user", avatar="user"): + st.markdown(prompt_text) + with st.chat_message(name="assistant", avatar="assistant"): + message_placeholder = st.empty() -button = st.button("发送", key="predict") - -if button: - input_placeholder.markdown(prompt_text) history, past_key_values = st.session_state.history, st.session_state.past_key_values for response, history, past_key_values in model.stream_chat(tokenizer, prompt_text, history, past_key_values=past_key_values,