Raymd9 commited on
Commit
b9575a2
·
1 Parent(s): cfc816f
Files changed (1) hide show
  1. app.py +21 -21
app.py CHANGED
@@ -3,28 +3,28 @@ 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__")
 
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__")