Update app.py

app.py

@@ -250,7 +250,7 @@ with tab1:
             choose_cols = st.container()
             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
+            disp_stats = basic_season_cols + choose_disp
             display = st.container()
             gamelog_table = gamelog_table[gamelog_table['Date'] >= low_date]
             gamelog_table = gamelog_table[gamelog_table['Date'] <= high_date]
@@ -260,12 +260,14 @@ 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')
-            season_long_table_disp = season_long_table[[disp_stats]]
+            season_long_table_disp = season_long_table.reindex(disp_stats,axis="columns")
             display.dataframe(season_long_table_disp.style.format(precision=2), height=750, use_container_width = True)  
             
         elif split_var1 == 'Gamelogs':
             choose_cols = st.container()
-            choose_cols.multiselect('Which stats would you like to view?', options = data_cols, default = data_cols, key='col_display')
+            with choose_cols:
+                choose_disp = st.multiselect('Which stats would you like to view?', options = data_cols, default = data_cols, key='col_display')
+            disp_stats = basic_cols + choose_disp
             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]]