Spaces:
Sleeping
Sleeping
Commit
·
a019aa8
1
Parent(s):
7915e18
Add application file
Browse files
app.py
CHANGED
|
@@ -1,29 +1,42 @@
|
|
| 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)
|
| 15 |
-
set_env_variable("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 == "博客写作助手":
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
import os
|
| 3 |
import runpy
|
| 4 |
+
|
| 5 |
+
# 设置页面配置
|
| 6 |
st.set_page_config(layout="wide", page_title="My Multi-Page App")
|
| 7 |
+
|
| 8 |
+
# 设置环境变量的函数
|
| 9 |
def set_env_variable(key, value):
|
| 10 |
os.environ[key] = value
|
| 11 |
+
|
| 12 |
+
# 首页内容
|
| 13 |
def home_page():
|
| 14 |
st.header("欢迎来到首页")
|
| 15 |
+
|
| 16 |
# 设置输入框为隐私状态
|
| 17 |
token = st.text_input("请输入浦语token:", type="password", key="token")
|
| 18 |
weather_token = st.text_input("请输入和风天气token:", type="password", key="weather_token")
|
| 19 |
+
|
| 20 |
if st.button("保存并体验agent"):
|
| 21 |
if token and weather_token:
|
| 22 |
+
set_env_variable("token", token) # 设置环境变量为 'token'
|
| 23 |
+
set_env_variable("weather_token", weather_token) # 设置环境变量为 'weather_token'
|
| 24 |
st.session_state.token_entered = True
|
| 25 |
st.rerun()
|
| 26 |
else:
|
| 27 |
st.error("请输入所有token")
|
| 28 |
+
|
| 29 |
+
# 初始化 session_state
|
| 30 |
if 'token_entered' not in st.session_state:
|
| 31 |
st.session_state.token_entered = False
|
| 32 |
+
|
| 33 |
+
# 根据 session_state 显示首页或子页面
|
| 34 |
if not st.session_state.token_entered:
|
| 35 |
home_page()
|
| 36 |
else:
|
| 37 |
# 动态加载子页面
|
| 38 |
page = st.sidebar.radio("选择页面", ["天气查询助手", "博客写作助手"])
|
| 39 |
+
|
| 40 |
if page == "天气查询助手":
|
| 41 |
runpy.run_path("examples/agent_api_web_demo.py", run_name="__main__")
|
| 42 |
elif page == "博客写作助手":
|