Update app.py

app.py

@@ -32,21 +32,38 @@ def clear_chat_history():
     del st.session_state.messages
 
 
+# def init_chat_history():
+#     with st.chat_message("assistant", avatar='🤖'):
+#         st.markdown("您好，我是百川大模型，很高兴为您服务🥰")
+
+#     if "messages" in st.session_state:
+#         for message in st.session_state.messages:
+#             avatar = '🧑‍💻' if message["role"] == "user" else '🤖'
+#             with st.chat_message(message["role"], avatar=avatar):
+#                 st.markdown(message["content"])
+#     else:
+#         st.session_state.messages = []
+
+#     return st.session_state.messages
+    
+# 初始化聊天历史
 def init_chat_history():
-    with st.chat_message("assistant", avatar='🤖'):
-        st.markdown("您好，我是百川大模型，很高兴为您服务🥰")
+    # 显示机器人的欢迎语
+    st.write("🤖: 您好，我是百川大模型，很高兴为您服务🥰")
 
+    # 如果会话状态中有消息记录，就显示出来
     if "messages" in st.session_state:
         for message in st.session_state.messages:
+            # 根据角色显示不同的头像
             avatar = '🧑‍💻' if message["role"] == "user" else '🤖'
-            with st.chat_message(message["role"], avatar=avatar):
-                st.markdown(message["content"])
+            # 用st.write或st.markdown显示消息内容
+            st.write(f"{avatar}: {message['content']}")
     else:
+        # 如果没有消息记录，就初始化一个空列表
         st.session_state.messages = []
 
     return st.session_state.messages
 
-
 def main():
     model, tokenizer = init_model()
     messages = init_chat_history()