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__")
|