Bread-F commited on
Commit
b9e68ee
·
verified ·
1 Parent(s): 817d2b6

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +47 -0
app.py ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ```python
3
+ import streamlit as st
4
+ import os
5
+ import runpy
6
+
7
+ # 从环境变量获取 token
8
+ token = os.getenv('token')
9
+ weather_token = os.getenv('weather_token')
10
+
11
+ st.set_page_config(layout="wide", page_title="My Multi-Page App")
12
+
13
+ def set_env_variable(key, value):
14
+ os.environ[key] = value
15
+
16
+ def home_page():
17
+ st.header("欢迎来到首页")
18
+ # 检查环境变量是否已存在
19
+ global token, weather_token
20
+ if not token:
21
+ token = st.text_input("请输入浦语 token:", type="password", key="token")
22
+ if not weather_token:
23
+ weather_token = st.text_input("请输入和风天气 token:", type="password", key="weather_token")
24
+
25
+ if st.button("保存并体验 agent"):
26
+ if token and weather_token:
27
+ set_env_variable("token", token) # 设置环境变量
28
+ set_env_variable("weather_token", weather_token)
29
+ st.session_state.token_entered = True
30
+ st.rerun()
31
+ else:
32
+ st.error("请输入所有 token")
33
+
34
+ # 检查会话状态
35
+ if 'token_entered' not in st.session_state:
36
+ # 如果 token 存在于环境变量中,直接跳过输入
37
+ st.session_state.token_entered = bool(token and weather_token)
38
+
39
+ if not st.session_state.token_entered:
40
+ home_page()
41
+ else:
42
+ # 动态加载子页面
43
+ page = st.sidebar.radio("选择页面", ["天气查询助手", "博客写作助手"])
44
+ if page == "天气查询助手":
45
+ runpy.run_path("examples/agent_api_web_demo.py", run_name="__main__")
46
+ elif page == "博客写作助手":
47
+ runpy.run_path("examples/multi_agents_api_web_demo.py", run_name="__main__")