Update app.py

app.py

@@ -38,7 +38,7 @@ def init_conn():
         
         uri = st.secrets['mongo_uri']
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
-        db = client["testing_db"]
+        db = client["NFL_Database"]
      
         NFL_Data = st.secrets['NFL_Data']
 
@@ -57,49 +57,68 @@ fd_columns = ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'sal
 @st.cache_data(ttl = 600)
 def init_DK_seed_frames():  
     
-        collection = db["DK_NFL_seed_frame"] 
-        cursor = collection.find()
+    collection = db["DK_NFL_seed_frame"] 
+    cursor = collection.find()
+
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    DK_seed = raw_display.to_numpy()
+
+    return DK_seed
+
+@st.cache_data(ttl = 600)
+def init_DK_Secondary_seed_frames():  
     
-        raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
-        DK_seed = raw_display.to_numpy()
+    collection = db["DK_NFL_Secondary_seed_frame"] 
+    cursor = collection.find()
+
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    DK_seed = raw_display.to_numpy()
 
-        return DK_seed
+    return DK_seed
 
 @st.cache_data(ttl = 599)
 def init_FD_seed_frames():  
     
-        collection = db["FD_NFL_seed_frame"] 
-        cursor = collection.find()
+    collection = db["FD_NFL_seed_frame"] 
+    cursor = collection.find()
+
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    FD_seed = raw_display.to_numpy()
+
+    return FD_seed
+
+@st.cache_data(ttl = 599)
+def init_FD_Secondary_seed_frames():  
     
-        raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
-        FD_seed = raw_display.to_numpy()
+    collection = db["FD_NFL_Secondary_seed_frame"] 
+    cursor = collection.find()
 
-        return FD_seed
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+    FD_seed = raw_display.to_numpy()
+
+    return FD_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_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_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'])
+    collection = db["DK_NFL_ROO"] 
+    cursor = collection.find()
+
+    raw_display = pd.DataFrame(list(cursor))
+    dk_raw = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
+                          'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
+    dk_raw['STDev'] = (dk_raw['Ceiling'] - dk_raw['Floor']) / 4
+
+    collection = db["FD_NFL_ROO"] 
+    cursor = collection.find()
+
+    raw_display = pd.DataFrame(list(cursor))
+    fd_raw = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
+                          'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
+    fd_raw['STDev'] = (fd_raw['Ceiling'] - fd_raw['Floor']) / 4
 
     return dk_raw, fd_raw
 
@@ -168,10 +187,14 @@ with tab2:
               FD_seed = init_FD_seed_frames()
               dk_raw, fd_raw = init_baselines()
               
-        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'))
+        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
         if site_var1 == 'Draftkings':
-            DK_seed = init_DK_seed_frames()
+            if slate_var1 == 'Main Slate':
+                DK_seed = init_DK_seed_frames()
+            elif slate_var1 == 'Secondary Slate':
+                DK_seed = init_DK_Secondary_seed_frames()
+                
             raw_baselines = dk_raw
             column_names = dk_columns
             
@@ -188,7 +211,11 @@ with tab2:
                     stack_var2 = [5, 4, 3, 2, 1, 0]
                     
         elif site_var1 == 'Fanduel':
-            FD_seed = init_FD_seed_frames()
+            if slate_var1 == 'Main Slate':
+                FD_seed = init_FD_seed_frames()
+            elif slate_var1 == 'Secondary Slate':
+                FD_seed = init_FD_Secondary_seed_frames()
+                
             raw_baselines = fd_raw
             column_names = fd_columns
             
@@ -252,14 +279,22 @@ with tab1:
               DK_seed = init_DK_seed_frames()
               FD_seed = init_FD_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_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate'), 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':
-            DK_seed = init_DK_seed_frames()
+            if sim_slate_var1 == 'Main Slate':
+                DK_seed = init_DK_seed_frames()
+            elif sim_slate_var1 == 'Secondary Slate':
+                DK_seed = init_DK_Secondary_seed_frames()
+                
             raw_baselines = dk_raw
             column_names = dk_columns
         elif sim_site_var1 == 'Fanduel':
-            FD_seed = init_FD_seed_frames()
+            if sim_slate_var1 == 'Main Slate':
+                FD_seed = init_FD_seed_frames()
+            elif sim_slate_var1 == 'Secondary Slate':
+                FD_seed = init_FD_Secondary_seed_frames()
+                
             raw_baselines = fd_raw
             column_names = fd_columns