Refactor init_baselines to return secondary slate data for Draftkings and Fanduel, updating function calls accordingly.

app.py

@@ -184,22 +184,22 @@ def init_baselines():
     load_display = load_display[load_display['Median'] > 0]
 
     dk_roo_raw = load_display[load_display['Site'] == 'Draftkings']
-    dk_roo_raw = dk_roo_raw[dk_roo_raw['Slate'] == 'Main Slate']
+    dk_roo_raw = dk_roo_raw[dk_roo_raw['Slate'] == 'main_slate']
     dk_roo_raw['STDev'] = dk_roo_raw['Median'] / 3
     dk_raw = dk_roo_raw.dropna(subset=['Median'])
 
     fd_roo_raw = load_display[load_display['Site'] == 'Fanduel']
-    fd_roo_raw = fd_roo_raw[fd_roo_raw['Slate'] == 'Main Slate']
+    fd_roo_raw = fd_roo_raw[fd_roo_raw['Slate'] == 'main_slate']
     fd_roo_raw['STDev'] = fd_roo_raw['Median'] / 3
     fd_raw = fd_roo_raw.dropna(subset=['Median'])
 
     dk_secondary_roo_raw = load_display[load_display['Site'] == 'Draftkings']
-    dk_secondary_roo_raw = dk_secondary_roo_raw[dk_secondary_roo_raw['Slate'] == 'Secondary Slate']
+    dk_secondary_roo_raw = dk_secondary_roo_raw[dk_secondary_roo_raw['Slate'] == 'secondary_slate']
     dk_secondary_roo_raw['STDev'] = dk_secondary_roo_raw['Median'] / 3
     dk_secondary = dk_secondary_roo_raw.dropna(subset=['Median'])
 
     fd_secondary_roo_raw = load_display[load_display['Site'] == 'Fanduel']
-    fd_secondary_roo_raw = fd_secondary_roo_raw[fd_secondary_roo_raw['Slate'] == 'Secondary Slate']
+    fd_secondary_roo_raw = fd_secondary_roo_raw[fd_secondary_roo_raw['Slate'] == 'secondary_slate']
     fd_secondary_roo_raw['STDev'] = fd_secondary_roo_raw['Median'] / 3
     fd_secondary = fd_secondary_roo_raw.dropna(subset=['Median'])
 
@@ -286,7 +286,7 @@ def sim_contest(Sim_size, seed_frame, maps_dict, Contest_Size, teams_playing_cou
         
     return Sim_Winners
 
-dk_raw, fd_raw, teams_playing_count = init_baselines()
+dk_raw, fd_raw, dk_secondary, fd_secondary, teams_playing_count = init_baselines()
 dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
 fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
 
@@ -300,7 +300,7 @@ with tab1:
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames(10000)
               FD_seed = init_FD_seed_frames(10000)
-              dk_raw, fd_raw, teams_playing_count = init_baselines()
+              dk_raw, fd_raw, dk_secondary, fd_secondary, teams_playing_count = init_baselines()
               dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
               fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
 
@@ -633,7 +633,7 @@ with tab2:
                     del st.session_state[key]
                 DK_seed = init_DK_seed_frames(10000)
                 FD_seed = init_FD_seed_frames(10000)
-                dk_raw, fd_raw, teams_playing_count = init_baselines()
+                dk_raw, fd_raw, dk_secondary, fd_secondary, teams_playing_count = init_baselines()
                 dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
                 fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))