Refactor session state management in app.py to enhance portfolio reset functionality and streamline player summary assignment. Introduce logic to toggle 'settings_base' state during form submissions, ensuring accurate handling of player exposures and improving user experience.

app.py

@@ -811,8 +811,8 @@ with tab2:
             col1, col2 = st.columns(2)
             with col1:
                 if st.button('Reset Portfolio', key='reset_port'):
+                    st.session_state['settings_base'] = True
                     st.session_state['working_frame'] = st.session_state['base_frame'].copy()
-                    st.session_state['player_summary'] = st.session_state['origin_player_exposures'].copy()
 
             with col2:
                 with st.form(key='contest_size_form'):
@@ -994,7 +994,9 @@ with tab2:
                         stack_remove = st.multiselect("If Specific Stacks, Which to remove?", options=sorted(list(set(stack_dict.values()))), default=[])
                     
                     submitted = st.form_submit_button("Submit")
+                    
                     if submitted:
+                        st.session_state['settings_base'] = False
                         parsed_frame = st.session_state['working_frame'].copy()
                         parsed_frame = parsed_frame[parsed_frame['Dupes'] <= max_dupes]
                         parsed_frame = parsed_frame[parsed_frame['salary'] >= min_salary]
@@ -1030,6 +1032,7 @@ with tab2:
                     
                     submitted = st.form_submit_button("Submit")
                     if submitted:
+                        st.session_state['settings_base'] = False
                         parsed_frame = st.session_state['working_frame'].copy()
                         if player_remove:
                             # Create mask for lineups that contain any of the removed players
@@ -1105,6 +1108,7 @@ with tab2:
                     
                     submitted = st.form_submit_button("Trim")
                     if submitted:
+                        st.session_state['settings_base'] = False
                         st.write('initiated')
                         parsed_frame = st.session_state['working_frame'].copy()
                         
@@ -1118,6 +1122,7 @@ with tab2:
                     lineup_target = st.number_input("Lineups to produce", value=150, min_value=1, step=1)
                     submitted = st.form_submit_button("Submit")
                     if submitted:
+                        st.session_state['settings_base'] = False
                         if preset_choice == 'Small Field (Heavy Own)':
                             parsed_frame = small_field_preset(st.session_state['working_frame'], lineup_target, excluded_cols, sport_var)
                         elif preset_choice == 'Large Field (Manage Diversity)':
@@ -1243,7 +1248,9 @@ with tab2:
             player_stats = []
             player_columns = [col for col in display_frame.columns if col not in excluded_cols]
             
-            if 'player_summary' not in st.session_state:
+            if st.session_state['settings_base'] and 'origin_player_exposures' in st.session_state:
+                st.session_state['player_summary'] = st.session_state['origin_player_exposures']
+            else:
                 if type_var == 'Showdown':
                     for player in player_names:
                         # Create mask for lineups where this player is Captain (first column)
@@ -1334,7 +1341,8 @@ with tab2:
                 player_summary = pd.DataFrame(player_stats)
                 player_summary = player_summary.sort_values('Lineup Count', ascending=False)
                 st.session_state['player_summary'] = player_summary.copy()
-                st.session_state['origin_player_exposures'] = player_summary.copy()
+                if 'origin_player_exposures' not in st.session_state:
+                    st.session_state['origin_player_exposures'] = player_summary.copy()
             
             st.subheader("Player Summary")
             st.dataframe(