mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-22 23:11:05 +08:00
71b528a2d1
* fix chat and knowledge_base_chat * 更新多卡部署 * update readme * update api and webui: 1. add download_doc to api 2. return local path or http url in kowledge_base_chat depends on no_remote_api 3. change assistant avater in webui * 解决多卡启动问题 * fix chat and knowledge_base_chat * 更新readme的lora加载方式 * update readme * 更新readme --------- Co-authored-by: liunux4odoo <liunu@qq.com>
57 lines
1.7 KiB
Python
57 lines
1.7 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 *
|
||
import os
|
||
|
||
api = ApiRequest(base_url="http://127.0.0.1:7861", no_remote_api=False)
|
||
|
||
if __name__ == "__main__":
|
||
st.set_page_config("Langchain-Chatchat WebUI", initial_sidebar_state="expanded")
|
||
|
||
if not chat_box.chat_inited:
|
||
st.toast(
|
||
f"欢迎使用 [`Langchain-Chatchat`](https://github.com/chatchat-space/Langchain-Chatchat) ! \n\n"
|
||
f"当前使用模型`{LLM_MODEL}`, 您可以开始提问了."
|
||
)
|
||
|
||
pages = {
|
||
"对话": {
|
||
"icon": "chat",
|
||
"func": dialogue_page,
|
||
},
|
||
"知识库管理": {
|
||
"icon": "hdd-stack",
|
||
"func": knowledge_base_page,
|
||
},
|
||
}
|
||
|
||
with st.sidebar:
|
||
st.image(
|
||
os.path.join(
|
||
"img",
|
||
"logo-long-chatchat-trans-v2.png"
|
||
),
|
||
use_column_width=True
|
||
)
|
||
options = list(pages)
|
||
icons = [x["icon"] for x in pages.values()]
|
||
|
||
default_index = 0
|
||
selected_page = option_menu(
|
||
"",
|
||
options=options,
|
||
icons=icons,
|
||
# menu_icon="chat-quote",
|
||
default_index=default_index,
|
||
)
|
||
|
||
if selected_page in pages:
|
||
pages[selected_page]["func"](api)
|