Refactor session state management in Streamlit app to clear non-maintained keys within specific tab selections. This change improves data handling by ensuring only relevant session data is retained, while also adding debug output for better visibility into session state changes.

src/streamlit_app.py

@@ -8,12 +8,6 @@ print(f"Streamlit version: {st.__version__}")
 
 maintained_keys = ['tab_selector', 'site_selectbox', 'view_selectbox']
 
-for key in list(st.session_state.keys()):
-    if key not in maintained_keys:
-        del st.session_state[key]
-
-print(st.session_state)
-
 st.set_page_config(layout="wide")
 
 @st.cache_resource
@@ -358,6 +352,11 @@ selected_tab = st.segmented_control(
 )
 
 if selected_tab == 'Handbuilder':
+    for key in list(st.session_state.keys()):
+        if key not in maintained_keys:
+            del st.session_state[key]
+
+    print(st.session_state)
     st.session_state['handbuilder_data'] = init_handbuilder_data(site_var)
     t_stamp = f"Last Update: " + str(st.session_state['handbuilder_data']['timestamp'][0]) + f" CST"
     
@@ -760,6 +759,11 @@ if selected_tab == 'Handbuilder':
                 del st.session_state['saved_lineups']
 
 elif selected_tab == 'Stacks ROO':
+    for key in list(st.session_state.keys()):
+        if key not in maintained_keys:
+            del st.session_state[key]
+
+    print(st.session_state)
     player_stats, dk_stacks_raw, fd_stacks_raw, dk_roo_raw, fd_roo_raw, dk_sd_roo_raw, fd_sd_roo_raw, dk_id_map, fd_id_map, dk_sd_id_map, fd_sd_id_map = init_baselines()
     t_stamp = f"Last Update: " + str(dk_roo_raw['timestamp'][0]) + f" CST"
     with st.expander("Info and Filters"):
@@ -797,6 +801,11 @@ elif selected_tab == 'Stacks ROO':
         )
 
 elif selected_tab == 'Player ROO':
+    for key in list(st.session_state.keys()):
+        if key not in maintained_keys:
+            del st.session_state[key]
+
+    print(st.session_state)
     player_stats, dk_stacks_raw, fd_stacks_raw, dk_roo_raw, fd_roo_raw, dk_sd_roo_raw, fd_sd_roo_raw, dk_id_map, fd_id_map, dk_sd_id_map, fd_sd_id_map = init_baselines()
     t_stamp = f"Last Update: " + str(dk_roo_raw['timestamp'][0]) + f" CST"
     with st.expander("Info and Filters"):