Refactor init_DK_lineups and init_FD_lineups functions in app.py to accept type_var and slate_var parameters, enhancing lineup initialization flexibility. Update related function calls throughout the code for consistency and improved user experience.

app.py

@@ -86,25 +86,71 @@ def init_baselines():
     return roo_data, sd_roo_data, scoring_percentages
 
 @st.cache_data(ttl = 60)
-def init_DK_lineups(slate_var):  
+def init_DK_lineups(type_var, slate_var):  
         
-        collection = db2['DK_MLB_SD1_seed_frame']
-        cursor = collection.find().limit(10000)
+        if type_var == 'Regular':
+            if slate_var == 'Main':
+                collection = db2['DK_MLB_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Secondary':
+                collection = db2['DK_MLB_Secondary_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Auxiliary':
+                collection = db2['DK_MLB_Turbo_seed_frame']
+                cursor = collection.find().limit(10000)
+            
+            raw_display = pd.DataFrame(list(cursor))
+            raw_display = raw_display[['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'salary', 'proj', 'Own']]
+
+        elif type_var == 'Showdown':
+            if slate_var == 'Main':
+                collection = db2['DK_MLB_SD1_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Secondary':
+                collection = db2['DK_MLB_SD2_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Auxiliary':
+                collection = db2['DK_MLB_SD3_seed_frame']
+                cursor = collection.find().limit(10000)
+
+            raw_display = pd.DataFrame(list(cursor))
+            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Own']]
     
-        raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Own']]
         DK_seed = raw_display.to_numpy()
 
         return DK_seed
 
 @st.cache_data(ttl = 60)
-def init_FD_lineups(slate_var):  
+def init_FD_lineups(type_var,slate_var):  
         
-        collection = db2['FD_MLB_SD1_seed_frame']
-        cursor = collection.find().limit(10000)
+        if type_var == 'Regular':
+            if slate_var == 'Main':
+                collection = db2['FD_MLB_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Secondary':
+                collection = db2['FD_MLB_Secondary_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Auxiliary':
+                collection = db2['FD_MLB_Turbo_seed_frame']
+                cursor = collection.find().limit(10000)
+            
+            raw_display = pd.DataFrame(list(cursor))
+            raw_display = raw_display[['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'salary', 'proj', 'Own']]
+
+        elif type_var == 'Showdown':
+            if slate_var == 'Main':
+                collection = db2['FD_MLB_SD1_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Secondary':
+                collection = db2['FD_MLB_SD2_seed_frame']
+                cursor = collection.find().limit(10000)
+            elif slate_var == 'Auxiliary':
+                collection = db2['FD_MLB_SD3_seed_frame']
+                cursor = collection.find().limit(10000)
+            
+            raw_display = pd.DataFrame(list(cursor))
+            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Own']]
     
-        raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Own']]
         FD_seed = raw_display.to_numpy()
 
         return FD_seed
@@ -124,8 +170,8 @@ with col1:
         st.cache_data.clear()
         roo_data, sd_roo_data, scoring_percentages = init_baselines()
         hold_display = roo_data
-        dk_lineups = init_DK_lineups('Main')
-        fd_lineups = init_FD_lineups('Main')
+        dk_lineups = init_DK_lineups('Regular', 'Main')
+        fd_lineups = init_FD_lineups('Regular', 'Main')
         for key in st.session_state.keys():
             del st.session_state[key]
 with col2:
@@ -222,14 +268,14 @@ with tab3:
 
         if slate_type_var3 == 'Regular':
             if site_var == 'Draftkings':
-                dk_lineups = init_DK_lineups(slate_var3)
+                dk_lineups = init_DK_lineups(slate_type_var3, slate_var3)
             elif site_var == 'Fanduel':
-                fd_lineups = init_FD_lineups(slate_var3)
+                fd_lineups = init_FD_lineups(slate_type_var3, slate_var3)
         elif slate_type_var3 == 'Showdown':
             if site_var == 'Draftkings':
-                dk_lineups = init_DK_lineups(slate_var3)
+                dk_lineups = init_DK_lineups(slate_type_var3, slate_var3)
             elif site_var == 'Fanduel':
-                fd_lineups = init_FD_lineups(slate_var3)
+                fd_lineups = init_FD_lineups(slate_type_var3, slate_var3)
         lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=1000, value=150, step=1)
         
         if slate_type_var3 == 'Regular':