Create app.py
Browse files
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__")
|