Update app.py

app.py

@@ -39,8 +39,10 @@ gcservice_account = init_conn()
 
 master_hold = 'https://docs.google.com/spreadsheets/d/1D526UlXmrz-8qxVcUKrA-u7f6FftUiBufxDnzQv980k/edit#gid=791804525'
 
+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)
+@st.cache_resource(ttl = 600)
 def init_baselines():
     sh = gcservice_account.open_by_url(master_hold)
     worksheet = sh.worksheet('Pitcher_Proj')
@@ -52,19 +54,41 @@ def init_baselines():
     raw_display = pd.DataFrame(worksheet.get_all_records())
     hitter_proj = raw_display.dropna()
     
-    return pitcher_proj, hitter_proj
+    sh = gcservice_account.open_by_url(master_hold)
+    worksheet = sh.worksheet('Wins_Proj')
+    raw_display = pd.DataFrame(worksheet.get_all_records())
+    wins_proj = raw_display.dropna()
+    
+    return pitcher_proj, hitter_proj, wins_proj
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-pitcher_proj, hitter_proj = init_baselines()
+pitcher_proj, hitter_proj, wins_proj = init_baselines()
 
-tab1, tab2, tab3, tab4 = st.tabs(["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 = init_baselines()
+              pitcher_proj, hitter_proj, wins_proj = init_baselines()
+    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)
+    
+    st.download_button(
+        label="Export Team Win Projections",
+        data=convert_df_to_csv(export_frame_team),
+        file_name='MLB_team_win_export.csv',
+        mime='text/csv',
+        key='team_win_export',
+    )
+
+with tab2:
+    if st.button("Reset Data", key='reset2'):
+              st.cache_data.clear()
+              pitcher_proj, hitter_proj, wins_proj = init_baselines()
     raw_frame = pitcher_proj.copy()
     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']]
@@ -80,14 +104,14 @@ with tab1:
         key='pitcher_proj_export',
     )
 
-with tab2:
-    if st.button("Reset Data", key='reset2'):
+with tab3:
+    if st.button("Reset Data", key='reset3'):
               st.cache_data.clear()
-              pitcher_proj, hitter_proj = init_baselines()
+              pitcher_proj, hitter_proj, wins_proj = init_baselines()
     raw_frame = hitter_proj.copy()
-    export_frame_h = raw_frame[['Name', 'Position', 'Team', 'PA', 'Ceiling_var', 'Walk%', 'Walks', 'xHits', 'Singles%', 'Singles', 'Doubles%', 'Doubles',
+    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', 'Position', 'Team', 'PA', 'Walks', 'xHits', 'Singles', 'Doubles',
+    disp_frame = raw_frame[['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)
     
@@ -99,10 +123,10 @@ with tab2:
         key='hitter_proj_export',
     )
     
-with tab3:
-    if st.button("Reset Data", key='reset3'):
+with tab4:
+    if st.button("Reset Data", key='reset4'):
               st.cache_data.clear()
-              pitcher_proj, hitter_proj = init_baselines()
+              pitcher_proj, hitter_proj, wins_proj = init_baselines()
     col1, col2 = st.columns([1, 5])
     
     with col2:
@@ -184,10 +208,10 @@ with tab3:
                         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)
     
-with tab4:
-    if st.button("Reset Data", key='reset4'):
+with tab5:
+    if st.button("Reset Data", key='reset5'):
               st.cache_data.clear()
-              pitcher_proj, hitter_proj = init_baselines()
+              pitcher_proj, hitter_proj, wins_proj = init_baselines()
     col1, col2 = st.columns([1, 5])
     
     with col2:
@@ -195,6 +219,7 @@ with tab4:
     
     with col1:
         prop_type_var_h = st.selectbox('Select type of prop to simulate', options = ['Hits', 'Doubles', 'Home Runs', 'RBI', 'Stolen Bases'], key='prop_type_var_h')
+        
 
         if st.button('Simulate Stat', key='sim_h'):
             with col2: