Update app.py

app.py

@@ -171,6 +171,7 @@ 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'])
 indv_teams = gamelog_table.drop_duplicates(subset='Team')
 total_teams = indv_teams.Team.values.tolist()
 indv_players = gamelog_table.drop_duplicates(subset='Player')
@@ -185,6 +186,7 @@ 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'])
                   indv_teams = gamelog_table.drop_duplicates(subset='Team')
                   total_teams = indv_teams.Team.values.tolist()
                   indv_players = gamelog_table.drop_duplicates(subset='Player')
@@ -223,6 +225,9 @@ with tab1:
     
     with col2:
         if split_var1 == 'Season Logs':
+            choose_cols = st.container()
+            with choose_cols:
+                col_display = st.multiselect('Which stats would you like to view?', options = data_cols, default = data_cols, key='col_display')
             display = st.container()
             gamelog_table = gamelog_table[gamelog_table['Date'] >= low_date]
             gamelog_table = gamelog_table[gamelog_table['Date'] <= high_date]
@@ -235,6 +240,9 @@ with tab1:
             display.dataframe(season_long_table.style.format(precision=2), height=750, use_container_width = True)  
             
         elif split_var1 == 'Gamelogs':
+            choose_cols = st.container()
+            with choose_cols:
+                col_display = st.multiselect('Which stats would you like to view?', options = data_cols, default = data_cols, key='col_display')
             gamelog_table = gamelog_table[gamelog_table['Date'] >= low_date]
             gamelog_table = gamelog_table[gamelog_table['Date'] <= high_date]
             gamelog_table = gamelog_table[gamelog_table['TOI'] >= min_var1[0]]