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 = 300)
-def load_dk_player_projections():
+def init_baslines():
     sh = gcservice_account.open_by_url('https://docs.google.com/spreadsheets/d/1Yq0vGriWK-bS79e-bD6_u9pqrYE6Yrlbb_wEkmH-ot0/edit#gid=172632260')
     worksheet = sh.worksheet('DK_Build_Up')
     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
+    load_display.rename(columns={"Fantasy": "Median"}, inplace = True)
+    dk_roo_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/1Yq0vGriWK-bS79e-bD6_u9pqrYE6Yrlbb_wEkmH-ot0/edit#gid=172632260')
     worksheet = sh.worksheet('FD_Build_Up')
     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
+    load_display.rename(columns={"Fantasy": "Median"}, inplace = True)
+    fd_roo_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/1Yq0vGriWK-bS79e-bD6_u9pqrYE6Yrlbb_wEkmH-ot0/edit#gid=172632260')
     worksheet = sh.worksheet('DK_Salaries')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
     raw_display = load_display.dropna(subset=['Median'])
+    raw_display.rename(columns={"name": "Player"}, inplace = True)
     dk_ids = dict(zip(raw_display['Player'], raw_display['player_id']))
     
     worksheet = sh.worksheet('FD_Salaries')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
     raw_display = load_display.dropna(subset=['Median'])
+    raw_display.rename(columns={"name": "Player"}, inplace = True)
     fd_ids = dict(zip(raw_display['Player'], raw_display['player_id']))
 
-    return dk_ids, fd_ids
+    return dk_roo_raw, fd_roo_raw, dk_ids, fd_ids
 
-dk_roo_raw = load_dk_player_projections()
-fd_roo_raw = load_fd_player_projections()
-t_stamp = f"Last Update: " + str(dk_roo_raw['timestamp'][0]) + f" CST"
-dkid_dict, fdid_dict = set_export_ids()
+dk_roo_raw, fd_roo_raw, dkid_dict, fdid_dict = init_baslines()
+t_stamp = f"Last Update: " + "idk man recently probably" + f" CST"
 
 static_exposure = pd.DataFrame(columns=['Player', 'count'])
 overall_exposure = pd.DataFrame(columns=['Player', 'count'])
@@ -628,10 +620,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()
-              t_stamp = f"Last Update: " + str(dk_roo_raw['timestamp'][0]) + f" CST"
-              dkid_dict, fdid_dict = set_export_ids()
+              dk_roo_raw, fd_roo_raw, dkid_dict, fdid_dict = init_baslines()
+              t_stamp = f"Last Update: " + "idk man recently probably" + f" CST"
               
         slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'User'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))