Update app.py

app.py

@@ -65,7 +65,19 @@ def init_DK_seed_frames():
         DK_seed = raw_display.to_numpy()
 
         return DK_seed
+    
+@st.cache_data(ttl = 599)
+def init_DK_secondary_seed_frames():  
+    
+        collection = db["DK_NFL_Secondary_SD_seed_frame"] 
+        cursor = collection.find()
+    
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+        DK_second_seed = raw_display.to_numpy()
 
+        return DK_second_seed
+    
 @st.cache_data(ttl = 599)
 def init_FD_seed_frames():  
     
@@ -77,6 +89,18 @@ def init_FD_seed_frames():
         FD_seed = raw_display.to_numpy()
 
         return FD_seed
+    
+@st.cache_data(ttl = 599)
+def init_FD_secondary_seed_frames():  
+    
+        collection = db["FD_NFL_Secondary_SD_seed_frame"] 
+        cursor = collection.find()
+    
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+        FD_second_seed = raw_display.to_numpy()
+
+        return FD_second_seed
 
 @st.cache_data(ttl = 599)
 def init_baselines():
@@ -154,8 +178,6 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
         
     return Sim_Winners
 
-DK_seed = init_DK_seed_frames()
-FD_seed = init_FD_seed_frames()
 dk_raw, fd_raw = init_baselines()
 
 tab1, tab2 = st.tabs(['Contest Sims', 'Data Export'])
@@ -166,13 +188,16 @@ with tab2:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              DK_seed = init_DK_seed_frames()
-              FD_seed = init_FD_seed_frames()
               dk_raw, fd_raw = init_baselines()
               
-        slate_var1 = st.radio("Which data are you loading?", ('Showdown', 'Other Showdown'))
+        slate_var1 = st.radio("Which data are you loading?", ('Showdown', 'Secondary Showdown'))
+            
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
         if site_var1 == 'Draftkings':
+            if slate_var1 == 'Showdown':
+                DK_seed = init_DK_seed_frames()
+            elif slate_var1 == 'Secondary Showdown':
+                DK_seed = init_DK_secondary_seed_frames()
             raw_baselines = dk_raw
             column_names = dk_columns
             
@@ -189,6 +214,10 @@ with tab2:
                     stack_var2 = [5, 4, 3, 2, 1, 0]
                     
         elif site_var1 == 'Fanduel':
+            if slate_var1 == 'Showdown':
+                FD_seed = init_FD_seed_frames()
+            elif slate_var1 == 'Secondary Showdown':
+                FD_seed = init_FD_secondary_seed_frames()
             raw_baselines = fd_raw
             column_names = fd_columns