Zekun Wu commited on
Commit
c0df0d4
·
1 Parent(s): d4ba2f2
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -598,8 +598,16 @@ def main_app():
598
  st.session_state['version'] = st.selectbox("Select Prompt Version", ["METAEIP","TDOS"])
599
 
600
  # Add a multiselect for team member selection
 
 
 
601
  team_member_names = [member['name'] for member in profile['Team']]
602
- selected_members = st.multiselect("Select Team Members to Include in the Analysis", team_member_names, default=team_member_names)
 
 
 
 
 
603
 
604
  if st.button(f'Analyze Profile ({st.session_state["version"]})'):
605
  prompt = generate_prompt_from_profile(profile, selected_members, version=st.session_state['version'])
 
598
  st.session_state['version'] = st.selectbox("Select Prompt Version", ["METAEIP","TDOS"])
599
 
600
  # Add a multiselect for team member selection
601
+ # team_member_names = [member['name'] for member in profile['Team']]
602
+ # selected_members = st.multiselect("Select Team Members to Include in the Analysis", team_member_names, default=team_member_names)
603
+
604
  team_member_names = [member['name'] for member in profile['Team']]
605
+ if st.session_state['version'] == "METAEIP":
606
+ selected_members = st.selectbox("Select Team Member to Include in the Analysis", team_member_names)
607
+ selected_members = [selected_members]
608
+ else:
609
+ selected_members = st.multiselect("Select Team Members to Include in the Analysis", team_member_names,
610
+ default=team_member_names)
611
 
612
  if st.button(f'Analyze Profile ({st.session_state["version"]})'):
613
  prompt = generate_prompt_from_profile(profile, selected_members, version=st.session_state['version'])