chatbot/chat.py at main · baixing/chatbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import streamlit as st

from utils import render_messages, create_chatbot


# ------------------------------网页------------------------------
page_title = "ChatBot"  # 网页标题
st.set_page_config(
    page_title=page_title,
    page_icon="random",
    menu_items={
        "About": "Hi! **Jamie** developed me! Contact him [here](https://github.com/dwjamie/chatbot) if you have any problems."
    },
)

# ------------------------------配置------------------------------
# 侧边栏配置
with st.sidebar:
    st.header("机器人配置")
    system_message = st.text_area(label="角色设定", placeholder="机器人需要扮演什么角色？")
    model = st.selectbox(
        "模型",
        options=["GPT-3.5", "GPT-3.5 (Azure)", "GPT-4", "Claude 1", "Claude 2", "Claude Instant"],
    )
    temperature = st.slider("随机性", min_value=0.0, max_value=1.0, step=0.01, value=0.0)
    change_config = st.button(label="确认配置")
    clean_history = st.button(label="清空对话历史")

# 若第一次进入网页或切换了页面，则重置对话历史
if "current_page" not in st.session_state:
    st.session_state.current_page = page_title
if st.session_state.current_page != page_title or "chatbot" not in st.session_state:
    st.session_state.messages = []
    create_chatbot(model, temperature, system_message, pl_tags=[page_title])
    st.session_state.current_page = page_title

# 清空对话历史并重置ChatBot
if clean_history:
    st.session_state.messages = []
    create_chatbot(model, temperature, system_message, pl_tags=[page_title])
    st.info("对话历史已清空！", icon="✅")

# 确认ChatBot配置
if change_config:
    create_chatbot(model, temperature, system_message, pl_tags=[page_title])
    st.info("机器人配置已确认！", icon="✅")

# ------------------------------对话------------------------------
st.title(page_title)  # 渲染标题
render_messages(st.session_state.messages)  # 渲染对话历史
if user_message := st.chat_input("你好！"):
    # 渲染并储存用户消息
    with st.chat_message(name="user", avatar="🧑‍💻"):
        st.markdown(user_message)
    st.session_state.messages.append({"role": "user", "content": user_message})

    # 发给ChatBot
    assistant_response = st.session_state.chatbot.chat(user_message)

    # 渲染并储存ChatBot消息
    assistant_message = ""
    with st.chat_message(name="assistant", avatar="🤖"):
        placeholder = st.empty()
        for token in assistant_response:
            assistant_message += token
            placeholder.write(assistant_message + "▌")
        placeholder.empty()
        st.markdown(assistant_message)

    st.session_state.chatbot.add_message("assistant", assistant_message)
    st.session_state.messages.append(
        {"role": "assistant", "content": assistant_message}
    )