mirror of https://github.com/THUDM/ChatGLM2-6B
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 madepull/358/head
parent
1679b014c6
commit
77e1bf7af1
17
web_demo2.py
17
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,
|
||||
|
|
Loading…
Reference in New Issue