35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
|
import streamlit as st
|
|||
|
|
from gitdb.pack import write_stream_to_pack
|
|||
|
|
|
|||
|
|
from deepseek import get_response_stream
|
|||
|
|
# 会话管理状态：在程序执行过程中，无论代码重新运行多少次，会话状态里面的变量都会保存
|
|||
|
|
|
|||
|
|
st.title("黑马智聊机器人")
|
|||
|
|
|
|||
|
|
# st.session_state 对象，本质就是json对象字典，
|
|||
|
|
if "messages" not in st.session_state:
|
|||
|
|
st.session_state["messages"] = [{"role": "assistant", "content": "你好，我是黑马智聊机器人，有什么可以帮助你的么？"}]
|
|||
|
|
|
|||
|
|
for message in st.session_state.messages:
|
|||
|
|
with st.chat_message(message["role"]):
|
|||
|
|
st.write(message["content"])
|
|||
|
|
# 聊天输入
|
|||
|
|
user_input = st.chat_input() #你好
|
|||
|
|
|
|||
|
|
if user_input:
|
|||
|
|
#1 把用户的输入添加到会话状态里面messages中
|
|||
|
|
st.session_state.messages.append({"role": "user", "content": user_input})
|
|||
|
|
#2 在网页上渲染用户输入
|
|||
|
|
with st.chat_message("user"):
|
|||
|
|
st.write(user_input)
|
|||
|
|
#3 调用deepseek.py中的get_response方法，获取大模型的回复
|
|||
|
|
# 调用get_response_stream方法,流式获取大模型的回复
|
|||
|
|
response = get_response_stream(st.session_state.messages)
|
|||
|
|
with st.chat_message("assistant"):
|
|||
|
|
st.write_stream(response)
|
|||
|
|
|
|||
|
|
#4 把大模型的回复添加到会话状态里面messages中
|
|||
|
|
st.session_state.messages.append({"role": "assistant", "content": response})
|
|||
|
|
|
|||
|
|
|