mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-22 23:11:05 +08:00
68 lines
2.8 KiB
Python
68 lines
2.8 KiB
Python
import streamlit as st
|
||
from loom_openai_plugins_frontend import loom_openai_plugins_frontend
|
||
|
||
from chatchat.webui_pages.utils import ApiRequest
|
||
from chatchat.webui_pages.loom_view_client import (
|
||
update_store,
|
||
start_plugin,
|
||
start_worker,
|
||
stop_worker,
|
||
)
|
||
|
||
|
||
def openai_plugins_page(api: ApiRequest, is_lite: bool = None):
|
||
|
||
|
||
with (st.container()):
|
||
|
||
if "worker_id" not in st.session_state:
|
||
st.session_state.worker_id = ''
|
||
if "plugins_name" not in st.session_state and "status" in st.session_state:
|
||
|
||
for k, v in st.session_state.status.get("status", {}).get("subscribe_status", []).items():
|
||
st.session_state.plugins_name = v.get("plugins_names", [])[0]
|
||
break
|
||
|
||
col1, col2 = st.columns([0.8, 0.2])
|
||
|
||
with col1:
|
||
event = loom_openai_plugins_frontend(plugins_status=st.session_state.status,
|
||
run_list_plugins=st.session_state.run_plugins_list,
|
||
launch_subscribe_info=st.session_state.launch_subscribe_info,
|
||
list_running_models=st.session_state.list_running_models,
|
||
model_config=st.session_state.model_config)
|
||
|
||
with col2:
|
||
st.write("操作")
|
||
if not st.session_state.run_plugins_list:
|
||
button_type_disabled = False
|
||
button_start_text = '启动'
|
||
else:
|
||
button_type_disabled = True
|
||
button_start_text = '已启动'
|
||
|
||
if event:
|
||
event_type = event.get("event")
|
||
if event_type == "BottomNavigationAction":
|
||
st.session_state.plugins_name = event.get("data")
|
||
st.session_state.worker_id = ''
|
||
# 不存在run_plugins_list，打开启动按钮
|
||
if st.session_state.plugins_name not in st.session_state.run_plugins_list \
|
||
or st.session_state.run_plugins_list:
|
||
button_type_disabled = False
|
||
button_start_text = '启动'
|
||
else:
|
||
button_type_disabled = True
|
||
button_start_text = '已启动'
|
||
if event_type == "CardCoverComponent":
|
||
st.session_state.worker_id = event.get("data")
|
||
|
||
st.button(button_start_text, disabled=button_type_disabled, key="start",
|
||
on_click=start_plugin)
|
||
|
||
if st.session_state.worker_id != '':
|
||
st.button("启动" + st.session_state.worker_id, key="start_worker",
|
||
on_click=start_worker)
|
||
st.button("停止" + st.session_state.worker_id, key="stop_worker",
|
||
on_click=stop_worker)
|