Update app.py

app.py

@@ -66,6 +66,18 @@ def init_DK_seed_frames():
 
         return DK_seed
 
+@st.cache_data(ttl = 600)
+def init_DK_secondary_seed_frames():  
+    
+        collection = db["DK_NBA_Secondary_seed_frame"] 
+        cursor = collection.find()
+    
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'FLEX', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+        DK_secondary = raw_display.to_numpy()
+
+        return DK_secondary
+
 @st.cache_data(ttl = 599)
 def init_FD_seed_frames():  
     
@@ -78,6 +90,18 @@ def init_FD_seed_frames():
 
         return FD_seed
 
+@st.cache_data(ttl = 599)
+def init_FD_secondary_seed_frames():  
+    
+        collection = db["FD_NBA_Secondary_seed_frame"] 
+        cursor = collection.find()
+    
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['PG1', 'PG2', 'SG1', 'SG2', 'SF1', 'SF2', 'PF1', 'PF2', 'C1', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+        FD_secondary = raw_display.to_numpy()
+
+        return FD_secondary
+
 @st.cache_resource(ttl = 301)
 def init_baselines():
     sh = gcservice_account.open_by_url('https://docs.google.com/spreadsheets/d/1Yq0vGriWK-bS79e-bD6_u9pqrYE6Yrlbb_wEkmH-ot0/edit#gid=172632260')
@@ -97,7 +121,17 @@ def init_baselines():
     fd_roo_raw['STDev'] = fd_roo_raw['Median'] / 4
     fd_raw = fd_roo_raw.dropna(subset=['Median'])
 
-    return dk_raw, fd_raw
+    dk_secondary_roo_raw = load_display[load_display['site'] == 'Draftkings']
+    dk_secondary_roo_raw = dk_secondary_roo_raw[dk_secondary_roo_raw['slate'] == 'Secondary Slate']
+    dk_secondary_roo_raw['STDev'] = dk_secondary_roo_raw['Median'] / 4
+    dk_secondary = dk_secondary_roo_raw.dropna(subset=['Median'])
+
+    fd_secondary_roo_raw = load_display[load_display['site'] == 'Fanduel']
+    fd_secondary_roo_raw = fd_secondary_roo_raw[fd_secondary_roo_raw['slate'] == 'Secondary Slate']
+    fd_secondary_roo_raw['STDev'] = fd_secondary_roo_raw['Median'] / 4
+    fd_secondary = fd_secondary_roo_raw.dropna(subset=['Median'])
+
+    return dk_raw, fd_raw, dk_secondary, fd_secondary
 
 @st.cache_data
 def convert_df(array):
@@ -154,7 +188,9 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
 
 DK_seed = init_DK_seed_frames()
 FD_seed = init_FD_seed_frames()
-dk_raw, fd_raw = init_baselines()
+DK_secondary = init_DK_secondary_seed_frames()
+FD_secondary = init_FD_secondary_seed_frames()
+dk_raw, fd_raw, dk_secondary, fd_secondary = init_baselines()
 
 tab1, tab2 = st.tabs(['Contest Sims', 'Data Export'])
 
@@ -167,15 +203,21 @@ with tab2:
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames()
               FD_seed = init_FD_seed_frames()
-              dk_raw, fd_raw = init_baselines()
+              DK_secondary = init_DK_secondary_seed_frames()
+              FD_secondary = init_FD_secondary_seed_frames()
+              dk_raw, fd_raw, dk_secondary, fd_secondary = init_baselines()
               
-        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Just the Main Slate'))
-        site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
+        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'), key='slate_var1')
+        site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='site_var1')
         lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=500, value=10, step=1)
 
         if site_var1 == 'Draftkings':
-            raw_baselines = dk_raw
-            column_names = dk_columns
+            if slate_var1 == 'Main Slate':
+                raw_baselines = dk_raw
+                column_names = dk_columns
+            elif slate_var1 == 'Secondary Slate':
+                raw_baselines = dk_secondary
+                column_names = dk_columns
             
             player_var1 = st.radio("Do you want a frame with specific Players?", ('Full Slate', 'Specific Players'), key='player_var1')
             if player_var1 == 'Specific Players':
@@ -184,8 +226,12 @@ with tab2:
                     player_var2 = dk_raw.Player.values.tolist()
                     
         elif site_var1 == 'Fanduel':
-            raw_baselines = fd_raw
-            column_names = fd_columns
+            if slate_var1 == 'Main Slate':
+                raw_baselines = fd_raw
+                column_names = fd_columns
+            elif slate_var1 == 'Secondary Slate':
+                raw_baselines = fd_secondary
+                column_names = fd_columns
             
             player_var1 = st.radio("Do you want a frame with specific Players?", ('Full Slate', 'Specific Players'), key='player_var1')
             if player_var1 == 'Specific Players':
@@ -258,15 +304,25 @@ with tab1:
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames()
               FD_seed = init_FD_seed_frames()
-              dk_raw, fd_raw = init_baselines()
-        sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'), key='sim_slate_var1')
+              DK_secondary = init_DK_secondary_seed_frames()
+              FD_secondary = init_FD_secondary_seed_frames()
+              dk_raw, fd_raw, dk_secondary, fd_secondary = init_baselines()
+        sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'), key='sim_slate_var1')
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='sim_site_var1')
         if sim_site_var1 == 'Draftkings':
-            raw_baselines = dk_raw
-            column_names = dk_columns
+            if sim_slate_var1 == 'Main Slate':
+                raw_baselines = dk_raw
+                column_names = dk_columns
+            elif sim_slate_var1 == 'Secondary Slate':
+                raw_baselines = dk_secondary
+                column_names = dk_columns
         elif sim_site_var1 == 'Fanduel':
-            raw_baselines = fd_raw
-            column_names = fd_columns
+            if sim_slate_var1 == 'Main Slate':
+                raw_baselines = fd_raw
+                column_names = fd_columns
+            elif sim_slate_var1 == 'Secondary Slate':
+                raw_baselines = fd_secondary
+                column_names = fd_columns
             
         contest_var1 = st.selectbox("What contest size are you simulating?", ('Small', 'Medium', 'Large', 'Custom'))
         if contest_var1 == 'Small':