Update app.py

app.py

@@ -39,46 +39,38 @@ gcservice_account = init_conn()
 freq_format = {'Proj Own': '{:.2%}', 'Exposure': '{:.2%}', 'Edge': '{:.2%}'}
 
 @st.cache_resource(ttl = 360)
-def load_dk_player_projections():
+def init_baselines():
     sh = gcservice_account.open_by_url('https://docs.google.com/spreadsheets/d/1I_1Ve3F4tftgfLQQoRKOJ351XfEG48s36OxXUKxmgS8/edit#gid=1391856348')
     worksheet = sh.worksheet('DK_ROO')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
-    raw_display = load_display.dropna(subset=['Median'])
-
-    return raw_display
+    
+    dk_raw = load_display.dropna(subset=['Median'])
 
-@st.cache_resource(ttl = 300)
-def load_fd_player_projections():
-    sh = gcservice_account.open_by_url('https://docs.google.com/spreadsheets/d/1I_1Ve3F4tftgfLQQoRKOJ351XfEG48s36OxXUKxmgS8/edit#gid=1391856348')
     worksheet = sh.worksheet('FD_ROO')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
-    raw_display = load_display.dropna(subset=['Median'])
-
-    return raw_display
+    
+    fd_raw = load_display.dropna(subset=['Median'])
 
-@st.cache_resource(ttl = 300)
-def set_export_ids():
-    sh = gcservice_account.open_by_url('https://docs.google.com/spreadsheets/d/1I_1Ve3F4tftgfLQQoRKOJ351XfEG48s36OxXUKxmgS8/edit#gid=1391856348')
     worksheet = sh.worksheet('DK_ROO')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
     raw_display = load_display.dropna(subset=['Median'])
+    
     dk_ids = dict(zip(raw_display['Player'], raw_display['player_id']))
     
     worksheet = sh.worksheet('FD_ROO')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
     raw_display = load_display.dropna(subset=['Median'])
+    
     fd_ids = dict(zip(raw_display['Player'], raw_display['player_id']))
 
-    return dk_ids, fd_ids
+    return dk_raw, fd_raw, dk_ids, fd_ids
 
-dk_roo_raw = load_dk_player_projections()
-fd_roo_raw = load_fd_player_projections()
+dk_roo_raw, fd_roo_raw, dkid_dict, fdid_dict = init_baselines()
 t_stamp = f"Last Update: " + str(dk_roo_raw['timestamp'][0]) + f" CST"
-dkid_dict, fdid_dict = set_export_ids()
 
 static_exposure = pd.DataFrame(columns=['Player', 'count'])
 overall_exposure = pd.DataFrame(columns=['Player', 'count'])
@@ -677,10 +669,8 @@ with tab2:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              dk_roo_raw = load_dk_player_projections()
-              fd_roo_raw = load_fd_player_projections()
+              dk_roo_raw, fd_roo_raw, dkid_dict, fdid_dict = init_baselines()
               t_stamp = f"Last Update: " + str(dk_roo_raw['timestamp'][0]) + f" CST"
-              dkid_dict, fdid_dict = set_export_ids()
               
         slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate', 'Thurs-Mon Slate', 'User'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))