Update app.py

app.py

@@ -26,11 +26,25 @@ def init_conn():
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=100000)
         db = client["testing_db"]
 
+        collection = db["DK_MLB_seed_frame"] 
+        cursor = collection.find()
+    
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3']]
+        DK_seed = raw_display.to_numpy()
+
+        collection = db["FD_MLB_seed_frame"] 
+        cursor = collection.find()
+    
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL']]
+        FD_seed = raw_display.to_numpy()
+
         gc_con = gspread.service_account_from_dict(credentials, scope)
       
-        return gc_con, client, db
+        return gc_con, client, db, DK_seed, FD_seed
 
-gcservice_account, client, db = init_conn()
+gcservice_account, client, db, DK_seed, FD_seedb = init_conn()
 
 MLB_Data = 'https://docs.google.com/spreadsheets/d/1f42Ergav8K1VsOLOK9MUn7DM_MLMvv4GR2Fy7EfnZTc/edit#gid=340831852'
 
@@ -56,28 +70,6 @@ def init_baselines():
 
     return dk_raw, fd_raw
 
-@st.cache_data(ttl = 59)
-def init_DK_seed_frame():
-    collection = db["DK_MLB_seed_frame"] 
-    cursor = collection.find()
-
-    raw_display = pd.DataFrame(list(cursor))
-    raw_display = raw_display[['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3']]
-    DK_seed = raw_display.to_numpy()
-
-    return DK_seed
-
-@st.cache_data(ttl = 59)
-def init_FD_seed_frame():
-    collection = db["FD_MLB_seed_frame"] 
-    cursor = collection.find()
-
-    raw_display = pd.DataFrame(list(cursor))
-    raw_display = raw_display[['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL']]
-    FD_seed = raw_display.to_numpy()
-
-    return FD_seed
-
 @st.cache_data
 def convert_df(array):
     array = pd.DataFrame(array, columns=column_names)
@@ -140,7 +132,7 @@ with tab1:
         if st.button("Load Seed Frame", key='seed_frame_load'):
             if site_var1 == 'Draftkings':
                 
-                working_seed = init_DK_seed_frame()
+                working_seed = DK_seed
 #                working_seed_parse = working_seed[np.isin(working_seed[:, 2], team_var2)]
 #                working_seed_parse = working_seed[np.isin(working_seed[:, 3], stack_var2)]
                 data_export_display = pd.DataFrame(working_seed[0:1000], columns=column_names)
@@ -157,7 +149,7 @@ with tab1:
                         
             elif site_var1 == 'Fanduel':
                 
-                working_seed = init_FD_seed_frame()
+                working_seed = FD_seed
 #                working_seed_parse = working_seed[np.isin(working_seed[:, 2], team_var2)]
 #                working_seed_parse = working_seed[np.isin(working_seed[:, 3], stack_var2)]
                 data_export_display = pd.DataFrame(working_seed[0:1000], columns=column_names)