mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-19 21:37:20 +08:00
5d422ca9a1
改变 Embeddings 模型改为使用框架 API，不再手动加载，删除自定义 Embeddings Keyword 代码 修改依赖文件，移除 torch transformers 等重依赖 暂时移出对 loom 的集成 后续： 1、优化目录结构 2、检查合并中有无被覆盖的 0.2.10 内容
102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
import streamlit as st
|
||
|
||
# from webui_pages.loom_view_client import update_store
|
||
# from webui_pages.openai_plugins import openai_plugins_page
|
||
from webui_pages.utils import *
|
||
from streamlit_option_menu import option_menu
|
||
from webui_pages.dialogue.dialogue import dialogue_page, chat_box
|
||
from webui_pages.knowledge_base.knowledge_base import knowledge_base_page
|
||
import os
|
||
import sys
|
||
from configs import VERSION
|
||
from server.utils import api_address
|
||
|
||
|
||
# def on_change(key):
|
||
# if key:
|
||
# update_store()
|
||
|
||
|
||
api = ApiRequest(base_url=api_address())
|
||
|
||
if __name__ == "__main__":
|
||
is_lite = "lite" in sys.argv
|
||
|
||
st.set_page_config(
|
||
"Langchain-Chatchat WebUI",
|
||
os.path.join("img", "chatchat_icon_blue_square_v2.png"),
|
||
initial_sidebar_state="expanded",
|
||
menu_items={
|
||
'Get Help': 'https://github.com/chatchat-space/Langchain-Chatchat',
|
||
'Report a bug': "https://github.com/chatchat-space/Langchain-Chatchat/issues",
|
||
'About': f"""欢迎使用 Langchain-Chatchat WebUI {VERSION}！"""
|
||
},
|
||
layout="wide"
|
||
|
||
)
|
||
|
||
# use the following code to set the app to wide mode and the html markdown to increase the sidebar width
|
||
st.markdown(
|
||
"""
|
||
<style>
|
||
[data-testid="stSidebar"][aria-expanded="true"] > div:first-child{
|
||
width: 350px;
|
||
}
|
||
[data-testid="stSidebar"][aria-expanded="false"] > div:first-child{
|
||
width: 600px;
|
||
margin-left: -600px;
|
||
}
|
||
|
||
""",
|
||
unsafe_allow_html=True,
|
||
)
|
||
pages = {
|
||
"对话": {
|
||
"icon": "chat",
|
||
"func": dialogue_page,
|
||
},
|
||
"知识库管理": {
|
||
"icon": "hdd-stack",
|
||
"func": knowledge_base_page,
|
||
},
|
||
# "模型服务": {
|
||
# "icon": "hdd-stack",
|
||
# "func": openai_plugins_page,
|
||
# },
|
||
}
|
||
# 更新状态
|
||
# if "status" not in st.session_state \
|
||
# or "run_plugins_list" not in st.session_state \
|
||
# or "launch_subscribe_info" not in st.session_state \
|
||
# or "list_running_models" not in st.session_state \
|
||
# or "model_config" not in st.session_state:
|
||
# update_store()
|
||
|
||
with st.sidebar:
|
||
st.image(
|
||
os.path.join(
|
||
"img",
|
||
"logo-long-chatchat-trans-v2.png"
|
||
),
|
||
use_column_width=True
|
||
)
|
||
st.caption(
|
||
f"""<p align="right">当前版本：{VERSION}</p>""",
|
||
unsafe_allow_html=True,
|
||
)
|
||
options = list(pages)
|
||
icons = [x["icon"] for x in pages.values()]
|
||
|
||
default_index = 0
|
||
selected_page = option_menu(
|
||
menu_title="",
|
||
key="selected_page",
|
||
options=options,
|
||
icons=icons,
|
||
# menu_icon="chat-quote",
|
||
default_index=default_index,
|
||
)
|
||
|
||
if selected_page in pages:
|
||
pages[selected_page]["func"](api=api, is_lite=is_lite)
|