Update app.py

app.py

@@ -23,14 +23,16 @@ def init_conn():
           "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/gspread-connection%40model-sheets-connect.iam.gserviceaccount.com"
         }
         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)
 
-        return gc_con, uri, MLB_Data
+        return gc_con, db, MLB_Data
     
-gcservice_account, uri, MLB_Data = init_conn()
+gcservice_account, db, MLB_Data = init_conn()
 
 percentages_format = {'Exposure': '{:.2%}'}
 freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
@@ -38,9 +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(uri):  
-        client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=100000)
-        db = client["testing_db"]
+def init_seed_frames():  
     
         collection = db["DK_MLB_seed_frame"] 
         cursor = collection.find()
@@ -56,8 +56,6 @@ def init_seed_frames(uri):
         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()
 
-        client.close()
-
         return DK_Seed, FD_Seed
 
 @st.cache_data(ttl = 599)
@@ -131,7 +129,7 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
         
     return Sim_Winners
 
-DK_Seed, FD_Seed = init_seed_frames(uri)
+DK_Seed, FD_Seed = init_seed_frames()
 dk_raw, fd_raw = init_baselines()
 
 tab1, tab2 = st.tabs(['Data Export', 'Contest Sims'])
@@ -142,7 +140,7 @@ 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(db)
+              DK_Seed, FD_Seed = init_seed_frames()
               dk_raw, fd_raw = init_baselines()
               
         slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'))
@@ -228,7 +226,7 @@ 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(db)
+              DK_Seed, FD_Seed = init_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')