Update app.py

app.py

@@ -30,19 +30,23 @@ def init_conn():
           "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/gspread-connection%40model-sheets-connect.iam.gserviceaccount.com"
         }
 
-        client = pymongo.MongoClient("mongodb+srv://multichem:[email protected]/testing_db")
-        db = client["testing_db"]
+        uri = "mongodb+srv://multichem:[email protected]/?retryWrites=true&w=majority"
 
         gc_con = gspread.service_account_from_dict(credentials, scope)
       
-        return gc_con, client, db
+        return gc_con, uri, db
 
-gcservice_account, client, db = init_conn()
+gcservice_account, uri, db = init_conn()
 
 percentages_format = {'PG': '{:.2%}', 'SG': '{:.2%}', 'SF': '{:.2%}', 'PF': '{:.2%}', 'C': '{:.2%}'}
 
 @st.cache_resource(ttl = 600)
 def init_baselines():
+    
+    # Create a new client and connect to the server
+    client = MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=100000)
+    db = client['testing_db']
+    
     collection = db["MLB_Hitters_DB"] 
     cursor = collection.find()  # Finds all documents in the collection
     
@@ -56,6 +60,12 @@ def init_baselines():
     
     hitter_gamelog_table = hitter_gamelog_table.set_axis(['Player', 'Team', 'Date', 'G', 'AB', 'PA', 'H', '1B', '2B', '3B', 'HR', 'R', 'RBI', 'BB', 'IBB', 'SO', 'HBP', 'SF', 'SH',
                                                           'GDP', 'SB', 'CS', 'AVG', 'SLG', 'wRC+', 'LD%', 'GB%', 'FB%', 'Hard%', 'Barrels', 'Barrel%'], axis=1)
+
+    client.close()
+
+    time.sleep(1)
+
+    client = MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=100000)
     
     collection = db["MLB_Pitchers_DB"] 
     cursor = collection.find()  # Finds all documents in the collection
@@ -78,6 +88,8 @@ def init_baselines():
                                                             'FB%', 'HR/FB', 'Hard%', 'Barrels', 'Barrel%', 'xERA', 'vFA', 'vFT', 'vFC', 'vFS', 'vFO', 'vSI',
                                                             'vSL', 'vCU', 'vKC', 'vEP', 'vCH', 'vSC', 'vKN'], axis=1)
     
+    client.close()
+    
     timestamp = pitcher_gamelog_table['Date'].max()
     
     return hitter_gamelog_table, pitcher_gamelog_table, timestamp