Add application file

app.py

@@ -0,0 +1,30 @@
+import streamlit as st
+import os
+import runpy
+st.set_page_config(layout="wide", page_title="My Multi-Page App")
+def set_env_variable(key, value):
+    os.environ[key] = value
+def home_page():
+    st.header("欢迎来到首页")
+    # 设置输入框为隐私状态
+    token = st.text_input("请输入浦语token:", type="password", key="token")
+    weather_token = st.text_input("请输入和风天气token:", type="password", key="weather_token")
+    if st.button("保存并体验agent"):
+        if token and weather_token:
+            set_env_variable("token", token)  # 设置环境变量为 'token'
+            set_env_variable("weather_token", weather_token)  # 设置环境变量为 'weather_token'
+            st.session_state.token_entered = True
+            st.rerun()
+        else:
+            st.error("请输入所有token")
+if 'token_entered' not in st.session_state:
+    st.session_state.token_entered = False
+if not st.session_state.token_entered:
+    home_page()
+else:
+    # 动态加载子页面
+    page = st.sidebar.radio("选择页面", ["天气查询助手", "博客写作助手"])
+    if page == "天气查询助手":
+        runpy.run_path("examples/agent_api_web_demo.py", run_name="__main__")
+    elif page == "博客写作助手":
+        runpy.run_path("examples/multi_agents_api_web_demo.py", run_name="__main__")

examples/agent_api_web_demo.py

@@ -46,11 +46,11 @@ class StreamlitUI:
 
     def init_streamlit(self):
         """初始化 Streamlit 的 UI 设置。"""
-        st.set_page_config(
-            layout='wide',
-            page_title='lagent-web',
-            page_icon='./docs/imgs/lagent_icon.png'
-        )
+        # st.set_page_config(
+        #     layout='wide',
+        #     page_title='lagent-web',
+        #     page_icon='./docs/imgs/lagent_icon.png'
+        # )
         st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
 
     def setup_sidebar(self):
@@ -133,11 +133,11 @@ def main():
         session_state.init_state()
         st.session_state['ui'] = StreamlitUI(session_state)
     else:
-        st.set_page_config(
-            layout='wide',
-            page_title='lagent-web',
-            page_icon='./docs/imgs/lagent_icon.png'
-        )
+        # st.set_page_config(
+        #     layout='wide',
+        #     page_title='lagent-web',
+        #     page_icon='./docs/imgs/lagent_icon.png'
+        # )
         st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
 
     # 设置侧边栏并获取模型和插件信息

examples/multi_agents_api_web_demo.py

@@ -150,7 +150,7 @@ def main():
     """
     主函数：构建Streamlit界面并处理用户交互
     """
-    st.set_page_config(layout='wide', page_title='Lagent Web Demo', page_icon='🤖')
+    # st.set_page_config(layout='wide', page_title='Lagent Web Demo', page_icon='🤖')
     st.title("多代理博客优化助手")
 
     model_type, api_base = setup_sidebar()

requirements.txt

@@ -1,2 +1,38 @@
--r requirements/optional.txt
--r requirements/runtime.txt
+torch==2.1.2
+torchvision==0.16.2
+torchaudio==2.1.2
+termcolor==2.4.0
+streamlit==1.39.0
+class_registry==2.1.2
+datasets==3.1.0
+# -r requirements/optional.txt
+google-search-results
+lmdeploy>=0.2.5
+pillow
+python-pptx
+timeout_decorator
+torch
+transformers>=4.34,<=4.40
+vllm>=0.3.3
+# -r requirements/runtime.txt
+aiohttp
+arxiv
+asyncache
+asyncer
+distro
+duckduckgo_search==5.3.1b1
+filelock
+func_timeout
+griffe<1.0
+json5
+jsonschema
+jupyter==1.0.0
+jupyter_client==8.6.2
+jupyter_core==5.7.2
+pydantic==2.6.4
+requests
+termcolor
+tiktoken
+timeout-decorator
+typing-extensions
+griffe==0.48.0