ElvisWang111 commited on
Commit
84295d5
·
verified ·
1 Parent(s): c74debe

Update workflow/preference/pref_render.py

Browse files
Files changed (1) hide show
  1. workflow/preference/pref_render.py +20 -3
workflow/preference/pref_render.py CHANGED
@@ -5,7 +5,9 @@ def preferences_select():
5
 
6
  modeling_requirements = st.text_area(
7
  "请描述你的数据分析目标与需求",
8
- placeholder="例如:请帮我对数据进行可视化", height=200
 
 
9
  )
10
  st.session_state.additional_preference = modeling_requirements
11
 
@@ -52,7 +54,7 @@ def preferences_select():
52
  )
53
 
54
  preferences = None
55
- if st.button("▶️ 保存偏好设置"):
56
  preferences = {
57
  "报告风格": report_style,
58
  "模型偏好": model_pref,
@@ -68,14 +70,29 @@ def preferences_select():
68
  return preferences
69
 
70
 
 
 
 
 
 
 
 
 
 
 
 
71
  if __name__ == "__main__":
72
 
73
  st.title("偏好设置")
74
  st.markdown("---")
75
 
76
  c = st.columns(2)
 
 
 
77
  with c[0].expander('偏好设置', True):
78
  preferences_select()
79
-
 
80
 
81
 
 
5
 
6
  modeling_requirements = st.text_area(
7
  "请描述你的数据分析目标与需求",
8
+ placeholder="例如:请帮我对数据进行可视化",
9
+ height=200,
10
+ key="modeling_requirements"
11
  )
12
  st.session_state.additional_preference = modeling_requirements
13
 
 
54
  )
55
 
56
  preferences = None
57
+ if st.button("▶️ 保存偏好设置", use_container_width=True):
58
  preferences = {
59
  "报告风格": report_style,
60
  "模型偏好": model_pref,
 
70
  return preferences
71
 
72
 
73
+ def prep_chat(agent):
74
+ """渲染对话式建议区"""
75
+
76
+ with st.chat_message("assistant"):
77
+ st.write("我是 Autostat 自动模式决策助手,很高兴为您服务!\n\n"
78
+ "您可以在左侧边栏开启自动模式,我会协助您决策并一键完成所有分析")
79
+
80
+ if agent.plan is not None:
81
+ st.chat_message("assistant").write(agent.plan)
82
+
83
+
84
  if __name__ == "__main__":
85
 
86
  st.title("偏好设置")
87
  st.markdown("---")
88
 
89
  c = st.columns(2)
90
+
91
+ planner = st.session_state.planner_agent
92
+
93
  with c[0].expander('偏好设置', True):
94
  preferences_select()
95
+ with c[1].expander('自动模式决策报告', True):
96
+ prep_chat(planner)
97
 
98