Update app.py

app.py

@@ -101,9 +101,13 @@ def init_baselines():
     overall_dem = pd.concat([pg_dem, sg_dem, sf_dem, pf_dem, c_dem])
     overall_dem = overall_dem[['Acro', 'G', 'Pts% Boost', 'Reb% Boost', 'Ast% Boost', '3p% Boost',
                                'Stl Boost%', 'Blk Boost%', 'TOV Boost%', 'FPPM', 'FPPM Boost', 'position']]
+    overall_dem['Team'] = overall_dem['Acro'] + '-' + overall_dem['position']
     overall_dem = overall_dem.reset_index()
+    
+    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 overall_dem
+    return export_dem
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
@@ -126,10 +130,11 @@ with col1:
     elif split_var2 == 'All':
         pos_var1 = overall_dem.position.values.tolist()
 with col2:
-    dem_display = overall_dem[overall_dem['Acro'].isin(team_var1)]
+    dem_display = overall_dem[overall_dem['Team'].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={"Acro": "Team (Giving Boost)"}, inplace = True)
+    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",