Update app.py

app.py

@@ -55,13 +55,21 @@ def load_overall_stats():
     raw_display = raw_display.loc[raw_display['Median'] > 0]
     raw_display = raw_display.apply(pd.to_numeric, errors='ignore')
     fd_raw = raw_display.sort_values(by='Median', ascending=False)
+    
+    worksheet = sh.worksheet('Player_Level_ROO')
+    raw_display = pd.DataFrame(worksheet.get_all_records())
+    raw_display.replace("", 'Welp', inplace=True)
+    raw_display = raw_display.loc[raw_display['Player'] != 'Welp']
+    raw_display = raw_display.loc[raw_display['Median'] > 0]
+    raw_display = raw_display.apply(pd.to_numeric, errors='ignore')
+    roo_raw = raw_display.sort_values(by='Median', ascending=False)
 
-    return dk_raw, fd_raw
+    return dk_raw, fd_raw, roo_raw
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-dk_raw, fd_raw = load_overall_stats()
+dk_raw, fd_raw, roo_raw = load_overall_stats()
 
 tab1, tab2, tab3 = st.tabs(['Uploads and Info', 'Range of Outcomes', 'Custom Range of Outcomes'])
 
@@ -88,14 +96,14 @@ with tab2:
     with col1:
         if st.button("Load/Reset Data", key='reset1'):
               st.cache_data.clear()
-              dk_raw, fd_raw = load_overall_stats()
+              dk_raw, fd_raw, roo_raw = load_overall_stats()
               for key in st.session_state.keys():
                   del st.session_state[key]
         site_var2 = st.radio("What table would you like to display?", ('Draftkings', 'Fanduel'), key='site_var2')
         if site_var2 == 'Draftkings':
-            raw_baselines = dk_raw
+            raw_baselines = roo_raw[roo_raw['Site'] == 'Draftkings']
         elif site_var2 == 'Fanduel':
-            raw_baselines = fd_raw
+            raw_baselines = roo_raw[roo_raw['Site'] == 'Fanduel']
         split_var2 = st.radio("Are you running the full slate or crtain games?", ('Full Slate Run', 'Specific Games'), key='split_var2')
         if split_var2 == 'Specific Games':
             team_var2 = st.multiselect('Which teams would you like to include in the ROO?', options = raw_baselines['Team'].unique(), key='team_var2')
@@ -136,7 +144,7 @@ with tab3:
     with col1:
         if st.button("Load/Reset Data", key='reset2'):
               st.cache_data.clear()
-              dk_raw, fd_raw = load_overall_stats()
+              dk_raw, fd_raw, roo_raw = load_overall_stats()
               for key in st.session_state.keys():
                   del st.session_state[key]
         slate_var1 = st.radio("Which data are you loading?", ('Paydirt', 'User'), key='slate_var1')