Update app.py

app.py

@@ -135,8 +135,6 @@ with col3:
 
 
 
-
-
 # Initialize session state variables
 st.session_state.setdefault('current_page', 'home')
 st.session_state.setdefault('premiums_df', None)
@@ -151,25 +149,9 @@ def is_mostly_numeric(s):
         return percent_numeric > 0.5
     return False
 
-# # Callback functions for processing each uploaded file
-# def process_premiums_file(uploaded_file):
-#     df = pd.read_csv(uploaded_file)
-#     st.session_state['premiums_df'] = df
-#     st.session_state.active_page = 'premiums_selection'
-
-# def process_claims_recoveries_file(uploaded_file):
-#     df = pd.read_csv(uploaded_file)
-#     st.session_state['claims_recoveries_df'] = df
-#     st.session_state.active_page = 'claims_recoveries_selection'
-
-# def process_claims_count_file(uploaded_file):
-#     df = pd.read_csv(uploaded_file)
-#     st.session_state['claims_count_df'] = df
-#     st.session_state.active_page = 'claims_count_selection'
-    
 def process_file(uploaded_file, key):
     df = pd.read_csv(uploaded_file)
-    st.session_state[key] = df
+    st.session_state[key + '_df'] = df
     st.session_state['current_page'] = key + '_selection'
 
 # Function to display the column selection UI
@@ -197,40 +179,32 @@ def column_selection_ui(dataframe_key, selection_key):
 
     # Filter the DataFrame based on selected 'lob' values
     if selected_lob_values:
-        filtered_df = st.session_state['premiums_df'][st.session_state['premiums_df'][lob_column].isin(selected_lob_values)]
-        st.session_state['filtered_premiums_df'] = filtered_df
+        filtered_df = st.session_state[dataframe_key][st.session_state[dataframe_key][lob_column].isin(selected_lob_values)]
+        st.session_state['filtered_' + dataframe_key] = filtered_df
 
         # Display other column selections after 'lob' has been selected and DataFrame has been filtered
         st.write("Now, select the columns for other parameters:")
-        valuation_as_at = st.selectbox('Valuation As At:', filtered_df.columns, key='valuation_as_at')
-        quarter_bracket = st.selectbox('Quarter Bracket:', filtered_df.columns, key='quarter_bracket')
+        valuation_as_at = st.selectbox('Valuation As At:', filtered_df.columns, key='valuation_as_at_' + selection_key)
+        quarter_bracket = st.selectbox('Quarter Bracket:', filtered_df.columns, key='quarter_bracket_' + selection_key)
         # Continue with other selections as required...
 
         # Store the column selections in the session state
-        st.session_state['column_selections'] = {
+        st.session_state[f'{selection_key}_selections'] = {
+            'lob_column': lob_column,
+            'selected_lob_values': selected_lob_values,
             'valuation_as_at': valuation_as_at,
             'quarter_bracket': quarter_bracket,
-            # ... include other selections as well
         }
 
-    # Store the column selections in the session state
-    st.session_state[f'{selection_key}_selections'] = {
-        'lob_column': lob_column,
-        'selected_lob_values': selected_lob_values,
-        'valuation_as_at': valuation_as_at,
-        'quarter_bracket': quarter_bracket,
-    }
-
-    
 # File uploaders and buttons
 if st.session_state.current_page == 'home':
     upload_col1, upload_col2, upload_col3 = st.columns(3)
     with upload_col1:
-        uploaded_file_premiums = st.file_uploader("Upload Premiums Data", key="premiums")
+        uploaded_file_premiums = st.file_uploader("Upload Premiums Data", key="premiums_file")
     with upload_col2:
-        uploaded_file_claims_recoveries = st.file_uploader("Upload Claims Recoveries Data", key="claims_recoveries")
+        uploaded_file_claims_recoveries = st.file_uploader("Upload Claims Recoveries Data", key="claims_recoveries_file")
     with upload_col3:
-        uploaded_file_claims_count = st.file_uploader("Upload Claims Count Data", key="claims_count")
+        uploaded_file_claims_count = st.file_uploader("Upload Claims Count Data", key="claims_count_file")
 
     button_col1, button_col2, button_col3 = st.columns(3)
     with button_col1:
@@ -248,10 +222,10 @@ if st.session_state.current_page == 'home':
 
 # Conditional rendering based on the current page
 if st.session_state.current_page.endswith('_selection'):
-    dataframe_key = st.session_state.current_page.replace('_selection', '')
-    column_selection_ui(dataframe_key, dataframe_key)
+    dataframe_key = st.session_state.current_page.replace('_selection', '_df')
+    column_selection_ui(dataframe_key, dataframe_key.replace('_df', ''))
 
 # Back to Home button
 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'