add appp.py
Browse files
app.py
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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("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__")
|