Update app.py

app.py

@@ -42,7 +42,15 @@ percentages_format = {'Pts% Boost': '{:.2%}', 'Reb% Boost': '{:.2%}', 'Ast% Boos
 @st.cache_resource(ttl = 600)
 def init_baselines():
     sh = gcservice_account.open_by_url(DEM_data)
-
+    
+    worksheet = sh.worksheet('DEM Matchups')
+    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)
+    matchups = raw_display[raw_display['Var'] != ""]
+    matchups_dict = dict(zip(matchups['Team'], matchups['Opp']))
+    
     worksheet = sh.worksheet('PG_DEM_Calc')
     raw_display = pd.DataFrame(worksheet.get_values())
     raw_display.columns = raw_display.iloc[0]
@@ -107,34 +115,45 @@ def init_baselines():
     export_dem = overall_dem[['Team', 'Acro', 'G', 'Pts% Boost', 'Reb% Boost', 'Ast% Boost', '3p% Boost',
                                'Stl Boost%', 'Blk Boost%', 'TOV Boost%', 'FPPM', 'FPPM Boost', 'position']]
 
-    return export_dem
+    return export_dem, matchups, matchups_dict
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-overall_dem = init_baselines()
+overall_dem, matchups, matchups_dict = init_baselines()
 
 col1, col2 = st.columns([1, 9])
 with col1:
     if st.button("Reset Data", key='reset1'):
               st.cache_data.clear()
-              overall_dem = init_baselines()
-    split_var1 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var1')
-    if split_var1 == 'Specific Teams':
-        team_var1 = st.multiselect('Which teams would you like to include in the tables?', options = overall_dem['Acro'].unique(), key='team_var1')
-    elif split_var1 == 'All':
-        team_var1 = overall_dem.Acro.values.tolist()
-    split_var2 = st.radio("Would you like to view all positions or specific ones?", ('All', 'Specific Positions'), key='split_var2')
-    if split_var2 == 'Specific Positions':
-        pos_var1 = st.multiselect('Which teams would you like to include in the tables?', options = overall_dem['position'].unique(), key='pos_var1')
-    elif split_var2 == 'All':
-        pos_var1 = overall_dem.position.values.tolist()
+              overall_dem, matchups, matchups_dict = init_baselines()
+    split_var1 = st.radio("View all teams or just this main slate's matchups?", ('Slate Matchups', 'All'), key='split_var1')
+    if split_var1 == 'Slate Matchups':
+        view_var1 = matchups.Opp.values.tolist()
+    if split_var1 == 'All':
+        split_var2 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var2')
+        if split_var2 == 'Specific Teams':
+            team_var1 = st.multiselect('Which teams would you like to include in the tables?', options = overall_dem['Acro'].unique(), key='team_var1')
+        elif split_var2 == 'All':
+            team_var1 = overall_dem.Acro.values.tolist()
+        split_var3 = st.radio("Would you like to view all positions or specific ones?", ('All', 'Specific Positions'), key='split_var3')
+        if split_var3 == 'Specific Positions':
+            pos_var1 = st.multiselect('Which teams would you like to include in the tables?', options = overall_dem['position'].unique(), key='pos_var1')
+        elif split_var3 == 'All':
+            pos_var1 = overall_dem.position.values.tolist()
 with col2:
-    dem_display = overall_dem[overall_dem['Acro'].isin(team_var1)]
-    dem_display = dem_display[dem_display['position'].isin(pos_var1)]
-    dem_display = dem_display.sort_values(by='FPPM Boost', ascending=False)
-    dem_display.rename(columns={"Team": "Team (Giving Boost)"}, inplace = True)
-    dem_display = dem_display.set_index('Team (Giving Boost)')
+    if split_var1 == 'Slate Matchups':
+        dem_display = overall_dem[overall_dem['Acro'].isin(view_var1)]
+        dem_display['Team (Getting Boost)'] = dem_display['Acro'].map(matchups_dict)
+        dem_display.rename(columns={"Acro": "Opp (Giving Boost)"}, inplace = True)
+        dem_display = dem_display.set_index('Team (Getting Boost)')
+        dem_display = dem_display.sort_values(by='FPPM Boost', ascending=False)
+    elif split_var1 == 'All':
+        dem_display = overall_dem[overall_dem['Acro'].isin(team_var1)]
+        dem_display = dem_display[dem_display['position'].isin(pos_var1)]
+        dem_display = dem_display.sort_values(by='FPPM Boost', ascending=False)
+        dem_display.rename(columns={"Team": "Team (Giving Boost)"}, inplace = True)
+        dem_display = dem_display.set_index('Team (Giving Boost)')
     st.dataframe(dem_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(percentages_format, precision=2), use_container_width = True)
     st.download_button(
         label="Export DEM Numbers",