Update app.py

app.py

@@ -137,15 +137,21 @@ with col3:
 
 
 
-# Set initial session states
-if 'active_page' not in st.session_state:
-    st.session_state.active_page = None
-if 'premiums_selection' not in st.session_state:
-    st.session_state.premiums_selection = False
-if 'claims_recoveries_selection' not in st.session_state:
-    st.session_state.claims_recoveries_selection = False
-if 'claims_count_selection' not in st.session_state:
-    st.session_state.claims_count_selection = False
+# Ensure the session state variables are initialized
+if 'current_page' not in st.session_state:
+    st.session_state.current_page = 'home'
+if 'premiums_df' not in st.session_state:
+    st.session_state.premiums_df = None
+if 'claims_recoveries_df' not in st.session_state:
+    st.session_state.claims_recoveries_df = None
+if 'claims_count_df' not in st.session_state:
+    st.session_state.claims_count_df = None
+
+if 'current_page' not in st.session_state:
+    st.session_state['current_page'] = 'home'
+if 'premiums_df' not in st.session_state:
+    st.session_state['premiums_df'] = None
+# ... Similarly for other dataframes
 
 # Helper function to check if a string has more than 50% numeric characters
 def is_mostly_numeric(s):
@@ -213,6 +219,7 @@ 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,7 +229,7 @@ with upload_col3:
     uploaded_file_claims_count = st.file_uploader("Upload Claims Count Data", key="claims_count")
 
 # Only show buttons if we're on the home page
-if st.session_state.current_page == 'home':
+if st.session_state['current_page'] == 'home':
     button_col1, button_col2, button_col3 = st.columns(3)
     with button_col1:
         if uploaded_file_premiums and st.button("Select Premiums columns"):
@@ -239,14 +246,14 @@ if st.session_state.current_page == 'home':
             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':
+if st.session_state['current_page'] == 'premiums_df':
     column_selection_ui('premiums_df')
-elif st.session_state.current_page == 'claims_recoveries_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':
+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.session_state['current_page'] != 'home':
     if st.button("Back to Home"):
-        st.session_state.current_page = 'home'
+        st.session_state['current_page'] = 'home'