Refactor player selection logic in Handbuilder tab of app.py to initialize session state DataFrame for player selections, ensuring all players are included when teams change. Update filtering mechanism to improve selection accuracy and user experience.

app.py

@@ -808,6 +808,8 @@ with tab4:
 
     if st.button("Clear Data", key='clear_handbuild'):
         st.session_state.handbuilder_editor_key += 1  # Change the key to force reset
+        # Reset all selections
+        st.session_state.handbuilder_editor_df = None
         st.rerun()
 
     # --- TEAM FILTER UI ---
@@ -824,20 +826,28 @@ with tab4:
         if checked:
             selected_teams.append(team)
 
-    # If no teams selected, show all teams
+    # --- PLAYER SELECTION STATE ---
+    # Initialize or update the session state DataFrame
+    if 'handbuilder_editor_df' not in st.session_state or st.session_state.handbuilder_editor_df is None:
+        # Start with all players, all unchecked
+        base_df = dk_roo[['Player', 'Position', 'Team', 'Salary', 'Median', 'Order', 'Hand', 'Own%']].copy()
+        base_df['Select'] = False
+        st.session_state.handbuilder_editor_df = base_df
+    else:
+        # Make sure all players are present (in case teams change)
+        base_df = st.session_state.handbuilder_editor_df
+
+    # Filter by selected teams (or all if none selected)
     if selected_teams:
-        player_select_df = dk_roo[
-            dk_roo['Team'].isin(selected_teams)
-        ][['Player', 'Position', 'Team', 'Salary', 'Median', 'Order', 'Hand', 'Own%']].copy()
+        filtered_df = base_df[base_df['Team'].isin(selected_teams)].copy()
     else:
-        player_select_df = dk_roo[['Player', 'Position', 'Team', 'Salary', 'Median', 'Order', 'Hand', 'Own%']].copy()
-    player_select_df['Select'] = False  # Add a checkbox column
+        filtered_df = base_df.copy()
 
     col1, col2 = st.columns([1, 2])
     with col2:
         st.subheader("Player Select")
         edited_df = st.data_editor(
-            player_select_df,
+            filtered_df,
             column_config={
                 "Select": st.column_config.CheckboxColumn(
                     "Select",
@@ -849,9 +859,12 @@ with tab4:
             hide_index=True,
             key=f"handbuilder_editor_{st.session_state.handbuilder_editor_key}"
         )
+        # Update the session state DataFrame with the new selections
+        # (merge back into the full base_df)
+        st.session_state.handbuilder_editor_df.update(edited_df)
 
     # Filter selected players for the lineup
-    selected_players = edited_df[edited_df['Select'] == True]
+    selected_players = st.session_state.handbuilder_editor_df[st.session_state.handbuilder_editor_df['Select'] == True]
     select_players_disp = selected_players[['Player', 'Team', 'Salary', 'Median', 'Own%']]
 
     with col1: