Update app.py

app.py

@@ -40,7 +40,7 @@ dk_columns = ['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 's
 fd_columns = ['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count']
 
 @st.cache_data(ttl = 599)
-def init_seed_frames():  
+def init_DK_seed_frames():  
     
         collection = db["DK_MLB_seed_frame"] 
         cursor = collection.find()
@@ -49,6 +49,11 @@ def init_seed_frames():
         raw_display = raw_display[['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count']]
         DK_seed = raw_display.to_numpy()
 
+        return DK_Seed
+
+@st.cache_data(ttl = 599)
+def init_FD_seed_frames():  
+    
         collection = db["FD_MLB_seed_frame"] 
         cursor = collection.find()
     
@@ -56,7 +61,7 @@ def init_seed_frames():
         raw_display = raw_display[['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count']]
         FD_seed = raw_display.to_numpy()
 
-        return DK_Seed, FD_Seed
+        return FD_seed
 
 @st.cache_data(ttl = 599)
 def init_baselines():
@@ -129,7 +134,8 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
         
     return Sim_Winners
 
-DK_Seed, FD_Seed = init_seed_frames()
+DK_Seed = init_DK_seed_frames()
+FD_Seed = init_FD_seed_frames()
 dk_raw, fd_raw = init_baselines()
 
 tab1, tab2 = st.tabs(['Data Export', 'Contest Sims'])
@@ -140,7 +146,8 @@ with tab1:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              DK_Seed, FD_Seed = init_seed_frames()
+              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?", ('Main Slate', 'Other Main Slate'))
@@ -226,7 +233,8 @@ with tab2:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              DK_Seed, FD_Seed = init_seed_frames()
+              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')
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='sim_site_var1')