Update app.py

app.py

@@ -63,9 +63,14 @@ def init_baselines():
           raw_display.replace('', np.nan, inplace=True)
           prop_frame = raw_display.dropna(subset='Team')
           
-          return pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp
+          worksheet = sh.worksheet('Pick6_ingest')
+          raw_display = pd.DataFrame(worksheet.get_all_records())
+          raw_display.replace('', np.nan, inplace=True)
+          pick_frame = raw_display.dropna(subset='Player')
+          
+          return pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp
 
-pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp = init_baselines()
+pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp = init_baselines()
 
 tab1, tab2, tab3, tab4, tab5 = st.tabs(["Game Betting Model", "Pitcher Prop Projections", "Hitter Prop Projections", "Player Prop Simulations", "Stat Specific Simulations"])
 
@@ -76,7 +81,7 @@ with tab1:
     st.info(t_stamp)
     if st.button("Reset Data", key='reset1'):
               st.cache_data.clear()
-              pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp = init_baselines()
+              pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp = init_baselines()
     line_var1 = st.radio('How would you like to display odds?', options = ['Percentage', 'American'], key='line_var1')
     if line_var1 == 'Percentage':
         team_frame = team_frame[['Names', 'Game', 'Win Percentage', 'Spread', 'Cover Spread Percentage', 'Avg Score', 'Game Total', 'Avg Fifth Inning', 'Fifth Inning Lead Percentage']]
@@ -100,7 +105,7 @@ with tab2:
     st.info(t_stamp)
     if st.button("Reset Data", key='reset2'):
               st.cache_data.clear()
-              pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp = init_baselines()
+              pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp = init_baselines()
     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 = pitcher_stats['Team'].unique(), key='team_var1')
@@ -122,7 +127,7 @@ with tab3:
     st.info(t_stamp)
     if st.button("Reset Data", key='reset3'):
               st.cache_data.clear()
-              pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp = init_baselines()
+              pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp = init_baselines()
     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 = hitter_stats['Team'].unique(), key='team_var2')
@@ -144,7 +149,7 @@ with tab4:
     st.info(t_stamp)
     if st.button("Reset Data", key='reset4'):
               st.cache_data.clear()
-              pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp = init_baselines()
+              pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp = init_baselines()
     col1, col2 = st.columns([1, 5])
     
     with col2:
@@ -311,7 +316,7 @@ with tab5:
               # Clear values from *all* all in-memory and on-disk data caches:
               # i.e. clear values from both square and cube
               st.cache_data.clear()
-              pitcher_stats, hitter_stats, team_frame, prop_frame, t_stamp = init_baselines()
+              pitcher_stats, hitter_stats, team_frame, prop_frame, pick_frame, t_stamp = init_baselines()
     col1, col2 = st.columns([1, 5])
     
     with col2:
@@ -321,6 +326,18 @@ with tab5:
         export_container = st.empty()
     
     with col1:
+        game_select_var = st.selectbox('Select prop source', options = ['Draftkings', 'Pick6'])
+        if game_select_var == 'Draftkings':
+            prop_df = prop_frame[['Player', 'over_prop', 'over_line', 'under_line', 'prop_type']]
+        elif game_select_var == 'Pick6':
+            prop_df = pick_frame[['Player', 'over_prop', 'over_line', 'under_line', 'prop_type']]
+        st.download_button(
+            label="Download Prop Source",
+            data=convert_df_to_csv(prop_df),
+            file_name='MLB_prop_source.csv',
+            mime='text/csv',
+            key='prop_source',
+        )
         prop_type_var = st.selectbox('Select prop category', options = ['Strikeouts (Pitchers)', 'Total Outs (Pitchers)'])
 
         if st.button('Simulate Prop Category'):