Update app.py

app.py

@@ -171,8 +171,14 @@ def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
 gamelog_table = init_baselines()
-data_cols = gamelog_table.columns.drop(['Player', 'Team', 'Position', 'Date', 'TOI'])
-season_data_cols = ['TOI', 'Goals', 'Total Assists', 'First Assists', 'Second Assists', 'Total Points',
+basic_cols = ['Player', 'Team', 'Position', 'Date', 'TOI']
+basic_season_cols = ['Player', 'Team', 'Position', 'TOI']
+data_cols = ['Goals', 'Total Assists', 'First Assists', 'Second Assists', 'Total Points', 'IPP',
+             'Shots', 'Shots%', 'ixG', 'iCF', 'iFF', 'iSCF', 'iHDCF', 'Rush Attempts', 'Rebounds Created', 'PIM', 'Total Penalties', 'Minor',
+             'Major', 'Penalties Drawn', 'Giveaways', 'Takeaways', 'Hits', 'Hits Taken', 'Shots Blocked', 'Faceoffs Won',
+             'Faceoffs Lost', 'Faceoffs%', 'dk_shots_bonus', 'dk_blocks_bonus', 'dk_goals_bonus', 'dk_points_bonus',
+             'dk_fantasy', 'fd_fantasy']
+season_data_cols = ['Goals', 'Total Assists', 'First Assists', 'Second Assists', 'Total Points',
                     'IPP', 'Shots', 'ixG', 'iCF', 'iFF', 'iSCF', 'iHDCF', 'Rush Attempts', 'Rebounds Created', 'PIM', 'Total Penalties',
                     'Minor', 'Major', 'Penalties Drawn', 'Giveaways', 'Takeaways', 'Hits', 'Hits Taken', 'Shots Blocked', 'Faceoffs Won',
                     'Faceoffs Lost', 'dk_shots_bonus', 'dk_blocks_bonus', 'dk_goals_bonus', 'dk_points_bonus',
@@ -191,7 +197,18 @@ with tab1:
         if st.button("Reset Data", key='reset1'):
                   st.cache_data.clear()
                   gamelog_table = init_baselines()
-                  data_cols = gamelog_table.columns.drop(['Player', 'Team', 'Position', 'Date', 'TOI'])
+                  basic_cols = ['Player', 'Team', 'Position', 'Date', 'TOI']
+                  basic_season_cols = ['Player', 'Team', 'Position', 'TOI']
+                  data_cols = ['Goals', 'Total Assists', 'First Assists', 'Second Assists', 'Total Points', 'IPP',
+                               'Shots', 'Shots%', 'ixG', 'iCF', 'iFF', 'iSCF', 'iHDCF', 'Rush Attempts', 'Rebounds Created', 'PIM', 'Total Penalties', 'Minor',
+                               'Major', 'Penalties Drawn', 'Giveaways', 'Takeaways', 'Hits', 'Hits Taken', 'Shots Blocked', 'Faceoffs Won',
+                               'Faceoffs Lost', 'Faceoffs%', 'dk_shots_bonus', 'dk_blocks_bonus', 'dk_goals_bonus', 'dk_points_bonus',
+                               'dk_fantasy', 'fd_fantasy']
+                  season_data_cols = ['Goals', 'Total Assists', 'First Assists', 'Second Assists', 'Total Points',
+                                      'IPP', 'Shots', 'ixG', 'iCF', 'iFF', 'iSCF', 'iHDCF', 'Rush Attempts', 'Rebounds Created', 'PIM', 'Total Penalties',
+                                      'Minor', 'Major', 'Penalties Drawn', 'Giveaways', 'Takeaways', 'Hits', 'Hits Taken', 'Shots Blocked', 'Faceoffs Won',
+                                      'Faceoffs Lost', 'dk_shots_bonus', 'dk_blocks_bonus', 'dk_goals_bonus', 'dk_points_bonus',
+                                      'dk_fantasy', 'fd_fantasy']
                   indv_teams = gamelog_table.drop_duplicates(subset='Team')
                   total_teams = indv_teams.Team.values.tolist()
                   indv_players = gamelog_table.drop_duplicates(subset='Player')
@@ -231,7 +248,9 @@ with tab1:
     with col2:
         if split_var1 == 'Season Logs':
             choose_cols = st.container()
-            choose_cols.multiselect('Which stats would you like to view?', options = season_data_cols, default = season_data_cols, key='col_display')
+            with choose_cols:
+                choose_disp = st.multiselect('Which stats would you like to view?', options = season_data_cols, default = season_data_cols, key='col_display')
+            disp_stats = basic_cols + choose_disp
             display = st.container()
             gamelog_table = gamelog_table[gamelog_table['Date'] >= low_date]
             gamelog_table = gamelog_table[gamelog_table['Date'] <= high_date]
@@ -241,7 +260,8 @@ with tab1:
             gamelog_table = gamelog_table[gamelog_table['Player'].isin(player_var1)]
             season_long_table = seasonlong_build(gamelog_table)
             season_long_table = season_long_table.set_index('Player')
-            display.dataframe(season_long_table.style.format(precision=2), height=750, use_container_width = True)  
+            season_long_table_disp = season_long_table[[disp_stats]]
+            display.dataframe(season_long_table_disp.style.format(precision=2), height=750, use_container_width = True)  
             
         elif split_var1 == 'Gamelogs':
             choose_cols = st.container()