James McCool commited on
Commit
9a0fceb
·
1 Parent(s): c0977b2

Refactor position selection in Streamlit app to set default values for multiselect and simplify position filtering logic

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -9
src/streamlit_app.py CHANGED
@@ -189,11 +189,7 @@ if selected_tab == 'Stack Finder':
189
  team_var2 = st.multiselect('Which teams would you like to include in the analysis?', options = raw_baselines['Team'].unique(), key='team_var2')
190
  elif split_var2 == 'Full Slate Run':
191
  team_var2 = raw_baselines.Team.unique().tolist()
192
- pos_split2 = st.radio("Are you viewing all positions, specific groups, or specific positions?", ('All Positions', 'Specific Positions'), key='pos_split2')
193
- if pos_split2 == 'Specific Positions':
194
- pos_var2 = st.multiselect('What Positions would you like to view?', options = ['WR', 'TE', 'RB'])
195
- elif pos_split2 == 'All Positions':
196
- pos_var2 = 'All'
197
  with stack_info_column:
198
  if site_var == 'Draftkings':
199
  max_sal2 = st.number_input('Max Salary', min_value = 5000, max_value = 50000, value = 35000, step = 100, key='max_sal2')
@@ -240,10 +236,7 @@ if selected_tab == 'Stack Finder':
240
 
241
  stack_hold_container = st.empty()
242
  comb_list = []
243
- if pos_split2 == 'All Positions':
244
- raw_baselines = raw_baselines
245
- elif pos_split2 != 'All Positions':
246
- raw_baselines = raw_baselines[raw_baselines['Position'].str.contains('|'.join(pos_var2 + ['QB']))]
247
 
248
  # Create a position dictionary mapping players to their eligible positions
249
  pos_dict = dict(zip(raw_baselines.Player, raw_baselines.Position))
 
189
  team_var2 = st.multiselect('Which teams would you like to include in the analysis?', options = raw_baselines['Team'].unique(), key='team_var2')
190
  elif split_var2 == 'Full Slate Run':
191
  team_var2 = raw_baselines.Team.unique().tolist()
192
+ pos_var2 = st.multiselect('What Positions would you like to view?', options = ['WR', 'TE', 'RB'], default = ['WR', 'TE', 'RB'], key='pos_var2')
 
 
 
 
193
  with stack_info_column:
194
  if site_var == 'Draftkings':
195
  max_sal2 = st.number_input('Max Salary', min_value = 5000, max_value = 50000, value = 35000, step = 100, key='max_sal2')
 
236
 
237
  stack_hold_container = st.empty()
238
  comb_list = []
239
+ raw_baselines = raw_baselines[raw_baselines['Position'].str.contains('|'.join(pos_var2 + ['QB']))]
 
 
 
240
 
241
  # Create a position dictionary mapping players to their eligible positions
242
  pos_dict = dict(zip(raw_baselines.Player, raw_baselines.Position))