Update app.py

app.py

@@ -42,7 +42,9 @@ master_hold = 'https://docs.google.com/spreadsheets/d/1D526UlXmrz-8qxVcUKrA-u7f6
 team_format = {'2B': '{:.2%}', 'HR': '{:.2%}', 'SB': '{:.2%}', 'P_SO': '{:.2%}', 'P_H': '{:.2%}', 'P_R': '{:.2%}',
                'P_HR': '{:.2%}', 'P_BB': '{:.2%}'}
 
-@st.cache_resource(ttl = 599)
+sim_format = {'Top_finish': '{:.2%}', 'Top_5_finish': '{:.2%}', 'Top_10_finish': '{:.2%}'}
+
+@st.cache_resource(ttl = 600)
 def init_baselines():
     sh = gcservice_account.open_by_url(master_hold)
     worksheet = sh.worksheet('Pitcher_Proj')
@@ -65,17 +67,19 @@ def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
 pitcher_proj, hitter_proj, wins_proj = init_baselines()
+total_teams = pitcher_proj['Team'].values.tolist()
 
-tab1, tab2, tab3, tab4, tab5 = st.tabs(["Team Win Projections", "Pitcher Projections", "Hitter Projections", "Pitcher Simulations", "Hitter Simulations"])
+tab1, tab2, tab3, tab4, tab5 = st.tabs(["Pitcher Projections", "Hitter Projections", "Pitcher Simulations", "Hitter Simulations"])
 
 with tab1:
     if st.button("Reset Data", key='reset1'):
               st.cache_data.clear()
               pitcher_proj, hitter_proj, wins_proj = init_baselines()
+              total_teams = pitcher_proj['Team'].values.tolist()
     raw_frame = wins_proj.copy()
     export_frame_team = raw_frame[['Team', '2B', 'HR', 'SB', 'P_SO', 'P_H', 'P_R', 'P_HR', 'P_BB', 'Upside', 'Projected']]
     disp_frame = raw_frame[['Team', '2B', 'HR', 'SB', 'P_SO', 'P_H', 'P_R', 'P_HR', 'P_BB', 'Upside', 'Projected']]
-    st.dataframe(disp_frame.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(team_format, precision=2), use_container_width = True, height = 500)
+    st.dataframe(disp_frame.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(team_format, precision=2), use_container_width = True)
     
     st.download_button(
         label="Export Team Win Projections",
@@ -89,12 +93,20 @@ with tab2:
     if st.button("Reset Data", key='reset2'):
               st.cache_data.clear()
               pitcher_proj, hitter_proj, wins_proj = init_baselines()
+              total_teams = pitcher_proj['Team'].values.tolist()
     raw_frame = pitcher_proj.copy()
+    split_var1 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var1')
+    if split_var1 == 'Specific Teams':
+        team_var1 = st.multiselect('Which teams would you like to include in the tables?', options = total_teams, key='team_var1')
+    elif split_var1 == 'All':
+        team_var1 = total_teams
+    
+    working_data = raw_frame[raw_frame['Team'].isin(team_var1)]
     export_frame_sp = raw_frame[['Name', 'Team', 'TBF', 'Ceiling_var', 'True_AVG', 'Hits', 'Singles%', 'Singles', 'Doubles%', 'Doubles', 'xHR%', 'Homeruns', 'Strikeout%', 'Strikeouts',
                               'Walk%', 'Walks', 'Runs%', 'Runs', 'ERA', 'Wins', 'Quality_starts', 'ADP', 'UD_fpts']]
-    disp_frame = raw_frame[['Name', 'Team', 'TBF', 'True_AVG', 'Hits', 'Singles', 'Doubles', 'Homeruns', 'Strikeouts',
+    disp_frame_sp = working_data[['Name', 'Team', 'TBF', 'True_AVG', 'Hits', 'Singles', 'Doubles', 'Homeruns', 'Strikeouts',
                             'Walks', 'Runs', 'ERA', 'Wins', 'Quality_starts', 'ADP', 'UD_fpts']]
-    st.dataframe(disp_frame.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True, height = 500)
+    st.dataframe(disp_frame_sp.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
     
     st.download_button(
         label="Export Pitcher Projections",
@@ -108,12 +120,20 @@ with tab3:
     if st.button("Reset Data", key='reset3'):
               st.cache_data.clear()
               pitcher_proj, hitter_proj, wins_proj = init_baselines()
+              total_teams = pitcher_proj['Team'].values.tolist()
     raw_frame = hitter_proj.copy()
+    split_var2 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var2')
+    if split_var2 == 'Specific Teams':
+        team_var2 = st.multiselect('Which teams would you like to include in the tables?', options = total_teams, key='team_var2')
+    elif split_var2 == 'All':
+        team_var2 = total_teams
+    
+    working_data = raw_frame[raw_frame['Team'].isin(team_var2)]
     export_frame_h = raw_frame[['Name', 'Team', 'PA', 'Ceiling_var', 'Walk%', 'Walks', 'xHits', 'Singles%', 'Singles', 'Doubles%', 'Doubles',
                               'xHR%', 'Homeruns', 'Runs%', 'Runs', 'RBI%', 'RBI', 'Steal%', 'Stolen_bases', 'UD_fpts', 'ADP']]
-    disp_frame = raw_frame[['Name', 'Team', 'PA', 'Walks', 'xHits', 'Singles', 'Doubles',
+    disp_frame_h = working_data[['Name', 'Team', 'PA', 'Walks', 'xHits', 'Singles', 'Doubles',
                             'Homeruns', 'Runs', 'RBI', 'Stolen_bases', 'UD_fpts', 'ADP']]
-    st.dataframe(disp_frame.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True, height = 500)
+    st.dataframe(disp_frame_h.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
     
     st.download_button(
         label="Export Hitter Projections",
@@ -127,6 +147,7 @@ with tab4:
     if st.button("Reset Data", key='reset4'):
               st.cache_data.clear()
               pitcher_proj, hitter_proj, wins_proj = init_baselines()
+              total_teams = pitcher_proj['Team'].values.tolist()
     col1, col2 = st.columns([1, 5])
     
     with col2:
@@ -206,12 +227,13 @@ with tab4:
                         
                     with df_hold_container.container():     
                         df_hold_container = st.empty()
-                        st.dataframe(final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+                        st.dataframe(final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(sim_format, precision=2), use_container_width = True)
     
 with tab5:
     if st.button("Reset Data", key='reset5'):
               st.cache_data.clear()
               pitcher_proj, hitter_proj, wins_proj = init_baselines()
+              total_teams = pitcher_proj['Team'].values.tolist()
     col1, col2 = st.columns([1, 5])
     
     with col2:
@@ -298,4 +320,4 @@ with tab5:
 
                     with df_hold_container.container():     
                         df_hold_container = st.empty()
-                        st.dataframe(final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+                        st.dataframe(final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(sim_format, precision=2), use_container_width = True)