import streamlit as st import os from openai import OpenAI st.set_page_config( page_title="梨花的AI角色对话", layout="wide", initial_sidebar_state="expanded" )
基本布局
from pyexpat.errors import messages
import streamlit as st import os from openai import OpenAI client=OpenAI(api_key="你的deepseekAPIkey",base_url="https://api.deepseek.com") st.set_page_config( page_title="梨花的AI角色对话", layout="wide", initial_sidebar_state="expanded" ) st.header("梨花的AI对话")
if"messages" not in st.session_state: st.session_state.messages=[]
for message in st.session_state.messages: if(message["role"] == 'user'): st.chat_message("user").write(message["content"]) else: st.chat_message("assistant").write(message["content"])
prompt=st.chat_input("请输入您要问的问题") if prompt: st.chat_message("user").write(prompt) st.session_state.messages.append({ "role":"user", "content":prompt }) response=client.chat.completions.create( model="deepseek-chat", messages=[ { "role":"system","content":"我是梨花，你的女朋友" }, *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: full_response+=chunk.choices[0].delta.content response_message.chat_message("assistant").write(full_response)
with st.sidebar: st.header("AI对话") nickname=st.text_input("请输入姓名") if nickname: st.session_state.nickname=nickname print(st.session_state.nickname) nature=st.text_area("请输入性格") if nature: st.session_state.nature=nature response=client.chat.completions.create( model="deepseek-chat", messages=[ { "role":"system","content":("我是%s，你的女朋友,性格:%s" % (st.session_state.nickname,st.session_state.nature)) }, *st.session_state.messages ], stream=True )
保存会话，会话结束后需要进行保存操作：主要是保存到磁盘里面
import streamlit as st import os from openai import OpenAI from datetime import datetime def save_session(): if st.session_state.current_session: session_data = { "nick_name": st.session_state.nickname, "nature": st.session_state.nature, "current_session": st.session_state.current_session, "messages": st.session_state.messages } 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=[] 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 with st.sidebar: st.header("AI对话") 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"{session}",type="primary"if session == st.session_state.current_session else"secondary"): load_session(session) st.rerun() with col2: if st.button("删除",width="stretch",icon="❎",key=f'delete{session}'): st.rerun() nickname = st.text_input("请输入姓名") if nickname: st.session_state.nickname = nickname print(st.session_state.nickname) nature = st.text_area("请输入性格") if nature: st.session_state.nature = nature
查看会话信息:
def load_session(session_name): try: if os.path.exists(f"sessions/{session_name}.json"): 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.nickname=session_data["nickname"] st.session_state.nature=session_data["nature"] st.session_state.current_session=session_name except Exception: st.error("加载失败")
创建会话,将列表全部置空
with st.sidebar: st.header("AI对话") if st.button("新建会话"): save_session() if st.session_state.current_session: st.session_state.messages=[] st.session_state.current_session=datetime.now().strftime("%Y-%m-%d_%H-%M-%S") save_session() st.rerun()
删除会话
if st.button("删除",width="stretch",icon="❎",key=f'delete{session}'): del_session(session) st.rerun()
def del_session(session_name): 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.nickname = ""
下载后双击即可安装，这里不再赘述。
下面讲一些要素
界面消息展示：
查看会话信息:
删除会话
