reverting handbuilder adjustments

src/streamlit_app.py

@@ -616,43 +616,43 @@ 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
-        player_select_event = st.dataframe(
+        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",  # Single row selection
+            selection_mode=["single-row"],
             key="player_select_dataframe",
             height=500,
             hide_index=True
         )
-        
-        # Handle player selection - add to lineup
-        if player_select_event and "rows" in player_select_event.selection:
-            selected_idx = player_select_event.selection["rows"][0]  # Single selection
-            selected_player_row = st.session_state['player_select_df'].iloc[selected_idx]
-            
-            # Check if player is already in lineup
-            if selected_player_row['Player'] not in st.session_state['handbuilder_lineup']['Player'].values:
-                eligible_positions = re.split(r'[/, ]+', selected_player_row['Position'])
-                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:
+        # 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:
                             slot_to_fill = slot
                             break
-
-                if slot_to_fill is not None:
-                    player_row = selected_player_row.to_frame().T.assign(Slot=slot_to_fill)
+                    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()
+
 
     with handbuilder_lineup_build_column:
         st.subheader("Lineup Build")
@@ -697,36 +697,27 @@ if selected_tab == 'Handbuilder':
 
         st.session_state['lineup_display_df'] = pd.DataFrame(display_rows, columns=display_columns)
 
-        # Display the lineup table as an interactive dataframe
-        lineup_select_event = st.dataframe(
+        # Show the lineup table with single-row selection for removal
+        event_remove = 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",  # Single row selection
-            key="lineup_display_dataframe",
-            height=445,
-            hide_index=True
-        )
-        
-        # Handle lineup selection - remove from lineup
-        if lineup_select_event and "rows" in lineup_select_event.selection:
-            selected_idx = lineup_select_event.selection["rows"][0]  # Single selection
-            selected_slot_row = st.session_state['lineup_display_df'].iloc[selected_idx]
-            
-            # Only remove if there's actually a player in that slot
-            if selected_slot_row['Player'] and selected_slot_row['Player'] != '':
-                player_to_remove = selected_slot_row['Player']
-                if player_to_remove in st.session_state['handbuilder_lineup']['Player'].values:
-                    st.session_state['handbuilder_lineup'] = st.session_state['handbuilder_lineup'][
-                        st.session_state['handbuilder_lineup']['Player'] != player_to_remove
-                    ]
-
-        # Show the lineup table as a static display (no selection needed)
-        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),
+            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()
+
         # --- SUMMARY ROW ---
         if not filled_lineup.empty:
             total_salary = filled_lineup['Salary'].sum()