Enhance team filtering functionality in Streamlit app for pitcher and hitter tabs

- Added filtering options for team selection in both pitcher and hitter tabs, allowing users to view data for specific teams.
- Updated the display logic to ensure that selected teams are reflected in the dataframes for 'True AVG Splits', 'HWSr Splits', and various baseline tables.
- Improved user experience by integrating team selection into the current slate overview and active baselines displays.

src/streamlit_app.py

@@ -117,20 +117,33 @@ with pitcher_tab:
     
     if table_var_sp == 'True AVG Splits':
         disp_raw = true_avg_split
+
+        if team_var_sp is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_sp)]
+
         disp_raw = disp_raw[['Player', 'Handedness', 'Team', 'Opp', 'Opp LHH', 'Opp RHH', 'True AVG (LHH)', 'True AVG (RHH)', 'True AVG (Overall)', 'Weighted True AVG']]
         st.session_state['sp_disp_frame'] = disp_raw
     elif table_var_sp == 'HWSr Splits':
         disp_raw = true_avg_split
+
+        if team_var_sp is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_sp)]
+
         disp_raw = disp_raw[['Player', 'Handedness', 'Team', 'Opp', 'Opp LHH', 'Opp RHH', 'HWSr (LHH)', 'HWSr (RHH)', 'HWSr (Overall)', 'Weighted HWSr']]
         st.session_state['sp_disp_frame'] = disp_raw
     elif table_var_sp == 'Current Slate Overview':
         st.session_state['sp_disp_frame'] = starting_pitchers
     elif table_var_sp == 'Active Baselines':
         disp_raw = pitcher_info
+
         if splits_var_sp != 'Overall':
             disp_raw = disp_raw[disp_raw['Set'] == splits_var_sp]
         else:
             disp_raw = disp_raw[disp_raw['Set'] == 'RHH']
+        
+        if team_var_sp is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_sp)]
+
         disp_raw = disp_raw[['Names', 'DK_Salary', 'FD_Salary', 'Team', 'Opp', 'Opp_TT', 'Hand', 'K%', 'BB%', 'True AVG', 'xSLG', 'xBA', 'Hits', 'xHRs', 'xHR/PA']]
         positive_set = ['K%']
         st.session_state['sp_disp_frame'] = disp_raw
@@ -163,7 +176,7 @@ with hitter_tab:
         with col1:
             site_var_hitter = st.selectbox('Site', ['DraftKings', 'FanDuel'], key = 'site_var_hitter')
         with col2:
-            table_var_hitter = st.selectbox('Table', ['Active Baselines', 'League Aggregate Baselines', 'League Short Term Baselines', 'League Long Term Baselines'], key = 'table_var_hitter')
+            table_var_hitter = st.selectbox('Table', ['Current Slate Overview', 'Active Baselines', 'League Aggregate Baselines', 'League Short Term Baselines', 'League Long Term Baselines'], key = 'table_var_hitter')
         with col3:
             splits_var_hitter = st.selectbox('Splits', ['Overall', 'RHP', 'LHP'], key = 'splits_var_hitter')
         with col4:
@@ -176,15 +189,39 @@ with hitter_tab:
                 st.write('All teams selected')
 
     if table_var_hitter == 'Current Slate Overview':
-        st.session_state['hitter_disp_frame'] = slate_hitters
+        disp_raw = slate_hitters
+        
+        if team_var_hitter is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+        st.session_state['hitter_disp_frame'] = disp_raw
     elif table_var_hitter == 'Active Baselines':
-        st.session_state['hitter_disp_frame'] = hitter_info
+        disp_raw = hitter_info
+        
+        if team_var_hitter is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+
+        st.session_state['hitter_disp_frame'] = disp_raw
     elif table_var_hitter == 'League Aggregate Baselines':
-        st.session_state['hitter_disp_frame'] = hitter_agg
+        disp_raw = hitter_agg
+        
+        if team_var_hitter is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+
+        st.session_state['hitter_disp_frame'] = disp_raw
     elif table_var_hitter == 'League Short Term Baselines':
-        st.session_state['hitter_disp_frame'] = hitter_short
+        disp_raw = hitter_short
+        
+        if team_var_hitter is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+
+        st.session_state['hitter_disp_frame'] = disp_raw
     elif table_var_hitter == 'League Long Term Baselines':
-        st.session_state['hitter_disp_frame'] = hitter_long
+        disp_raw = hitter_long
+        
+        if team_var_hitter is not None:
+            disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+
+        st.session_state['hitter_disp_frame'] = disp_raw
 
     hitter_disp_container = st.container(border = True)
     hitter_disp_container = hitter_disp_container.empty()