Enhance hitter tab functionality in Streamlit app with position filtering

- Added a new selection option for player positions in the hitter tab, allowing users to filter data based on specific positions.
- Updated the layout to accommodate an additional column for position selection, improving user interactivity.
- Implemented position-based filtering logic for both DraftKings and FanDuel sites, enhancing data accuracy based on user selections.
- Introduced a loading spinner for league baseline data to improve user experience during data retrieval.

src/streamlit_app.py

@@ -3,6 +3,7 @@ import numpy as np
 import pandas as pd
 import pymongo
 import os
+import time
 
 st.set_page_config(layout="wide")
 
@@ -172,7 +173,7 @@ with pitcher_tab:
 with hitter_tab:
     with st.container(border = True):
         st.info('Note: Splits options are available for all baseline tables')
-        col1, col2, col3, col4, col5 = st.columns(5)
+        col1, col2, col3, col4, col5, col6 = st.columns(6)
         with col1:
             site_var_hitter = st.selectbox('Site', ['DraftKings', 'FanDuel'], key = 'site_var_hitter')
         with col2:
@@ -180,8 +181,10 @@ with hitter_tab:
         with col3:
             splits_var_hitter = st.selectbox('Splits', ['Overall', 'RHP', 'LHP'], key = 'splits_var_hitter')
         with col4:
-            team_type_hitter = st.selectbox('Do you want to view all teams or Specific ones?', ['All', 'Specific'], key = 'team_type_hitter')
+            position_var_hitter = st.selectbox('Positions', ['C', '1B', '2B', '3B', 'SS', 'OF'], key = 'position_var_hitter')
         with col5:
+            team_type_hitter = st.selectbox('Do you want to view all teams or Specific ones?', ['All', 'Specific'], key = 'team_type_hitter')
+        with col6:
             if team_type_hitter == 'Specific':
                 team_var_hitter = st.multiselect('Select Teams', slate_hitters['Team'].unique(), key = 'team_var_hitter')
             else:
@@ -193,12 +196,27 @@ with hitter_tab:
         
         if team_var_hitter is not None:
             disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+        if position_var_hitter is not None:
+            if site_var_hitter == 'Draftkings':
+                position_mask = disp_raw['Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
+                disp_raw = disp_raw[position_mask]
+            elif site_var_hitter == 'Fanduel':
+                position_mask = disp_raw['FD_Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
+                disp_raw = disp_raw[position_mask]
         st.session_state['hitter_disp_frame'] = disp_raw
     elif table_var_hitter == 'Active Baselines':
         disp_raw = hitter_info
         
         if team_var_hitter is not None:
             disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
+        
+        if position_var_hitter is not None:
+            if site_var_hitter == 'Draftkings':
+                position_mask = disp_raw['Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
+                disp_raw = disp_raw[position_mask]
+            elif site_var_hitter == 'Fanduel':
+                position_mask = disp_raw['FD_Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
+                disp_raw = disp_raw[position_mask]
 
         st.session_state['hitter_disp_frame'] = disp_raw
     elif table_var_hitter == 'League Aggregate Baselines':
@@ -227,6 +245,9 @@ with hitter_tab:
     hitter_disp_container = hitter_disp_container.empty()
 
     with hitter_disp_container:
+        if table_var_hitter.isin(['League Aggregate Baselines', 'League Short Term Baselines', 'League Long Term Baselines']):
+            with st.spinner("Full league baselines can take some time to load", show_time=True):
+                time.sleep(5)
         st.dataframe(st.session_state['hitter_disp_frame'].style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), height = 750, use_container_width = True, hide_index = True)
 
 with team_tab: