Aimed to add NBA as an option for the seed frame source

app.py

@@ -38,16 +38,17 @@ def init_conn():
         
         uri = st.secrets['mongo_uri']
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
-        db = client["testing_db"]
      
         NFL_Data = st.secrets['NFL_Data']
 
+        NBA_Data = st.secrets['NBA_Data']
+
         gc = gspread.service_account_from_dict(credentials)
         gc2 = gspread.service_account_from_dict(credentials2)
 
-        return gc, gc2, db, NFL_Data
+        return gc, gc2, client, NFL_Data, NBA_Data
     
-gcservice_account, gcservice_account2, db, NFL_Data = init_conn()
+gcservice_account, gcservice_account2, client, NFL_Data, NBA_Data = init_conn()
 
 percentages_format = {'Exposure': '{:.2%}'}
 freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
@@ -55,9 +56,13 @@ dk_columns = ['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'pro
 fd_columns = ['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 
 @st.cache_data(ttl = 599)
-def init_DK_seed_frames():  
-    
-        collection = db["DK_NFL_SD_seed_frame"] 
+def init_DK_seed_frames(sport): 
+        if sport == 'NFL':
+            db = client["testing_db"]
+        elif sport == 'NBA':
+            db = client["NBA_DFS"]
+        
+        collection = db[f"DK_{sport}_SD_seed_frame"] 
         cursor = collection.find()
     
         raw_display = pd.DataFrame(list(cursor))
@@ -67,9 +72,14 @@ def init_DK_seed_frames():
         return DK_seed
     
 @st.cache_data(ttl = 599)
-def init_DK_secondary_seed_frames():  
+def init_DK_secondary_seed_frames(sport):  
+        
+        if sport == 'NFL':
+            db = client["testing_db"]
+        elif sport == 'NBA':
+            db = client["NBA_DFS"]
     
-        collection = db["DK_NFL_Secondary_SD_seed_frame"] 
+        collection = db[f"DK_{sport}_Secondary_SD_seed_frame"] 
         cursor = collection.find()
     
         raw_display = pd.DataFrame(list(cursor))
@@ -79,9 +89,14 @@ def init_DK_secondary_seed_frames():
         return DK_second_seed
     
 @st.cache_data(ttl = 599)
-def init_FD_seed_frames():  
+def init_FD_seed_frames(sport):  
+        
+        if sport == 'NFL':
+            db = client["testing_db"]
+        elif sport == 'NBA':
+            db = client["NBA_DFS"]
     
-        collection = db["FD_NFL_SD_seed_frame"] 
+        collection = db[f"FD_{sport}_SD_seed_frame"] 
         cursor = collection.find()
     
         raw_display = pd.DataFrame(list(cursor))
@@ -91,9 +106,14 @@ def init_FD_seed_frames():
         return FD_seed
     
 @st.cache_data(ttl = 599)
-def init_FD_secondary_seed_frames():  
+def init_FD_secondary_seed_frames(sport):  
+        
+        if sport == 'NFL':
+            db = client["testing_db"]
+        elif sport == 'NBA':
+            db = client["NBA_DFS"]
     
-        collection = db["FD_NFL_Secondary_SD_seed_frame"] 
+        collection = db[f"FD_{sport}_Secondary_SD_seed_frame"] 
         cursor = collection.find()
     
         raw_display = pd.DataFrame(list(cursor))
@@ -103,27 +123,53 @@ def init_FD_secondary_seed_frames():
         return FD_second_seed
 
 @st.cache_data(ttl = 599)
-def init_baselines():
-    try:
-        sh = gcservice_account.open_by_url(NFL_Data)
-    except:
-        sh = gcservice_account2.open_by_url(NFL_Data)
-    
-    worksheet = sh.worksheet('DK_SD_ROO')
-    load_display = pd.DataFrame(worksheet.get_all_records())
-    load_display.replace('', np.nan, inplace=True)
-    load_display['STDev'] = load_display['Median'] / 4
-    load_display = load_display.drop_duplicates(subset=['Player'], keep='first')
-    
-    dk_raw = load_display.dropna(subset=['Median'])
-    
-    worksheet = sh.worksheet('FD_SD_ROO')
-    load_display = pd.DataFrame(worksheet.get_all_records())
-    load_display.replace('', np.nan, inplace=True)
-    load_display['STDev'] = load_display['Median'] / 4
-    load_display = load_display.drop_duplicates(subset=['Player'], keep='first')
+def init_baselines(sport):
+    if sport == 'NFL':
+        try:
+            sh = gcservice_account.open_by_url(NFL_Data)
+        except:
+            sh = gcservice_account2.open_by_url(NFL_Data)
+        
+        worksheet = sh.worksheet('DK_SD_ROO')
+        load_display = pd.DataFrame(worksheet.get_all_records())
+        load_display.replace('', np.nan, inplace=True)
+        load_display['STDev'] = load_display['Median'] / 4
+        load_display = load_display.drop_duplicates(subset=['Player'], keep='first')
+        
+        dk_raw = load_display.dropna(subset=['Median'])
+        
+        worksheet = sh.worksheet('FD_SD_ROO')
+        load_display = pd.DataFrame(worksheet.get_all_records())
+        load_display.replace('', np.nan, inplace=True)
+        load_display['STDev'] = load_display['Median'] / 4
+        load_display = load_display.drop_duplicates(subset=['Player'], keep='first')
+        
+        fd_raw = load_display.dropna(subset=['Median'])
     
-    fd_raw = load_display.dropna(subset=['Median'])
+    elif sport == 'NBA':
+         
+        try:
+            sh = gcservice_account.open_by_url(NBA_Data)
+        except:
+            sh = gcservice_account2.open_by_url(NBA_Data)
+        
+        worksheet = sh.worksheet('Player_Level_SD_ROO')
+        load_display = pd.DataFrame(worksheet.get_all_records())
+        load_display.replace('', np.nan, inplace=True)
+        load_display['STDev'] = load_display['Median'] / 4
+        load_display = load_display[load_display['site'] == 'Draftkings']
+        load_display = load_display.drop_duplicates(subset=['Player'], keep='first')
+        
+        dk_raw = load_display.dropna(subset=['Median'])
+        
+        worksheet = sh.worksheet('Player_Level_SD_ROO')
+        load_display = pd.DataFrame(worksheet.get_all_records())
+        load_display.replace('', np.nan, inplace=True)
+        load_display['STDev'] = load_display['Median'] / 4
+        load_display = load_display[load_display['site'] == 'Fanduel']
+        load_display = load_display.drop_duplicates(subset=['Player'], keep='first')
+        
+        fd_raw = load_display.dropna(subset=['Median'])
 
     return dk_raw, fd_raw
 
@@ -189,15 +235,16 @@ with tab2:
               for key in st.session_state.keys():
                   del st.session_state[key]
               dk_raw, fd_raw = init_baselines()
-              
+        
+        sport_var1 = st.radio("What sport are you working with?", ('NFL', 'NBA'))
         slate_var1 = st.radio("Which data are you loading?", ('Showdown', 'Secondary Showdown'))
             
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
         if site_var1 == 'Draftkings':
             if slate_var1 == 'Showdown':
-                DK_seed = init_DK_seed_frames()
+                DK_seed = init_DK_seed_frames(sport_var1)
             elif slate_var1 == 'Secondary Showdown':
-                DK_seed = init_DK_secondary_seed_frames()
+                DK_seed = init_DK_secondary_seed_frames(sport_var1)
             raw_baselines = dk_raw
             column_names = dk_columns
             
@@ -215,9 +262,9 @@ with tab2:
                     
         elif site_var1 == 'Fanduel':
             if slate_var1 == 'Showdown':
-                FD_seed = init_FD_seed_frames()
+                FD_seed = init_FD_seed_frames(sport_var1)
             elif slate_var1 == 'Secondary Showdown':
-                FD_seed = init_FD_secondary_seed_frames()
+                FD_seed = init_FD_secondary_seed_frames(sport_var1)
             raw_baselines = fd_raw
             column_names = fd_columns
             
@@ -279,20 +326,21 @@ with tab1:
               for key in st.session_state.keys():
                   del st.session_state[key]
               dk_raw, fd_raw = init_baselines()
+        sim_sport_var1 = st.radio("What sport are you working with?", ('NFL', 'NBA'))
         sim_slate_var1 = st.radio("Which data are you loading?", ('Showdown', 'Secondary Showdown'), key='sim_slate_var1')
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'), key='sim_site_var1')
         if sim_site_var1 == 'Draftkings':
             if sim_slate_var1 == 'Showdown':
-                DK_seed = init_DK_seed_frames()
+                DK_seed = init_DK_seed_frames(sim_sport_var1)
             elif sim_slate_var1 == 'Secondary Showdown':
-                DK_seed = init_DK_secondary_seed_frames()
+                DK_seed = init_DK_secondary_seed_frames(sim_sport_var1)
             raw_baselines = dk_raw
             column_names = dk_columns
         elif sim_site_var1 == 'Fanduel':
             if sim_slate_var1 == 'Showdown':
-                FD_seed = init_FD_seed_frames()
+                FD_seed = init_FD_seed_frames(sim_sport_var1)
             elif sim_slate_var1 == 'Secondary Showdown':
-                FD_seed = init_FD_secondary_seed_frames()
+                FD_seed = init_FD_secondary_seed_frames(sim_sport_var1)
             raw_baselines = fd_raw
             column_names = fd_columns