Sunshine108 commited on
Commit
a019aa8
·
1 Parent(s): 7915e18

Add application file

Browse files
Files changed (1) hide show
  1. app.py +15 -2
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)  # 设置环境变量为 '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 == "博客写作助手":
 
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 == "博客写作助手":