Sunshine108 commited on
Commit
72cf1d4
·
1 Parent(s): 039ad96

Add application file

Browse files
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import os
3
+ import runpy
4
+ st.set_page_config(layout="wide", page_title="My Multi-Page App")
5
+ def set_env_variable(key, value):
6
+     os.environ[key] = value
7
+ def home_page():
8
+     st.header("欢迎来到首页")
9
+     # 设置输入框为隐私状态
10
+     token = st.text_input("请输入浦语token:", type="password", key="token")
11
+     weather_token = st.text_input("请输入和风天气token:", type="password", key="weather_token")
12
+     if st.button("保存并体验agent"):
13
+         if token and weather_token:
14
+             set_env_variable("token", token)  # 设置环境变量为 'token'
15
+             set_env_variable("weather_token", weather_token)  # 设置环境变量为 'weather_token'
16
+             st.session_state.token_entered = True
17
+             st.rerun()
18
+         else:
19
+             st.error("请输入所有token")
20
+ if 'token_entered' not in st.session_state:
21
+     st.session_state.token_entered = False
22
+ if not st.session_state.token_entered:
23
+     home_page()
24
+ else:
25
+     # 动态加载子页面
26
+     page = st.sidebar.radio("选择页面", ["天气查询助手", "博客写作助手"])
27
+     if page == "天气查询助手":
28
+         runpy.run_path("examples/agent_api_web_demo.py", run_name="__main__")
29
+     elif page == "博客写作助手":
30
+         runpy.run_path("examples/multi_agents_api_web_demo.py", run_name="__main__")
examples/agent_api_web_demo.py CHANGED
@@ -46,11 +46,11 @@ class StreamlitUI:
46
 
47
  def init_streamlit(self):
48
  """初始化 Streamlit 的 UI 设置。"""
49
- st.set_page_config(
50
- layout='wide',
51
- page_title='lagent-web',
52
- page_icon='./docs/imgs/lagent_icon.png'
53
- )
54
  st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
55
 
56
  def setup_sidebar(self):
@@ -133,11 +133,11 @@ def main():
133
  session_state.init_state()
134
  st.session_state['ui'] = StreamlitUI(session_state)
135
  else:
136
- st.set_page_config(
137
- layout='wide',
138
- page_title='lagent-web',
139
- page_icon='./docs/imgs/lagent_icon.png'
140
- )
141
  st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
142
 
143
  # 设置侧边栏并获取模型和插件信息
 
46
 
47
  def init_streamlit(self):
48
  """初始化 Streamlit 的 UI 设置。"""
49
+ # st.set_page_config(
50
+ # layout='wide',
51
+ # page_title='lagent-web',
52
+ # page_icon='./docs/imgs/lagent_icon.png'
53
+ # )
54
  st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
55
 
56
  def setup_sidebar(self):
 
133
  session_state.init_state()
134
  st.session_state['ui'] = StreamlitUI(session_state)
135
  else:
136
+ # st.set_page_config(
137
+ # layout='wide',
138
+ # page_title='lagent-web',
139
+ # page_icon='./docs/imgs/lagent_icon.png'
140
+ # )
141
  st.header(':robot_face: :blue[Lagent] Web Demo ', divider='rainbow')
142
 
143
  # 设置侧边栏并获取模型和插件信息
examples/multi_agents_api_web_demo.py CHANGED
@@ -150,7 +150,7 @@ def main():
150
  """
151
  主函数:构建Streamlit界面并处理用户交互
152
  """
153
- st.set_page_config(layout='wide', page_title='Lagent Web Demo', page_icon='🤖')
154
  st.title("多代理博客优化助手")
155
 
156
  model_type, api_base = setup_sidebar()
 
150
  """
151
  主函数:构建Streamlit界面并处理用户交互
152
  """
153
+ # st.set_page_config(layout='wide', page_title='Lagent Web Demo', page_icon='🤖')
154
  st.title("多代理博客优化助手")
155
 
156
  model_type, api_base = setup_sidebar()
requirements.txt CHANGED
@@ -1,2 +1,38 @@
1
- -r requirements/optional.txt
2
- -r requirements/runtime.txt
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.1.2
2
+ torchvision==0.16.2
3
+ torchaudio==2.1.2
4
+ termcolor==2.4.0
5
+ streamlit==1.39.0
6
+ class_registry==2.1.2
7
+ datasets==3.1.0
8
+ # -r requirements/optional.txt
9
+ google-search-results
10
+ lmdeploy>=0.2.5
11
+ pillow
12
+ python-pptx
13
+ timeout_decorator
14
+ torch
15
+ transformers>=4.34,<=4.40
16
+ vllm>=0.3.3
17
+ # -r requirements/runtime.txt
18
+ aiohttp
19
+ arxiv
20
+ asyncache
21
+ asyncer
22
+ distro
23
+ duckduckgo_search==5.3.1b1
24
+ filelock
25
+ func_timeout
26
+ griffe<1.0
27
+ json5
28
+ jsonschema
29
+ jupyter==1.0.0
30
+ jupyter_client==8.6.2
31
+ jupyter_core==5.7.2
32
+ pydantic==2.6.4
33
+ requests
34
+ termcolor
35
+ tiktoken
36
+ timeout-decorator
37
+ typing-extensions
38
+ griffe==0.48.0