James McCool commited on
Commit
c1a7ed8
·
1 Parent(s): 74fde7d

Refactor session state management in Manage Portfolio tab to ensure base_frame_names is initialized correctly and streamline assignment of working_frame to Default.

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1741,9 +1741,9 @@ if selected_tab == 'Data Load':
1741
  del st.session_state['portfolio'], st.session_state['export_portfolio']
1742
 
1743
  if selected_tab == 'Manage Portfolio':
1744
- if 'origin_portfolio' in st.session_state and 'projections_df' in st.session_state:
1745
- if 'base_frame_names' not in st.session_state:
1746
  st.session_state['base_frame_names'] = ['Default']
 
1747
  with st.container():
1748
  reset_port_col, recalc_div_col, set_base_col, blank_reset_col, contest_size_col = st.columns([1, 1, 1, 6, 4])
1749
  with reset_port_col:
@@ -1802,8 +1802,8 @@ if selected_tab == 'Manage Portfolio':
1802
  st.session_state['working_frame']['Size'] = st.session_state['working_frame'].index.map(st.session_state['size_dict'])
1803
 
1804
  st.session_state['base_frame'] = predict_dupes(st.session_state['working_frame'], st.session_state['map_dict'], site_var, type_var, Contest_Size, strength_var, sport_var, salary_max)
1805
- st.session_state[f'Default'] = st.session_state['base_frame']
1806
- st.session_state['working_frame'] = st.session_state['base_frame']
1807
  # st.session_state['highest_owned_teams'] = st.session_state['projections_df'][~st.session_state['projections_df']['position'].isin(['P', 'SP'])].groupby('team')['ownership'].sum().sort_values(ascending=False).head(3).index.tolist()
1808
  # st.session_state['highest_owned_pitchers'] = st.session_state['projections_df'][st.session_state['projections_df']['position'].isin(['P', 'SP'])]['player_names'].sort_values(by='ownership', ascending=False).head(3).tolist()
1809
 
 
1741
  del st.session_state['portfolio'], st.session_state['export_portfolio']
1742
 
1743
  if selected_tab == 'Manage Portfolio':
1744
+ if 'base_frame_names' not in st.session_state:
 
1745
  st.session_state['base_frame_names'] = ['Default']
1746
+ if 'origin_portfolio' in st.session_state and 'projections_df' in st.session_state:
1747
  with st.container():
1748
  reset_port_col, recalc_div_col, set_base_col, blank_reset_col, contest_size_col = st.columns([1, 1, 1, 6, 4])
1749
  with reset_port_col:
 
1802
  st.session_state['working_frame']['Size'] = st.session_state['working_frame'].index.map(st.session_state['size_dict'])
1803
 
1804
  st.session_state['base_frame'] = predict_dupes(st.session_state['working_frame'], st.session_state['map_dict'], site_var, type_var, Contest_Size, strength_var, sport_var, salary_max)
1805
+ st.session_state['Default'] = st.session_state['base_frame']
1806
+ st.session_state['working_frame'] = st.session_state['Default']
1807
  # st.session_state['highest_owned_teams'] = st.session_state['projections_df'][~st.session_state['projections_df']['position'].isin(['P', 'SP'])].groupby('team')['ownership'].sum().sort_values(ascending=False).head(3).index.tolist()
1808
  # st.session_state['highest_owned_pitchers'] = st.session_state['projections_df'][st.session_state['projections_df']['position'].isin(['P', 'SP'])]['player_names'].sort_values(by='ownership', ascending=False).head(3).tolist()
1809