diff --git a/webui.py b/webui.py index 8c0387b3..9e300003 100644 --- a/webui.py +++ b/webui.py @@ -1,5 +1,6 @@ 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 @@ -11,6 +12,11 @@ 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__": @@ -58,6 +64,13 @@ if __name__ == "__main__": "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( @@ -76,11 +89,13 @@ if __name__ == "__main__": default_index = 0 selected_page = option_menu( - "", + menu_title="", + key="selected_page", options=options, icons=icons, # menu_icon="chat-quote", default_index=default_index, + on_change=on_change, ) if selected_page in pages: diff --git a/webui_pages/openai_plugins/base.py b/webui_pages/openai_plugins/base.py index 4f041d4e..ffe85455 100644 --- a/webui_pages/openai_plugins/base.py +++ b/webui_pages/openai_plugins/base.py @@ -11,12 +11,7 @@ from webui_pages.loom_view_client import ( def openai_plugins_page(api: ApiRequest, is_lite: bool = None): - 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.container()): @@ -69,4 +64,4 @@ def openai_plugins_page(api: ApiRequest, is_lite: bool = None): 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) \ No newline at end of file + on_click=stop_worker)