Refactor init_DK_seed_frames and init_FD_seed_frames functions in app.py to accept slate_var as a parameter, allowing for dynamic database collection selection based on slate type. Updated related function calls throughout the application to enhance flexibility in data loading for Draftkings and Fanduel simulations.

app.py

@@ -52,28 +52,40 @@ st.markdown("""
 </style>""", unsafe_allow_html=True)
 
 @st.cache_data(ttl = 600)
-def init_DK_seed_frames(sharp_split):  
-    
-        collection = db["DK_NHL_seed_frame"] 
-        cursor = collection.find().limit(sharp_split)
-    
-        raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['C1', 'C2', 'W1', 'W2', 'W3', 'D1', 'D2', 'G', 'FLEX', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
-        DK_seed = raw_display.to_numpy()
+def init_DK_seed_frames(sharp_split, slate_var):  
+
+    if slate_var == 'Main Slate':
+        collection = db['DK_NHL_seed_frame_Main Slate']
+    elif slate_var == 'Secondary Slate':
+        collection = db['DK_NHL_seed_frame_Secondary Slate']
+    elif slate_var == 'Auxiliary Slate':
+        collection = db['DK_NHL_seed_frame_Auxiliary Slate']
 
-        return DK_seed
+    cursor = collection.find().limit(sharp_split)
+
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[['C1', 'C2', 'W1', 'W2', 'W3', 'D1', 'D2', 'G', 'FLEX', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    DK_seed = raw_display.to_numpy()
+
+    return DK_seed
 
 @st.cache_data(ttl = 599)
-def init_FD_seed_frames(sharp_split):  
-    
-        collection = db["FD_NHL_seed_frame"] 
-        cursor = collection.find().limit(sharp_split)
+def init_FD_seed_frames(sharp_split, slate_var):  
     
-        raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['C1', 'C2', 'W1', 'W2', 'D1', 'D2', 'FLEX1', 'FLEX2', 'G', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
-        FD_seed = raw_display.to_numpy()
+    if slate_var == 'Main Slate':
+        collection = db['FD_NHL_seed_frame_Main Slate']
+    elif slate_var == 'Secondary Slate':
+        collection = db['FD_NHL_seed_frame_Secondary Slate']
+    elif slate_var == 'Auxiliary Slate':
+        collection = db['FD_NHL_seed_frame_Auxiliary Slate']
+
+    cursor = collection.find().limit(sharp_split)
+
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[['C1', 'C2', 'W1', 'W2', 'D1', 'D2', 'FLEX1', 'FLEX2', 'G', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    FD_seed = raw_display.to_numpy()
 
-        return FD_seed
+    return FD_seed
 
 @st.cache_data(ttl = 599)
 def init_baselines():
@@ -167,13 +179,13 @@ with tab1:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              DK_seed = init_DK_seed_frames(10000)
-              FD_seed = init_FD_seed_frames(10000)
+              DK_seed = init_DK_seed_frames(10000, 'Main Slate')
+              FD_seed = init_FD_seed_frames(10000, 'Main Slate')
               dk_raw, fd_raw, 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))
 
-        sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'), key='sim_slate_var1')
+        sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate', 'Auxiliary Slate'), key='sim_slate_var1')
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='sim_site_var1')
             
         contest_var1 = st.selectbox("What contest size are you simulating?", ('Small', 'Medium', 'Large', 'Custom'))
@@ -237,17 +249,15 @@ with tab1:
             
         else:
             if sim_site_var1 == 'Draftkings':
-                if sim_slate_var1 == 'Main Slate':
-                    st.session_state.working_seed = init_DK_seed_frames(sharp_split)
-                    dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
-                    raw_baselines = dk_raw
-                    column_names = dk_columns
+                st.session_state.working_seed = init_DK_seed_frames(sharp_split, sim_slate_var1)
+                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)
-                    fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
-                    raw_baselines = fd_raw
-                    column_names = fd_columns
+                st.session_state.working_seed = init_FD_seed_frames(sharp_split, sim_slate_var1)
+                fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
+                raw_baselines = fd_raw
+                column_names = fd_columns
 
             st.session_state.maps_dict = {
                     'Projection_map':dict(zip(raw_baselines.Player,raw_baselines.Median)),
@@ -628,7 +638,7 @@ with tab2:
                 dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
                 fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
                 
-        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'))
+        slate_var2 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate', 'Auxiliary Slate'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
         sharp_split_var = st.number_input("How many lineups do you want?", value=10000, max_value=500000, min_value=10000, step=10000)
         lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=500, value=10, step=1)
@@ -675,20 +685,18 @@ with tab2:
                     st.session_state.data_export_display = st.session_state.working_seed[0:lineup_num_var]
                 elif 'working_seed' not in st.session_state:
                     if site_var1 == 'Draftkings':
-                        if slate_var1 == 'Main Slate':
-                            st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
+                        st.session_state.working_seed = init_DK_seed_frames(sharp_split_var, slate_var2)
 
-                            dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
-                            raw_baselines = dk_raw
-                            column_names = dk_columns
+                        dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
+                        raw_baselines = dk_raw
+                        column_names = dk_columns
                 
                     elif site_var1 == 'Fanduel':
-                        if slate_var1 == 'Main Slate':
-                            st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
+                        st.session_state.working_seed = init_FD_seed_frames(sharp_split_var, slate_var2)
 
-                            fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
-                            raw_baselines = fd_raw
-                            column_names = fd_columns
+                        fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))
+                        raw_baselines = fd_raw
+                        column_names = fd_columns
                     st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
                     st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                     st.session_state.data_export_display = st.session_state.working_seed[0:lineup_num_var]
@@ -709,7 +717,7 @@ with tab2:
                 st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                 st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
             elif 'working_seed' not in st.session_state:
-                if slate_var1 == 'Main Slate':
+                if slate_var2 == 'Main Slate':
                     st.session_state.working_seed = init_DK_seed_frames(sharp_split_var)
                     dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_id))
 
@@ -726,7 +734,7 @@ with tab2:
                 st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
                 st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
             elif 'working_seed' not in st.session_state:
-                if slate_var1 == 'Main Slate':
+                if slate_var2 == 'Main Slate':
                     st.session_state.working_seed = init_FD_seed_frames(sharp_split_var)
                     fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_id))