1
with st.sidebar:   # with 是 streamlit 中的上下文管理器
2
    st.subheader("AI控制面板")
3

4
    # 新建会话
5
    if st.button("新建会话", width="stretch", icon="✏️"):
6
        # 1. 保存当前会话信息
7
        save_session()
8
        # 2. 创建新的会话
9
        if st.session_state.messages:      # 如果聊天信息非空, True; 否则, False
10
            st.session_state.messages = []
11
            st.session_state.current_session = generate_session_name()
12
            save_session()
13
            st.rerun()                     # 重新运行当前页面
14

15
    # 会话历史
16
    st.text("会话历史")
17
    session_list = load_sessions()
18
    for session in session_list:
19
        col1, col2 = st.columns([4, 1])
20
        with col1:
21
            # 加载会话信息
22
            # 三元运算符: 值1 if 条件 else 值2
23
            if st.button(session, width="stretch", icon="📄", key=f"load_{session}",
24
                         type="primary" if session == st.session_state.current_session else "secondary"):
25
                load_session(session)
26
                st.rerun()
27
        with col2:
28
            # 删除会话信息
29
            if st.button("", width="stretch", icon="❌️", key=f"delete_{session}"):
30
                delete_session(session)
31
                st.rerun()
32

33
    st.divider()
34

35
    # 伴侣信息
36
    st.subheader("伴侣信息")
37
    nick_name = st.text_input("昵称", placeholder="请输入昵称", value=st.session_state.nick_name)
38
    if nick_name:
39
        st.session_state.nick_name = nick_name
40
    nature = st.text_area("性格", placeholder="请输入性格", value=st.session_state.nature)
41
    if nature:
42
        st.session_state.nature = nature

1
import streamlit as st
2
import os
3
from openai import OpenAI
4
from datetime import datetime
5
import json
6

7
# 设置页面的配置项
8
st.set_page_config(
9
    page_title="AI智能伴侣",
10
    page_icon="🤖",
11
    layout="wide",
12
    initial_sidebar_state="expanded",
13
    menu_items={}
14
)
15

16
# 生成会话标识函数
17
def generate_session_name():
18
    return datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
19

20
# 保存会话信息函数
21
def save_session():
22
    if st.session_state.current_session:
23
        session_data = {
24
            "nick_name": st.session_state.nick_name,
25
            "nature": st.session_state.nature,
26
            "current_session": st.session_state.current_session,
27
            "messages": st.session_state.messages
28
        }
29
        if not os.path.exists("sessions"):
30
            os.mkdir("sessions")
31
        with open(f"sessions/{st.session_state.current_session}.json", "w", encoding="utf-8") as f:
32
            json.dump(session_data, f, ensure_ascii=False, indent=2)
33

34
# 加载所有的会话列表信息
35
def load_sessions():
36
    session_list = []
37
    if os.path.exists("sessions"):
38
        file_list = os.listdir("sessions")
39
        for filename in file_list:
40
            if filename.endswith(".json"):
41
                session_list.append(filename[:-5])
42
    session_list.sort(reverse=True)
43
    return session_list
44

45
# 加载指定的会话信息
46
def load_session(session_name):
47
    try:
48
        if os.path.exists(f"sessions/{session_name}.json"):
49
            with open(f"sessions/{session_name}.json", "r", encoding="utf-8") as f:
50
                session_data = json.load(f)
51
                st.session_state.messages = session_data["messages"]
52
                st.session_state.nick_name = session_data["nick_name"]
53
                st.session_state.nature = session_data["nature"]
54
                st.session_state.current_session = session_name
55
    except Exception:
56
        st.error("加载会话失败!")
57

58
# 删除会话信息函数
59
def delete_session(session_name):
60
    try:
61
        if os.path.exists(f"sessions/{session_name}.json"):
62
            os.remove(f"sessions/{session_name}.json")
63
            if session_name == st.session_state.current_session:
64
                st.session_state.messages = []
65
                st.session_state.current_session = generate_session_name()
66
    except Exception:
67
        st.error("删除会话失败!")
68

69
# 大标题 / Logo
70
st.title("AI智能伴侣")
71
st.logo("resources/logo.png")
72

73
# 系统提示词
74
system_prompt = """
75
        你叫 %s，现在是用户的真实伴侣，请完全代入伴侣角色。
76
        规则：
77
            1. 每次只回1条消息
78
            2. 禁止任何场景或状态描述性文字
79
            3. 匹配用户的语言
80
            4. 回复简短，像微信聊天一样
81
            5. 有需要的话可以用❤️🌸等emoji表情
82
            6. 用符合伴侣性格的方式对话
83
            7. 回复的内容, 要充分体现伴侣的性格特征
84
        伴侣性格：
85
            - %s
86
        你必须严格遵守上述规则来回复用户。
87
    """
88

89
# 初始化会话状态
90
if "messages" not in st.session_state:
91
    st.session_state.messages = []
92
if "nick_name" not in st.session_state:
93
    st.session_state.nick_name = "小甜甜"
94
if "nature" not in st.session_state:
95
    st.session_state.nature = "活泼开朗的东北姑娘"
96
if "current_session" not in st.session_state:
97
    st.session_state.current_session = generate_session_name()
98

99
# 展示聊天信息
100
st.text(f"会话名称: {st.session_state.current_session}")
101
for message in st.session_state.messages:
102
    st.chat_message(message["role"]).write(message["content"])
103

104
# 创建与AI大模型交互的客户端对象
105
client = OpenAI(api_key=os.environ.get('DEEPSEEK_API_KEY'), base_url="https://api.deepseek.com")
106

107
# 左侧的侧边栏
108
with st.sidebar:
109
    st.subheader("AI控制面板")
110

111
    # 新建会话
112
    if st.button("新建会话", width="stretch", icon="✏️"):
113
        save_session()
114
        if st.session_state.messages:
115
            st.session_state.messages = []
116
            st.session_state.current_session = generate_session_name()
117
            save_session()
118
            st.rerun()
119

120
    # 会话历史
121
    st.text("会话历史")
122
    session_list = load_sessions()
123
    for session in session_list:
124
        col1, col2 = st.columns([4, 1])
125
        with col1:
126
            if st.button(session, width="stretch", icon="📄", key=f"load_{session}",
127
                         type="primary" if session == st.session_state.current_session else "secondary"):
128
                load_session(session)
129
                st.rerun()
130
        with col2:
131
            if st.button("", width="stretch", icon="❌️", key=f"delete_{session}"):
132
                delete_session(session)
133
                st.rerun()
134

135
    st.divider()
136

137
    # 伴侣信息
138
    st.subheader("伴侣信息")
139
    nick_name = st.text_input("昵称", placeholder="请输入昵称", value=st.session_state.nick_name)
140
    if nick_name:
141
        st.session_state.nick_name = nick_name
142
    nature = st.text_area("性格", placeholder="请输入性格", value=st.session_state.nature)
143
    if nature:
144
        st.session_state.nature = nature
145

146
# 消息输入框
147
prompt = st.chat_input("请输入您要问的问题")
148
if prompt:
149
    st.chat_message("user").write(prompt)
150
    st.session_state.messages.append({"role": "user", "content": prompt})
151

152
    # 调用AI大模型（流式）
153
    response = client.chat.completions.create(
154
        model="deepseek-chat",
155
        messages=[
156
            {"role": "system", "content": system_prompt % (st.session_state.nick_name, st.session_state.nature)},
157
            *st.session_state.messages
158
        ],
159
        stream=True
160
    )
161

162
    response_message = st.empty()
163
    full_response = ""
164
    for chunk in response:
165
        if chunk.choices[0].delta.content is not None:
166
            content = chunk.choices[0].delta.content
167
            full_response += content
168
            response_message.chat_message("assistant").write(full_response)
169

170
    st.session_state.messages.append({"role": "assistant", "content": full_response})
171

172
    # 保存会话信息
173
    save_session()