Enhance player data management in app.py by adding 'Team_Total' and 'Opp_Total' columns to the lineup. Updated DataFrame structures and quick fill options to incorporate these new metrics, improving the clarity and depth of player selection information.

app.py

@@ -89,6 +89,8 @@ def init_baselines():
     hold_frame['Order'] = np.where(hold_frame['pos_group'] == 'Hitters', hold_frame['Player'].map(RHP_Info.set_index('Player')['Order']), 0)
     hold_frame['Hand'] = np.where(hold_frame['pos_group'] == 'Hitters', hold_frame['Player'].map(RHP_Info.set_index('Player')['bats']), hold_frame['Player'].map(RHH_Info.set_index('Player')['Hand']))
     hold_frame['Opp'] = hold_frame['Team'].map(RHH_Info.set_index('Team')['Opp'])
+    hold_frame['Team_Total'] = hold_frame['Team'].map(RHP_Info.set_index('Team')['Opp_TT'])
+    hold_frame['Opp_Total'] = hold_frame['Team'].map(RHH_Info.set_index('Team')['Opp_TT'])
     
     roo_data.insert(3, 'Opp', hold_frame['Opp'])
     roo_data.insert(4, 'Hand', hold_frame['Hand'])
@@ -96,6 +98,8 @@ def init_baselines():
         roo_data.insert(5, 'Order', hold_frame['Order'].astype(int))
     except:
         roo_data.insert(5, 'Order', hold_frame['Order'])
+    roo_data.insert(6, 'Team_Total', hold_frame['Team_Total'])
+    roo_data.insert(7, 'Opp_Total', hold_frame['Opp_Total'])
     
     dk_roo = roo_data[roo_data['Site'] == 'Draftkings']
     dk_id_map = dict(zip(dk_roo['Player'], dk_roo['player_ID']))
@@ -832,7 +836,7 @@ with tab4:
 
     # --- LINEUP STATE ---
     if 'handbuilder_lineup' not in st.session_state:
-        st.session_state['handbuilder_lineup'] = pd.DataFrame(columns=['Player', 'Order', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Own%'])
+        st.session_state['handbuilder_lineup'] = pd.DataFrame(columns=['Player', 'Order', 'Position', 'Team', 'Team_Total', 'Opp_Total', 'Salary', 'Median', '2x%', 'Own%'])
     if 'handbuilder_select_key' not in st.session_state:
         st.session_state['handbuilder_select_key'] = 0
 
@@ -859,14 +863,14 @@ with tab4:
     if selected_teams:
         player_select_df = handbuild_roo[
             handbuild_roo['Team'].isin(selected_teams)
-        ][['Player', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Order', 'Hand', 'Own%']].drop_duplicates(subset=['Player', 'Team']).sort_values(by='Order', ascending=True).copy()
+        ][['Player', 'Position', 'Team', 'Team_Total', 'Opp_Total', 'Salary', 'Median', '2x%', 'Order', 'Hand', 'Own%']].drop_duplicates(subset=['Player', 'Team']).sort_values(by='Order', ascending=True).copy()
     else:
-        player_select_df = handbuild_roo[['Player', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Order', 'Hand', 'Own%']].drop_duplicates(subset=['Player', 'Team']).copy()
+        player_select_df = handbuild_roo[['Player', 'Position', 'Team', 'Team_Total', 'Opp_Total', 'Salary', 'Median', '2x%', 'Order', 'Hand', 'Own%']].drop_duplicates(subset=['Player', 'Team']).copy()
     
     with st.expander("Quick Fill Options"):
         auto_team_var = st.selectbox("Auto Fill Team", options=all_teams)
         auto_size_var = st.selectbox("Auto Fill Size", options=[3, 4, 5])
-        auto_range_var = st.selectbox("Auto Fill Order", options=['Top (1-5)', 'Mid (3-7)', 'Wrap (8-3)'])
+        auto_range_var = st.selectbox("Auto Fill Order", options=['Top (1-5)', 'Mid (4-8)', 'Wrap (7-2)'])
         # --- QUICK FILL LOGIC ---
         if st.button("Quick Fill", key="quick_fill"):
             # 1. Get all eligible players from the selected team, not already in the lineup
@@ -907,7 +911,7 @@ with tab4:
                         add_row['Slot'] = slot_to_fill
                         st.session_state['handbuilder_lineup'] = pd.concat(
                             [st.session_state['handbuilder_lineup'], pd.DataFrame([add_row[[
-                                'Player', 'Order', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Own%', 'Slot'
+                                'Player', 'Order', 'Position', 'Team', 'Team_Total', 'Opp_Total', 'Salary', 'Median', '2x%', 'Own%', 'Slot'
                             ]]])],
                             ignore_index=True
                         )
@@ -954,7 +958,7 @@ with tab4:
                     # 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', 'Order', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Own%', 'Slot']]],
+                        [st.session_state['handbuilder_lineup'], player_row[['Player', 'Order', 'Position', 'Team', 'Team_Total', 'Opp_Total', 'Salary', 'Median', '2x%', 'Own%', 'Slot']]],
                         ignore_index=True
                     )
                 st.session_state['handbuilder_select_key'] += 1
@@ -1087,5 +1091,5 @@ with tab4:
 
     # Optionally, add a button to clear the lineup
     if st.button("Clear Lineup", key='clear_lineup'):
-        st.session_state['handbuilder_lineup'] = pd.DataFrame(columns=['Player', 'Position', 'Team', 'Salary', 'Median', '2x%', 'Own%', 'Slot', 'Order'])
+        st.session_state['handbuilder_lineup'] = pd.DataFrame(columns=['Player', 'Position', 'Team', 'Team_Total', 'Opp_Total', 'Salary', 'Median', '2x%', 'Own%', 'Slot', 'Order'])
         st.rerun()