Update app.py

app.py

@@ -28,9 +28,13 @@ def init_model():
     return model, tokenizer
 
 
+# def clear_chat_history():
+#     del st.session_state.messages
+# 清空聊天历史
 def clear_chat_history():
-    del st.session_state.messages
-
+    # 把会话状态中的消息记录清空，并刷新页面
+    st.session_state.messages = []
+    st.experimental_rerun()
 
 # def init_chat_history():
 #     with st.chat_message("assistant", avatar='🤖'):
@@ -64,26 +68,50 @@ def init_chat_history():
 
     return st.session_state.messages
 
+# def main():
+#     model, tokenizer = init_model()
+#     messages = init_chat_history()
+
+#     if prompt := st.chat_input("Shift + Enter 换行, Enter 发送"):
+#         with st.chat_message("user", avatar='🧑‍💻'):
+#             st.markdown(prompt)
+#         messages.append({"role": "user", "content": prompt})
+#         print(f"[user] {prompt}", flush=True)
+#         with st.chat_message("assistant", avatar='🤖'):
+#             placeholder = st.empty()
+#             for response in model.chat(tokenizer, messages, stream=True):
+#                 placeholder.markdown(response)
+#                 if torch.backends.mps.is_available():
+#                     torch.mps.empty_cache()
+#         messages.append({"role": "assistant", "content": response})
+#         print(json.dumps(messages, ensure_ascii=False), flush=True)
+
+#         st.button("清空对话", on_click=clear_chat_history)
+# 主函数
 def main():
     model, tokenizer = init_model()
     messages = init_chat_history()
 
-    if prompt := st.chat_input("Shift + Enter 换行, Enter 发送"):
-        with st.chat_message("user", avatar='🧑‍💻'):
-            st.markdown(prompt)
-        messages.append({"role": "user", "content": prompt})
-        print(f"[user] {prompt}", flush=True)
-        with st.chat_message("assistant", avatar='🤖'):
-            placeholder = st.empty()
-            for response in model.chat(tokenizer, messages, stream=True):
-                placeholder.markdown(response)
-                if torch.backends.mps.is_available():
-                    torch.mps.empty_cache()
-        messages.append({"role": "assistant", "content": response})
+    # 在侧边栏或表单中创建一个输入框和一个提交按钮
+    with st.sidebar.form("chat_form"):
+        user_input = st.text_input("请输入您的问题")
+        submit_button = st.form_submit_button("发送")
+
+    # 如果用户点击了发送按钮，就把用户的输入添加到消息记录中，并显示出来
+    if submit_button:
+        messages.append({"role": "user", "content": user_input})
+        st.write(f"🧑‍💻: {user_input}")
+        print(f"[user] {user_input}", flush=True)
+
+        # 调用模型来生成回复，并添加到消息记录中，并显示出来
+        reply = model.chat(tokenizer, messages, stream=False)
+        messages.append({"role": "assistant", "content": reply})
+        st.write(f"🤖: {reply}")
         print(json.dumps(messages, ensure_ascii=False), flush=True)
 
-        st.button("清空对话", on_click=clear_chat_history)
-
+        # 如果有清空对话的按钮，就绑定清空聊天历史的函数
+        if st.button("清空对话"):
+            clear_chat_history()
 
 if __name__ == "__main__":
     main()