minimum-generated-pig commited on
Commit
87757ba
·
verified ·
1 Parent(s): 743550b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -26
app.py CHANGED
@@ -1,30 +1,20 @@
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("agent_api_web_demo.py", run_name="__main__")
29
-     elif page == "博客写作助手":
30
-         runpy.run_path("multi_agents_api_web_demo.py", run_name="__main__")
 
1
  import streamlit as st
 
2
  import runpy
3
+
4
+ # 设置页面配置
5
  st.set_page_config(layout="wide", page_title="My Multi-Page App")
6
+
 
7
  def home_page():
8
+ st.header("欢迎来到首页")
9
+
10
+ # 选择页面
11
+ page = st.sidebar.radio("选择页面", ["天气查询助手", "博客写作助手"])
12
+
13
+ # 根据选择加载子页面
14
+ if page == "天气查询助手":
15
+ runpy.run_path("agent_api_web_demo.py", run_name="__main__")
16
+ elif page == "博客写作助手":
17
+ runpy.run_path("multi_agents_api_web_demo.py", run_name="__main__")
18
+
19
+ # 加载页面
20
+ home_page()