Update app.py

app.py

@@ -6,7 +6,7 @@ import gspread
 import pymongo
 import time
 
-@st.cache_data(ttl = 599)
+@st.cache_resource
 def init_conn():
         scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/auth/drive']
 
@@ -25,7 +25,20 @@ def init_conn():
         uri = "mongodb+srv://multichem:[email protected]/?retryWrites=true&w=majority&appName=TestCluster"
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=100000)
         db = client["testing_db"]
+        
+        MLB_Data = 'https://docs.google.com/spreadsheets/d/1f42Ergav8K1VsOLOK9MUn7DM_MLMvv4GR2Fy7EfnZTc/edit#gid=340831852'
+
+        gc_con = gspread.service_account_from_dict(credentials, scope)
+    
+gcservice_account, db = init_conn()
+
+percentages_format = {'Exposure': '{:.2%}'}
+freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
+dk_columns = ['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count']
+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(database):      
         collection = db["DK_MLB_seed_frame"] 
         cursor = collection.find()
     
@@ -39,19 +52,8 @@ def init_conn():
         raw_display = pd.DataFrame(list(cursor))
         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()
-        
-        MLB_Data = 'https://docs.google.com/spreadsheets/d/1f42Ergav8K1VsOLOK9MUn7DM_MLMvv4GR2Fy7EfnZTc/edit#gid=340831852'
-
-        gc_con = gspread.service_account_from_dict(credentials, scope)
 
-        client.close()
-      
-        return gc_con, client, db, DK_seed, FD_seed, MLB_Data
-
-percentages_format = {'Exposure': '{:.2%}'}
-freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
-dk_columns = ['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count']
-fd_columns = ['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count']
+        return DK_Seed, FD_Seed
 
 @st.cache_data(ttl = 599)
 def init_baselines():
@@ -124,7 +126,7 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
         
     return Sim_Winners
 
-gcservice_account, client, db, DK_seed, FD_seed, MLB_Data = init_conn()
+DK_Seed, FD_Seed = init_seed_frames(db)
 dk_raw, fd_raw = init_baselines()
 
 tab1, tab2 = st.tabs(['Data Export', 'Contest Sims'])
@@ -135,7 +137,7 @@ with tab1:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              gcservice_account, client, db, DK_seed, FD_seed, MLB_Data = init_conn()
+              DK_Seed, FD_Seed = init_seed_frames(db)
               dk_raw, fd_raw = init_baselines()
               
         slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'))
@@ -221,7 +223,7 @@ with tab2:
               st.cache_data.clear()
               for key in st.session_state.keys():
                   del st.session_state[key]
-              gcservice_account, client, db, DK_seed, FD_seed, MLB_Data = init_conn()
+              DK_Seed, FD_Seed = init_seed_frames(db)
               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')