Refactor Streamlit app to streamline data handling and improve display logic

- Updated data retrieval to drop unnecessary '_id' columns from DataFrames for various tables.
- Enhanced display logic by storing DataFrames in session state for better performance and user experience.
- Improved organization of data presentation across different tabs for pitchers, hitters, and teams.

src/streamlit_app.py

@@ -64,31 +64,31 @@ def init_baselines():
         df = pd.DataFrame(cursor)
 
         if table == 'Bullpen_Data':
-            bp_data = df
+            bp_data = df.drop(columns = ['_id'])
         elif table == 'Hitter_Agg_Merge':
-            hitter_agg = df
+            hitter_agg = df.drop(columns = ['_id'])
         elif table == 'Hitter_Long_Merge':
-            hitter_long = df
+            hitter_long = df.drop(columns = ['_id'])
         elif table == 'Hitter_Short_Merge':
-            hitter_short = df
+            hitter_short = df.drop(columns = ['_id'])
         elif table == 'Pitcher_Agg_Merge':
-            pitcher_agg = df
+            pitcher_agg = df.drop(columns = ['_id'])
         elif table == 'Pitcher_Long_Merge':
-            pitcher_long = df
+            pitcher_long = df.drop(columns = ['_id'])
         elif table == 'Pitcher_Short_Merge':
-            pitcher_short = df
+            pitcher_short = df.drop(columns = ['_id'])
         elif table == 'Slate_Hitters_Merge':
-            slate_hitters = df
+            slate_hitters = df.drop(columns = ['_id'])
         elif table == 'Slate_Team_Merge':
-            slate_team = df
+            slate_team = df.drop(columns = ['_id'])
         elif table == 'Starting_Pitchers':  
-            starting_pitchers = df
+            starting_pitchers = df.drop(columns = ['_id'])
         elif table == 'True_AVG_Split':
-            true_avg_split = df
+            true_avg_split = df.drop(columns = ['_id'])
         elif table == 'Pitcher_Info':
-            pitcher_info = df
+            pitcher_info = df.drop(columns = ['_id'])
         elif table == 'Hitter_Info':    
-            hitter_info = df
+            hitter_info = df.drop(columns = ['_id'])
 
     return bp_data, hitter_agg, hitter_long, hitter_short, pitcher_agg, pitcher_long, pitcher_short, slate_hitters, slate_team, starting_pitchers, true_avg_split, pitcher_info, hitter_info
 
@@ -108,19 +108,21 @@ with pitcher_tab:
             splits_var_sp = st.selectbox('Splits', ['Overall', 'RHH', 'LHH'], key = 'splits_var_sp')
     
     if table_var_sp == 'True AVG Splits':
-        st.dataframe(true_avg_split)
+        st.session_state['sp_disp_frame'] = true_avg_split
     elif table_var_sp == 'HWSr Splits':
-        st.dataframe(true_avg_split)
+        st.session_state['sp_disp_frame'] = true_avg_split
     elif table_var_sp == 'Current Slate Overview':
-        st.dataframe(starting_pitchers)
+        st.session_state['sp_disp_frame'] = starting_pitchers
     elif table_var_sp == 'Active Baselines':
-        st.dataframe(pitcher_info)
+        st.session_state['sp_disp_frame'] = pitcher_info
     elif table_var_sp == 'League Aggregate Baselines':
-        st.dataframe(pitcher_agg)
+        st.session_state['sp_disp_frame'] = pitcher_agg
     elif table_var_sp == 'League Short Term Baselines':
-        st.dataframe(pitcher_short)
+        st.session_state['sp_disp_frame'] = pitcher_short
     elif table_var_sp == 'League Long Term Baselines':
-        st.dataframe(pitcher_long)
+        st.session_state['sp_disp_frame'] = pitcher_long
+    
+    st.dataframe(st.session_state['sp_disp_frame'])
 
 with hitter_tab:
     with st.expander('Info and Display Options'):
@@ -134,15 +136,17 @@ with hitter_tab:
             splits_var_hitter = st.selectbox('Splits', ['Overall', 'RHP', 'LHP'], key = 'splits_var_hitter')
 
     if table_var_hitter == 'Current Slate Overview':
-        st.dataframe(starting_pitchers)
+        st.session_state['hitter_disp_frame'] = starting_pitchers
     elif table_var_hitter == 'Active Baselines':
-        st.dataframe(hitter_info)
+        st.session_state['hitter_disp_frame'] = hitter_info
     elif table_var_hitter == 'League Aggregate Baselines':
-        st.dataframe(hitter_agg)
+        st.session_state['hitter_disp_frame'] = hitter_agg
     elif table_var_hitter == 'League Short Term Baselines':
-        st.dataframe(hitter_short)
+        st.session_state['hitter_disp_frame'] = hitter_short
     elif table_var_hitter == 'League Long Term Baselines':
-        st.dataframe(hitter_long)
+        st.session_state['hitter_disp_frame'] = hitter_long
+
+    st.dataframe(st.session_state['hitter_disp_frame'])
 
 with team_tab:
     with st.expander('Info and Display Options'):
@@ -153,6 +157,8 @@ with team_tab:
             table_var_team = st.selectbox('Table', ['Team Baselines', 'Bullpen Baselines'], key = 'table_var_team')
     
     if table_var_team == 'Team Baselines':
-        st.dataframe(slate_team)
+        st.session_state['team_disp_frame'] = slate_team
     elif table_var_team == 'Bullpen Baselines':
-        st.dataframe(bp_data)
+        st.session_state['team_disp_frame'] = bp_data
+
+    st.dataframe(st.session_state['team_disp_frame'])