Refactor init_baselines function in app.py to accept a slate variable, allowing dynamic selection of slate data for both DK and FD seed frame initializations. This change enhances flexibility in data retrieval and improves accuracy in contest simulations by ensuring the correct slate is used throughout the application.

app.py

@@ -106,12 +106,12 @@ def init_FD_Secondary_seed_frames(sharp_split):
     return FD_seed
 
 @st.cache_data(ttl = 599)
-def init_baselines():
+def init_baselines(slate_var):
     collection = db["DK_NFL_ROO"] 
     cursor = collection.find()
 
     raw_display = pd.DataFrame(list(cursor))
-    raw_display = raw_display[raw_display['slate'] == 'Main Slate']
+    raw_display = raw_display[raw_display['slate'] == slate_var]
     raw_display = raw_display[raw_display['version'] == 'overall']
     dk_raw = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
                           'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
@@ -121,7 +121,7 @@ def init_baselines():
     cursor = collection.find()
 
     raw_display = pd.DataFrame(list(cursor))
-    raw_display = raw_display[raw_display['slate'] == 'Main Slate']
+    raw_display = raw_display[raw_display['slate'] == slate_var]
     raw_display = raw_display[raw_display['version'] == 'overall']
     fd_raw = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
                           'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
@@ -179,7 +179,7 @@ def sim_contest(Sim_size, seed_frame, maps_dict, Contest_Size):
         
     return Sim_Winners
 
-dk_raw, fd_raw = init_baselines()
+dk_raw, fd_raw = init_baselines('Main Slate')
 dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
 fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
 
@@ -193,7 +193,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 = init_baselines()
+              dk_raw, fd_raw = init_baselines('Main Slate')
               dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
               fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
               
@@ -279,9 +279,11 @@ with tab2:
                     if slate_var1 == 'Main Slate':
                         st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
                         dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_id))
+                        dk_raw, fd_raw = init_baselines('Main Slate')
                     elif slate_var1 == 'Secondary Slate':
                         st.session_state.working_seed = init_DK_Secondary_seed_frames(sharp_split_var)
                         dk_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_id))
+                        dk_raw, fd_raw = init_baselines('Secondary Slate')
                         
                     raw_baselines = dk_raw
                     column_names = dk_columns
@@ -299,9 +301,11 @@ with tab2:
                     if slate_var1 == 'Main Slate':
                         st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
                         fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_id))
+                        dk_raw, fd_raw = init_baselines('Main Slate')
                     elif slate_var1 == 'Secondary Slate':
                         st.session_state.working_seed = init_FD_Secondary_seed_frames(sharp_split_var)
                         fd_id_dict = dict(zip(st.session_state.working_seed.Player, st.session_state.working_seed.player_id))
+                        dk_raw, fd_raw = init_baselines('Secondary Slate')
                         
                     raw_baselines = fd_raw
                     column_names = fd_columns
@@ -322,7 +326,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 = init_baselines()
+              dk_raw, fd_raw = init_baselines('Main Slate')
               dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
               fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))