Update app.py
Browse files
app.py
CHANGED
@@ -32,21 +32,38 @@ def clear_chat_history():
|
|
32 |
del st.session_state.messages
|
33 |
|
34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
def init_chat_history():
|
36 |
-
|
37 |
-
|
38 |
|
|
|
39 |
if "messages" in st.session_state:
|
40 |
for message in st.session_state.messages:
|
|
|
41 |
avatar = '🧑💻' if message["role"] == "user" else '🤖'
|
42 |
-
|
43 |
-
|
44 |
else:
|
|
|
45 |
st.session_state.messages = []
|
46 |
|
47 |
return st.session_state.messages
|
48 |
|
49 |
-
|
50 |
def main():
|
51 |
model, tokenizer = init_model()
|
52 |
messages = init_chat_history()
|
|
|
32 |
del st.session_state.messages
|
33 |
|
34 |
|
35 |
+
# def init_chat_history():
|
36 |
+
# with st.chat_message("assistant", avatar='🤖'):
|
37 |
+
# st.markdown("您好,我是百川大模型,很高兴为您服务🥰")
|
38 |
+
|
39 |
+
# if "messages" in st.session_state:
|
40 |
+
# for message in st.session_state.messages:
|
41 |
+
# avatar = '🧑💻' if message["role"] == "user" else '🤖'
|
42 |
+
# with st.chat_message(message["role"], avatar=avatar):
|
43 |
+
# st.markdown(message["content"])
|
44 |
+
# else:
|
45 |
+
# st.session_state.messages = []
|
46 |
+
|
47 |
+
# return st.session_state.messages
|
48 |
+
|
49 |
+
# 初始化聊天历史
|
50 |
def init_chat_history():
|
51 |
+
# 显示机器人的欢迎语
|
52 |
+
st.write("🤖: 您好,我是百川大模型,很高兴为您服务🥰")
|
53 |
|
54 |
+
# 如果会话状态中有消息记录,就显示出来
|
55 |
if "messages" in st.session_state:
|
56 |
for message in st.session_state.messages:
|
57 |
+
# 根据角色显示不同的头像
|
58 |
avatar = '🧑💻' if message["role"] == "user" else '🤖'
|
59 |
+
# 用st.write或st.markdown显示消息内容
|
60 |
+
st.write(f"{avatar}: {message['content']}")
|
61 |
else:
|
62 |
+
# 如果没有消息记录,就初始化一个空列表
|
63 |
st.session_state.messages = []
|
64 |
|
65 |
return st.session_state.messages
|
66 |
|
|
|
67 |
def main():
|
68 |
model, tokenizer = init_model()
|
69 |
messages = init_chat_history()
|