Update app.py

app.py

@@ -12,7 +12,7 @@ import gspread
 import gc
 import plotly.express as px
 import plotly.io as pio
-pio.renderers.default='browser'
+import pymongo
 
 @st.cache_resource
 def init_conn():
@@ -32,10 +32,12 @@ def init_conn():
         }
 
         gc_con = gspread.service_account_from_dict(credentials, scope)
+
+        client = pymongo.MongoClient("mongodb+srv://multichem:[email protected]/testing_db")
       
-        return gc_con
+        return gc_con, client
 
-gcservice_account = init_conn()
+gcservice_account, client = init_conn()
 
 NBA_Data = 'https://docs.google.com/spreadsheets/d/1Yq0vGriWK-bS79e-bD6_u9pqrYE6Yrlbb_wEkmH-ot0/edit#gid=1808117109'
 
@@ -45,11 +47,12 @@ percentages_format = {'PG': '{:.2%}', 'SG': '{:.2%}', 'SF': '{:.2%}', 'PF': '{:.
 def init_baselines():
     sh = gcservice_account.open_by_url(NBA_Data)
     
-    worksheet = sh.worksheet('Gamelog')
-    raw_display = pd.DataFrame(worksheet.get_values())
-    raw_display.columns = raw_display.iloc[0]
-    raw_display = raw_display[1:]
-    raw_display = raw_display.reset_index(drop=True)
+    db = client["gamelog"]
+    collection = db["rotations"] 
+    
+    cursor = collection.find()  # Finds all documents in the collection
+    
+    raw_display = pd.DataFrame(list(cursor))
     gamelog_table = raw_display[raw_display['PLAYER_NAME'] != ""]
     gamelog_table = gamelog_table[['PLAYER_NAME', 'POS', 'GAME_ID', 'TEAM_NAME', 'OPP_NAME', 'SEASON_ID', 'GAME_DATE', 'MATCHUP', 'MIN', 'touches', 'PTS', 'FGM', 'FGA', 'FG_PCT', 'FG3M', 'FG3A',
                                    'FG3_PCT', 'FTM', 'FTA', 'FT_PCT', 'reboundChancesOffensive', 'OREB', 'reboundChancesDefensive', 'DREB', 'reboundChancesTotal', 'REB',
@@ -90,11 +93,12 @@ def init_baselines():
                                             'Passes', 'Alt Assists', 'FT Assists', 'Assists', 'Stl', 'Blk', 'Tov', 'PF', 'DD', 'TD', 'Fantasy', 'FD_Fantasy', 'FPPM',
                                             'Rebound%', 'Assists/Pass', 'Touch_per_min', 'Fantasy/Touch', 'FD Fantasy/Touch', 'team_score', 'opp_score', 'spread'], axis=1)
     
-    worksheet = sh.worksheet('Rotations')
-    raw_display = pd.DataFrame(worksheet.get_values())
-    raw_display.columns = raw_display.iloc[0]
-    raw_display = raw_display[1:]
-    raw_display = raw_display.reset_index(drop=True)
+    db = client["testing_db"]
+    collection = db["rotations"] 
+    
+    cursor = collection.find()  # Finds all documents in the collection
+    
+    raw_display = pd.DataFrame(list(cursor))
     rot_table = raw_display[raw_display['Player'] != ""]
     rot_table = rot_table[['Player', 'Team', 'PG', 'SG', 'SF', 'PF', 'C', 'Given Pos']]
     data_cols = ['PG', 'SG', 'SF', 'PF', 'C']