Modify init_DK_lineups and init_FD_lineups to support Regular and Showdown slate types

- Add 'type' parameter to differentiate between Regular and Showdown slates
- Update database collection selection based on slate type
- Modify tab2 to include slate type selection for DraftKings and FanDuel
- Remove hardcoded lineup initializations at the start of the script

app.py

@@ -50,14 +50,20 @@ def init_baselines():
     return roo_data, sd_roo_data, timestamp
 
 @st.cache_data(ttl = 60)
-def init_DK_lineups():  
+def init_DK_lineups(type):  
         
-        collection = db['PGA_DK_Seed_Frame_Name_Map']
+        if type == 'Regular':
+            collection = db['PGA_DK_Seed_Frame_Name_Map']
+        elif type == 'Showdown':
+            collection = db['PGA_DK_SD_Seed_Frame_Name_Map']
         cursor = collection.find()
         raw_data = pd.DataFrame(list(cursor))
         names_dict = dict(zip(raw_data['key'], raw_data['value']))
-    
-        collection = db["PGA_DK_Seed_Frame"] 
+        
+        if type == 'Regular':
+            collection = db["PGA_DK_Seed_Frame"] 
+        elif type == 'Showdown':
+            collection = db["PGA_DK_SD_Seed_Frame"]
         cursor = collection.find().limit(10000)
     
         raw_display = pd.DataFrame(list(cursor))
@@ -70,14 +76,20 @@ def init_DK_lineups():
         return DK_seed
 
 @st.cache_data(ttl = 60)
-def init_FD_lineups():  
+def init_FD_lineups(type):  
         
-        collection = db['PGA_DK_Seed_Frame_Name_Map']
+        if type == 'Regular':
+            collection = db['PGA_DK_Seed_Frame_Name_Map']
+        elif type == 'Showdown':
+            collection = db['PGA_DK_SD_Seed_Frame_Name_Map']
         cursor = collection.find()
         raw_data = pd.DataFrame(list(cursor))
         names_dict = dict(zip(raw_data['key'], raw_data['value']))
-    
-        collection = db["PGA_DK_Seed_Frame"] 
+        
+        if type == 'Regular':
+            collection = db["PGA_DK_Seed_Frame"] 
+        elif type == 'Showdown':
+            collection = db["PGA_DK_SD_Seed_Frame"]
         cursor = collection.find().limit(10000)
     
         raw_display = pd.DataFrame(list(cursor))
@@ -98,8 +110,6 @@ def convert_df(array):
     return array.to_csv().encode('utf-8')
 
 roo_data, sd_roo_data, timestamp = init_baselines()
-dk_lineups = init_DK_lineups()
-fd_lineups = init_FD_lineups()
 hold_display = roo_data
 lineup_display = []
 check_list = []
@@ -115,8 +125,8 @@ with tab1:
             # i.e. clear values from both square and cube
             st.cache_data.clear()
             roo_data, sd_roo_data, timestamp = init_baselines()
-            dk_lineups = init_DK_lineups()
-            fd_lineups = init_FD_lineups()
+            dk_lineups = init_DK_lineups('Regular')
+            fd_lineups = init_FD_lineups('Regular')
             hold_display = roo_data
             for key in st.session_state.keys():
                 del st.session_state[key]
@@ -169,14 +179,25 @@ with tab2:
             st.cache_data.clear()
             roo_data, sd_roo_data, timestamp = init_baselines()
             hold_display = roo_data
-            dk_lineups = init_DK_lineups()
-            fd_lineups = init_FD_lineups()
+            dk_lineups = init_DK_lineups('Regular')
+            fd_lineups = init_FD_lineups('Regular')
             t_stamp = f"Last Update: " + str(timestamp) + f" CST"
             for key in st.session_state.keys():
                 del st.session_state[key]
               
-        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Just the Main Slate'))
+        slate_var1 = st.radio("Which data are you loading?", ('Regular', 'Showdown'))
+        
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
+        if slate_var1 == 'Regular':
+            if site_var1 == 'Draftkings':
+                dk_lineups = init_DK_lineups('Regular')
+            elif site_var1 == 'Fanduel':
+                fd_lineups = init_FD_lineups('Regular')
+        elif slate_var1 == 'Showdown':
+            if site_var1 == 'Draftkings':
+                dk_lineups = init_DK_lineups('Showdown')
+            elif site_var1 == 'Fanduel':
+                fd_lineups = init_FD_lineups('Showdown')
         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 site_var1 == 'Draftkings':