93 lines
3.6 KiB
Python
93 lines
3.6 KiB
Python
# 运行方式：
|
||
# 1. 安装必要的包：pip install streamlit-option-menu streamlit-chatbox>=1.1.6
|
||
# 2. 运行本机fastchat服务：python server\llm_api.py 或者 运行对应的sh文件
|
||
# 3. 运行API服务器：python server/api.py。如果使用api = ApiRequest(no_remote_api=True)，该步可以跳过。
|
||
# 4. 运行WEB UI：streamlit run webui.py --server.port 7860
|
||
|
||
import streamlit as st
|
||
from webui_pages.utils import *
|
||
from streamlit_option_menu import option_menu
|
||
from webui_pages import *
|
||
|
||
api = ApiRequest(base_url="http://127.0.0.1:7861", no_remote_api=False)
|
||
|
||
if __name__ == "__main__":
|
||
st.set_page_config("langchain-chatglm WebUI", initial_sidebar_state="expanded")
|
||
|
||
if not chat_box.chat_inited:
|
||
st.toast(
|
||
f"欢迎使用 [`Langchain-Chatglm`](https://github.com/chatchat-space/langchain-chatglm) ! \n\n"
|
||
f"当前使用模型`{LLM_MODEL}`, 您可以开始提问了."
|
||
)
|
||
|
||
if "chat_list" not in st.session_state:
|
||
st.session_state["chat_list"] = {"对话1": {"need_rename": True, "chat_no": 1}}
|
||
if "cur_chat_name" not in st.session_state:
|
||
st.session_state["cur_chat_name"] = list(st.session_state["chat_list"].keys())[0]
|
||
if "need_chat_name" not in st.session_state:
|
||
st.session_state["need_chat_name"] = True
|
||
if "chat_count" not in st.session_state:
|
||
st.session_state["chat_count"] = 1
|
||
|
||
chat_list = [{"name": k, "chat_no": v.get("chat_no", 0)} for k, v in st.session_state.chat_list.items()]
|
||
chat_list = [x["name"] for x in sorted(chat_list, key=lambda x: x["chat_no"])]
|
||
pages1 = {i: {
|
||
"icon": "chat",
|
||
"func": dialogue_page,
|
||
} for i in chat_list}
|
||
|
||
pages2 = {
|
||
"新建对话": {
|
||
"icon": "plus-circle",
|
||
"func": dialogue_page,
|
||
},
|
||
"---": {
|
||
"icon": None,
|
||
"func": None
|
||
},
|
||
"知识库管理": {
|
||
"icon": "hdd-stack",
|
||
"func": knowledge_base_page,
|
||
},
|
||
}
|
||
pages = {**pages1, **pages2}
|
||
|
||
with st.sidebar:
|
||
options = chat_list + list(pages2)
|
||
icons = ["chat"] * len(chat_list) + [x["icon"] for x in pages2.values()]
|
||
|
||
default_index = list(pages).index(st.session_state["cur_chat_name"])
|
||
selected_page = option_menu(
|
||
"langchain-chatglm",
|
||
options=options,
|
||
icons=icons,
|
||
menu_icon="chat-quote",
|
||
default_index=default_index,
|
||
)
|
||
|
||
if selected_page == "新建对话":
|
||
cur_chat_name = st.session_state.get("cur_chat_name")
|
||
if (not st.session_state.get("create_chat")
|
||
and not st.session_state.get("renamde_chat")
|
||
and not st.session_state.get("delete_chat")):
|
||
st.session_state.chat_count += 1
|
||
chat_no = st.session_state.chat_count
|
||
new_chat_name = f"对话{chat_no}"
|
||
st.session_state.chat_list[new_chat_name] = {"need_rename": True, "chat_no": chat_no}
|
||
st.session_state["cur_chat_name"] = new_chat_name
|
||
st.experimental_rerun()
|
||
else:
|
||
if st.session_state.get("create_chat"):
|
||
st.session_state.create_chat = False
|
||
if st.session_state.get("renamde_chat"):
|
||
st.session_state.renamde_chat = False
|
||
st.experimental_rerun()
|
||
if st.session_state.get("delete_chat"):
|
||
st.session_state.delete_chat = False
|
||
st.experimental_rerun()
|
||
elif selected_page in st.session_state.chat_list:
|
||
st.session_state["selected_page"] = selected_page
|
||
|
||
if selected_page in pages:
|
||
pages[selected_page]["func"](api)
|