Update app.py

app.py

@@ -213,7 +213,6 @@ def column_selection_ui(dataframe_key, selection_key):
         }
 
 
-# File uploader and buttons
 upload_col1, upload_col2, upload_col3 = st.columns(3)
 with upload_col1:
     uploaded_file_premiums = st.file_uploader("Upload Premiums Data", key="premiums")
@@ -222,35 +221,32 @@ with upload_col2:
 with upload_col3:
     uploaded_file_claims_count = st.file_uploader("Upload Claims Count Data", key="claims_count")
 
-button_col1, button_col2, button_col3 = st.columns(3)
-premiumlabel = "Select Premiums columns"
-incurredlabel = "Select Incurred claims columns"
-recoverlabel = "Select Recoveries columns"
-claimscount = "Select Claims count columns"
-
-# Display buttons if the corresponding file has been uploaded
-if st.session_state.active_page is None:
+# Only show buttons if we're on the home page
+if st.session_state.current_page == 'home':
+    button_col1, button_col2, button_col3 = st.columns(3)
     with button_col1:
-        if uploaded_file_premiums is not None:
-            if st.button(premiumlabel):
-                process_premiums_file(uploaded_file_premiums)
-
+        if uploaded_file_premiums and st.button("Select Premiums columns"):
+            process_file(uploaded_file_premiums, 'premiums_df')
     with button_col2:
-        if uploaded_file_claims_recoveries is not None:
-            if st.button(incurredlabel):
-                process_claims_recoveries_file(uploaded_file_claims_recoveries)
-            if st.button(recoverlabel):
-                process_claims_recoveries_file(uploaded_file_claims_recoveries)  # Modify as needed for different functionality
-
+        if uploaded_file_claims_recoveries:
+            if st.button("Select Incurred claims columns"):
+                process_file(uploaded_file_claims_recoveries, 'claims_recoveries_df')
+            if st.button("Select Recoveries columns"):
+                # Use a different key or logic for processing recoveries
+                process_file(uploaded_file_claims_recoveries, 'claims_recoveries_df')
     with button_col3:
-        if uploaded_file_claims_count is not None:
-            if st.button(claimscount):
-                process_claims_count_file(uploaded_file_claims_count)
-
-# Conditional rendering based on the active page
-if st.session_state.active_page == 'premiums_selection' and st.session_state.premiums_selection:
-    column_selection_ui('premiums_df', 'premiums_selection')
-elif st.session_state.active_page == 'claims_recoveries_selection' and st.session_state.claims_recoveries_selection:
-    column_selection_ui('claims_recoveries_df', 'claims_recoveries_selection')
-elif st.session_state.active_page == 'claims_count_selection' and st.session_state.claims_count_selection:
-    column_selection_ui('claims_count_df', 'claims_count_selection')
+        if uploaded_file_claims_count and st.button("Select Claims count columns"):
+            process_file(uploaded_file_claims_count, 'claims_count_df')
+
+# Display the column selection UI based on the current page
+if st.session_state.current_page == 'premiums_df':
+    column_selection_ui('premiums_df')
+elif st.session_state.current_page == 'claims_recoveries_df':
+    column_selection_ui('claims_recoveries_df')
+elif st.session_state.current_page == 'claims_count_df':
+    column_selection_ui('claims_count_df')
+
+# Add a button to go back to the home page from any selection page
+if st.session_state.current_page != 'home':
+    if st.button("Back to Home"):
+        st.session_state.current_page = 'home'