Update app.py

app.py

@@ -252,11 +252,11 @@ if st.session_state['current_page'] == 'home':
 
 # Display the column selection UI based on the current page
 if st.session_state['current_page'] == 'premiums_df':
-    column_selection_ui('premiums_df')
+    column_selection_ui('premiums_df', 'premiums')
 elif st.session_state['current_page'] == 'claims_recoveries_df':
-    column_selection_ui('claims_recoveries_df')
+    column_selection_ui('claims_recoveries_df', 'claims_recoveries')
 elif st.session_state['current_page'] == 'claims_count_df':
-    column_selection_ui('claims_count_df')
+    column_selection_ui('claims_count_df', 'claims_count')
 
 # Add a button to go back to the home page from any selection page
 if st.session_state['current_page'] != 'home':