Refactor init_baselines function in app.py to accept a slate variable, allowing dynamic filtering of player data based on the selected slate. Update all relevant calls to init_baselines to ensure consistency across the application. Adjust sharp_split values for improved simulation accuracy. This enhances flexibility in data retrieval for DraftKings and FanDuel contests.

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,10 +179,6 @@ def sim_contest(Sim_size, seed_frame, maps_dict, Contest_Size):
         
     return Sim_Winners
 
-dk_raw, fd_raw = 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))
-
 tab1, tab2 = st.tabs(['Contest Sims', 'Data Export'])
 with tab2:
     col1, col2 = st.columns([1, 7])
@@ -193,7 +189,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))
               
@@ -205,8 +201,10 @@ with tab2:
             
             team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
             if team_var1 == 'Specific Teams':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
                     team_var2 = st.multiselect('Which teams do you want?', options = dk_raw['Team'].unique())
             elif team_var1 == 'Full Slate':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
                     team_var2 = dk_raw.Team.values.tolist()
             
             stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
@@ -219,8 +217,10 @@ with tab2:
             
             team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
             if team_var1 == 'Specific Teams':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
                     team_var2 = st.multiselect('Which teams do you want?', options = fd_raw['Team'].unique())
             elif team_var1 == 'Full Slate':
+                    dk_raw, fd_raw = init_baselines('Main Slate')
                     team_var2 = fd_raw.Team.values.tolist()
             
             stack_var1 = st.radio("Do you want a frame with specific stack sizes?", ('Full Slate', 'Specific Stack Sizes'), key='stack_var1')
@@ -279,9 +279,12 @@ 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 +302,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 +327,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))
 
@@ -342,13 +347,13 @@ with tab1:
         if strength_var1 == 'Not Very':
             sharp_split = 500000
         elif strength_var1 == 'Below Average':
-            sharp_split = 400000
+            sharp_split = 250000
         elif strength_var1 == 'Average':
-            sharp_split = 300000
+            sharp_split = 100000
         elif strength_var1 == 'Above Average':
-            sharp_split = 200000
+            sharp_split = 50000
         elif strength_var1 == 'Very':
-            sharp_split = 100000
+            sharp_split = 10000
 
     
     with col2:
@@ -390,16 +395,24 @@ with tab1:
                 if sim_site_var1 == 'Draftkings':
                     if sim_slate_var1 == 'Main Slate':
                         st.session_state.working_seed = init_DK_seed_frames(sharp_split)
+                        dk_raw, fd_raw = init_baselines('Main Slate')
+                        dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id)) 
                     elif sim_slate_var1 == 'Secondary Slate':
                         st.session_state.working_seed = init_DK_Secondary_seed_frames(sharp_split)
+                        dk_raw, fd_raw = init_baselines('Secondary Slate')
+                        dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id)) 
                         
                     raw_baselines = dk_raw
                     column_names = dk_columns
                 elif sim_site_var1 == 'Fanduel':
                     if sim_slate_var1 == 'Main Slate':
                         st.session_state.working_seed = init_FD_seed_frames(sharp_split)
+                        dk_raw, fd_raw = init_baselines('Main Slate')
+                        fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
                     elif sim_slate_var1 == 'Secondary Slate':
                         st.session_state.working_seed = init_FD_Secondary_seed_frames(sharp_split)
+                        dk_raw, fd_raw = init_baselines('Secondary Slate')
+                        fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
                         
                     raw_baselines = fd_raw
                     column_names = fd_columns