实现代码
import streamlit as st
import os
from openai import OpenAI
from streamlit import sidebar
import datetime
import jsonst.set_page_config(page_title="AI智能大模型",page_icon="🤖",# 页面布局layout="wide",#控制侧边栏状态initial_sidebar_state="expanded",menu_items={}
)# 保存会话信息
def save_session():# 构建新的会话对象session_data = {"friend_name": st.session_state.friend_name,"friend_nature": st.session_state.friend_nature,"current_session": st.session_state.current_session,"messages": st.session_state.messages}print("保存会话信息：",st.session_state.current_session)if not os.path.exists("sessions"):os.mkdir("sessions")# 保存会话数据with open(f"sessions/{st.session_state.current_session}.json", "w", encoding="utf-8") as f:json.dump(session_data, f, ensure_ascii=False, indent=2)# 加载所有的会话信息
def load_sessions():session_list = []# 加载sessions目录下的所有会话信息if os.path.exists("sessions"):file_list = os.listdir("sessions")for filename in file_list:if filename.endswith(".json"):session_list.append(filename[:-5])session_list.sort(reverse=True)return session_list# 加载指定会话信息
def load_session(session_name):try:if os.path.exists(f"sessions/{session_name}.json"):print("session:"+session_name)# 读取会话数据with open(f"sessions/{session_name}.json","r",encoding="utf-8") as f:session_data = json.load(f)st.session_state.messages = session_data["messages"]st.session_state.friend_name = session_name["friend_name"]st.session_state.friend_nature = session_data["friend_nature"]st.session_state.current_session = session_nameexcept Exception:st.error("加载会话失败！")# 删除指定会话信息
def delete_session(session_name):try:if os.path.exists(f"sessions/{session_name}.json"):os.remove(f"sessions/{session_name}.json")# 如果删除的是当前会话，则更新当前会话列表展示if session_name == st.session_state.current_session:st.session_state.messages = []st.session_state.current_session = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")except Exception:st.error("删除会话失败！")# 大标题
st.title("AI智能大模型")# 调用AI大模型
client = OpenAI(api_key=os.environ.get('DEEPSEEK_API_KEY'),base_url="https://api.deepseek.com")# 系统提示词
system_prompt = ("你是%s，你是用户的真实朋友，请完全代入朋友角色""朋友性格：%s")# 初始化聊天信息
if "messages" not in st.session_state:st.session_state.messages = []if "friend_name" not in st.session_state:st.session_state.friend_name = "天天"if "friend_nature" not in st.session_state:st.session_state.friend_nature = "活泼开朗热情的河北姑娘"# 会话标识
if "current_session" not in st.session_state:now = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")st.session_state.current_session = nowst.text(f"会话名称：{st.session_state.current_session}")
# 展示聊天信息
for message in st.session_state.messages:st.chat_message(message["role"]).write(message["content"])# 左侧侧边栏
with sidebar:# 会话信息管理st.subheader("AI控制面板")# 新建会话if st.button("新建会话",width="stretch",icon="🖋"):# 1、保存当前会话信息if st.session_state.current_session:save_session()# 2、创建新的会话if st.session_state.messages:st.session_state.messages = []st.session_state.current_session = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")save_session()st.rerun()# 会话历史st.text("会话历史")session_list = load_sessions()for session in session_list:col1,col2 = st.columns([4,1])with col1:# 加载会话信息if st.button(session, width="stretch", icon="📄",key=f"load_{session}", type="primary" if session == st.session_state.current_session else "secondary"):st.session_state.current_session = sessionload_session(session)st.rerun()with col2:# 删除会话信息if st.button("",width="stretch",icon="❌",key=f"delete_{session}"):delete_session(session)st.rerun()# 分割线st.divider()st.subheader("朋友信息")friend_name = st.text_input("昵称",placeholder="请输入朋友的昵称",value=st.session_state.friend_name)if friend_name:st.session_state.friend_name = friend_namefriend_nature = st.text_input("性格",placeholder="请输入朋友的性格",value=st.session_state.friend_nature)if friend_nature:st.session_state.friend_nature = friend_nature# 消息输入框
prompt = st.chat_input("请输入您要问的问题")
if prompt:st.chat_message("user").write(prompt)print("调用AI大模型发送的提示词")# 保存用户输入的提示词st.session_state.messages.append({"role": "user", "content": prompt})# 调用AI模型，与其交互response = client.chat.completions.create(model="deepseek-chat",messages=[{"role": "system", "content": system_prompt%(st.session_state.friend_name, st.session_state.friend_nature)},# 会话记忆的实现*st.session_state.messages],stream=True)# 流式输出的结果解析方式response_message = st.empty() # 创建一个空的组件，用于展示大模型返回的结果full_response = ""for chunk in response:if chunk.choices[0].delta.content is not None:content = chunk.choices[0].delta.contentfull_response += contentresponse_message.chat_message("assistant").write(full_response)# 保存大模型返回的提示词st.session_state.messages.append({"role": "assistant", "content": full_response})# 保存会话信息save_session()
实现效果
左侧侧边栏可以设置AI的角色，灵活性较强