Enhance player selection and lineup management in Handbuilder tab of Streamlit app by introducing dedicated buttons for adding and removing players. Update selection mode to 'single' for clarity, and improve feedback messages for user actions. Streamline the process of managing player slots and ensure better state management during updates.

src/streamlit_app.py

@@ -616,43 +616,52 @@ if selected_tab == 'Handbuilder':
     handbuilder_lineup_build_column, handbuilder_player_select_column = st.columns([1, 2])
     with handbuilder_player_select_column:
         st.subheader("Player Select")
+        
+        # Display player selection dataframe with single row selection
         event = st.dataframe(
             st.session_state['player_select_df'].style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').background_gradient(cmap='RdYlGn_r', subset=['Salary', 'Own']).format(precision=2),
             on_select="rerun",
-            selection_mode=["single-row"],
+            selection_mode="single-row",
             key="player_select_dataframe",
             height=500,
             hide_index=True
         )
-        # If a row is selected, add that player to the lineup and reset selection
-        if event and "rows" in event.selection and len(event.selection["rows"]) > 0:
-            idx = event.selection["rows"][0]
-            player_row = st.session_state['player_select_df'].iloc[[idx]]
-            eligible_positions = re.split(r'[/, ]+', player_row['Position'].iloc[0])
-            # Find the first eligible slot that is not full
-            slot_to_fill = None
-
-            for slot in ['QB', 'RB', 'WR', 'TE', 'UTIL', 'DST']:
-                if slot_counts.get(slot, 0) < position_limits.get(slot, 0):
-                    if slot == 'UTIL':
-                        if 'DST' not in eligible_positions and 'QB' not in eligible_positions:
+        
+        # Add Player button
+        if st.button("Add Selected Player to Lineup", key="add_player_button"):
+            if event and "rows" in event.selection and len(event.selection["rows"]) > 0:
+                idx = event.selection["rows"][0]
+                player_row = st.session_state['player_select_df'].iloc[[idx]]
+                eligible_positions = re.split(r'[/, ]+', player_row['Position'].iloc[0])
+                
+                # Find the first eligible slot that is not full
+                slot_to_fill = None
+                for slot in ['QB', 'RB', 'WR', 'TE', 'UTIL', 'DST']:
+                    if slot_counts.get(slot, 0) < position_limits.get(slot, 0):
+                        if slot == 'UTIL':
+                            if 'DST' not in eligible_positions and 'QB' not in eligible_positions:
+                                slot_to_fill = slot
+                                break
+                        elif slot in eligible_positions:
                             slot_to_fill = slot
                             break
-                    elif slot in eligible_positions:
-                        slot_to_fill = slot
-                        break
-
-            if slot_to_fill is not None:
-                # Avoid duplicates
-                if not player_row['Player'].iloc[0] in st.session_state['handbuilder_lineup']['Player'].values:
-                    # Add the slot info
-                    player_row = player_row.assign(Slot=slot_to_fill)
-                    st.session_state['handbuilder_lineup'] = pd.concat(
-                        [st.session_state['handbuilder_lineup'], player_row[['Player', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Own', 'Slot']]],
-                        ignore_index=True
-                    )
-            st.rerun()
 
+                if slot_to_fill is not None:
+                    # Avoid duplicates
+                    if not player_row['Player'].iloc[0] in st.session_state['handbuilder_lineup']['Player'].values:
+                        # Add the slot info
+                        player_row = player_row.assign(Slot=slot_to_fill)
+                        st.session_state['handbuilder_lineup'] = pd.concat(
+                            [st.session_state['handbuilder_lineup'], player_row[['Player', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Own', 'Slot']]],
+                            ignore_index=True
+                        )
+                        st.success(f"Added {player_row['Player'].iloc[0]} to {slot_to_fill} slot")
+                    else:
+                        st.warning(f"{player_row['Player'].iloc[0]} is already in the lineup")
+                else:
+                    st.error("No available slots for this player")
+            else:
+                st.warning("Please select a player first")
 
     with handbuilder_lineup_build_column:
         st.subheader("Lineup Build")
@@ -697,26 +706,29 @@ if selected_tab == 'Handbuilder':
 
         st.session_state['lineup_display_df'] = pd.DataFrame(display_rows, columns=display_columns)
 
-        # Show the lineup table with single-row selection for removal
-        event_remove = st.dataframe(
+        # Show the lineup table as a static display
+        st.dataframe(
             st.session_state['lineup_display_df'].style.background_gradient(axis=0).background_gradient(cmap='RdYlGn', subset=['Median']).background_gradient(cmap='RdYlGn_r', subset=['Salary', 'Own']).format(precision=2),
-            on_select="rerun",
-            selection_mode=["single-row"],
-            key="lineup_remove_dataframe",
             height=445,
             hide_index=True
         )
-
-        # If a row is selected and not blank, remove that player from the lineup
-        if event_remove and "rows" in event_remove.selection and len(event_remove.selection["rows"]) > 0:
-            idx = event_remove.selection["rows"][0]
-            player_to_remove = st.session_state['lineup_display_df'].iloc[idx]['Player']
-            slot_to_remove = st.session_state['lineup_display_df'].iloc[idx]['Slot']
-            if player_to_remove:  # Only remove if not blank
-                st.session_state['handbuilder_lineup'] = filled_lineup[
-                    ~((filled_lineup['Player'] == player_to_remove) & (filled_lineup['Slot'] == slot_to_remove))
-                ]
-                st.rerun()
+        
+        # Remove Player button
+        if st.button("Remove Selected Player from Lineup", key="remove_player_button"):
+            # For removal, we can use a simple dropdown or text input
+            if not filled_lineup.empty:
+                player_to_remove = st.selectbox(
+                    "Select player to remove:",
+                    options=filled_lineup['Player'].tolist(),
+                    key="remove_player_selectbox"
+                )
+                if st.button("Confirm Remove", key="confirm_remove_button"):
+                    st.session_state['handbuilder_lineup'] = st.session_state['handbuilder_lineup'][
+                        st.session_state['handbuilder_lineup']['Player'] != player_to_remove
+                    ]
+                    st.success(f"Removed {player_to_remove} from lineup")
+            else:
+                st.warning("No players in lineup to remove")
 
         # --- SUMMARY ROW ---
         if not filled_lineup.empty: