Update app.py

app.py

@@ -30,88 +30,168 @@ with col2:
 with col3:
     header("Claims Count",'h3')
 
-# Second row with three file upload buttons
-upload_col1, upload_col2, upload_col3 = st.columns(3)
-with upload_col1:
-    uploaded_file_premiums = st.file_uploader("", key="premiums")
-with upload_col2:
-    uploaded_file_claims_recoveries = st.file_uploader("", key="claims_recoveries")
-with upload_col3:
-    uploaded_file_claims_count = st.file_uploader("", key="claims_count")
+# # Second row with three file upload buttons
+# upload_col1, upload_col2, upload_col3 = st.columns(3)
+# with upload_col1:
+#     uploaded_file_premiums = st.file_uploader("", key="premiums")
+# with upload_col2:
+#     uploaded_file_claims_recoveries = st.file_uploader("", key="claims_recoveries")
+# with upload_col3:
+#     uploaded_file_claims_count = st.file_uploader("", key="claims_count")
+
+# # Third row with three buttons, enabled only if the corresponding file is uploaded
+# 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"
+
+# def process_premiums_file(uploaded_file):
+#     # Read the CSV file into a DataFrame
+#     df = pd.read_csv(uploaded_file)
+    
+#     # Store the DataFrame in the session state
+#     st.session_state['premiums_df'] = df
+
+#     # Indicate that the CSV has been processed and we're ready to select columns
+#     st.session_state['select_columns'] = True
+
+# # Button to select premiums, only displayed and active if a file has been uploaded
+# with button_col1:
+#     # Check if the file has been uploaded
+#     file_uploaded = uploaded_file_premiums is not None
+#     # The button is enabled if a file has been uploaded, and disabled otherwise
+#     if st.button(premiumlabel, disabled=not file_uploaded, key="premiums_button"):
+#         # If the button is clicked (active), call the callback function
+#         process_premiums_file(uploaded_file_premiums)
+
+# with button_col2:
+#     if uploaded_file_claims_recoveries is not None:
+#         st.button(incurredlabel)
+#         st.button(recoverlabel)
+#     else:
+#         st.button(incurredlabel, disabled=True)
+#         st.button(recoverlabel, disabled=True)
+
+# with button_col3:
+#     if uploaded_file_claims_count is not None:
+#         st.button(claimscount)
+#     else:
+#         st.button(claimscount, disabled=True)
 
-# Third row with three buttons, enabled only if the corresponding file is uploaded
-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"
 
+# # If the CSV has been processed, display the column selection UI
+# if st.session_state.get('select_columns', False):
+#     st.empty()
+#     st.write("Please choose the 'lob' column and the values to filter by:")
+
+#     # Determine columns with only string values and where no value is mostly numeric
+#     string_columns = [
+#         col for col in st.session_state['premiums_df'].columns 
+#         if not any(not isinstance(x, str) or is_mostly_numeric(x) for x in st.session_state['premiums_df'][col].dropna())
+#     ]
+
+#     # Create two columns: one for the 'lob' column dropdown, one for the checkboxes
+#     col1, col2 = st.columns(2)
+    
+#     # Dropdown to select the 'lob' column, filtered to only string columns
+#     with col1:
+#         lob_column = st.selectbox('Choose the column for "lob":', string_columns, key='lob_column')
+#     # Initialize an empty list to hold the selected values
+#     selected_lob_values = []
+
+#     # Generate checkboxes for each unique value in the 'lob' column
+#     with col2:
+#         st.write("Select values for filtering:")
+#         # Ensure that 'lob_column' is in the session state and it's in the DataFrame columns
+#         if 'lob_column' in st.session_state and st.session_state['lob_column'] in st.session_state['premiums_df'].columns:
+#             # Retrieve unique values from the 'lob' column
+#             unique_values = st.session_state['premiums_df'][st.session_state['lob_column']].unique()
+#             # Sort the values for consistent checkbox ordering
+#             unique_values.sort()
+#             for value in unique_values:
+#                 # Create a checkbox for each unique value in the 'lob' column
+#                 if st.checkbox(f"{value}", key=f"checkbox_{value}"):
+#                     selected_lob_values.append(value)
+
+#     # 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
+
+#         # 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')
+#         # Continue with other selections as required...
+
+#         # Store the column selections in the session state
+#         st.session_state['column_selections'] = {
+#             'valuation_as_at': valuation_as_at,
+#             'quarter_bracket': quarter_bracket,
+#             # ... include other selections as well
+#         }
+
+
+
+
+
+
+# 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
+
+# Helper function to check if a string has more than 50% numeric characters
+def is_mostly_numeric(s):
+    if isinstance(s, str):
+        num_numeric = sum(c.isdigit() for c in s)
+        percent_numeric = num_numeric / len(s) if s else 0
+        return percent_numeric > 0.5
+    return False
+
+# Callback functions for processing each uploaded file
 def process_premiums_file(uploaded_file):
-    # Read the CSV file into a DataFrame
     df = pd.read_csv(uploaded_file)
-    
-    # Store the DataFrame in the session state
     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'
+
+# Function to display the column selection UI
+def column_selection_ui(dataframe_key, selection_key):
+    st.write(f"Please choose the 'lob' column and the values to filter by for {selection_key}:")
 
-    # Indicate that the CSV has been processed and we're ready to select columns
-    st.session_state['select_columns'] = True
-
-# Button to select premiums, only displayed and active if a file has been uploaded
-with button_col1:
-    # Check if the file has been uploaded
-    file_uploaded = uploaded_file_premiums is not None
-    # The button is enabled if a file has been uploaded, and disabled otherwise
-    if st.button(premiumlabel, disabled=not file_uploaded, key="premiums_button"):
-        # If the button is clicked (active), call the callback function
-        process_premiums_file(uploaded_file_premiums)
-
-with button_col2:
-    if uploaded_file_claims_recoveries is not None:
-        st.button(incurredlabel)
-        st.button(recoverlabel)
-    else:
-        st.button(incurredlabel, disabled=True)
-        st.button(recoverlabel, disabled=True)
-
-with button_col3:
-    if uploaded_file_claims_count is not None:
-        st.button(claimscount)
-    else:
-        st.button(claimscount, disabled=True)
-
-
-# If the CSV has been processed, display the column selection UI
-if st.session_state.get('select_columns', False):
-    st.empty()
-    st.write("Please choose the 'lob' column and the values to filter by:")
-
-    # Determine columns with only string values and where no value is mostly numeric
     string_columns = [
-        col for col in st.session_state['premiums_df'].columns 
-        if not any(not isinstance(x, str) or is_mostly_numeric(x) for x in st.session_state['premiums_df'][col].dropna())
+        col for col in st.session_state[dataframe_key].columns 
+        if not any(not isinstance(x, str) or is_mostly_numeric(x) for x in st.session_state[dataframe_key][col].dropna())
     ]
 
-    # Create two columns: one for the 'lob' column dropdown, one for the checkboxes
     col1, col2 = st.columns(2)
-    
-    # Dropdown to select the 'lob' column, filtered to only string columns
     with col1:
-        lob_column = st.selectbox('Choose the column for "lob":', string_columns, key='lob_column')
-    # Initialize an empty list to hold the selected values
+        lob_column = st.selectbox('Choose the column for "lob":', string_columns, key=f'lob_column_{selection_key}')
     selected_lob_values = []
 
-    # Generate checkboxes for each unique value in the 'lob' column
     with col2:
         st.write("Select values for filtering:")
-        # Ensure that 'lob_column' is in the session state and it's in the DataFrame columns
-        if 'lob_column' in st.session_state and st.session_state['lob_column'] in st.session_state['premiums_df'].columns:
-            # Retrieve unique values from the 'lob' column
-            unique_values = st.session_state['premiums_df'][st.session_state['lob_column']].unique()
-            # Sort the values for consistent checkbox ordering
+        if lob_column in st.session_state[dataframe_key].columns:
+            unique_values = st.session_state[dataframe_key][lob_column].unique()
             unique_values.sort()
             for value in unique_values:
-                # Create a checkbox for each unique value in the 'lob' column
-                if st.checkbox(f"{value}", key=f"checkbox_{value}"):
+                if st.checkbox(f"{value}", key=f"checkbox_{value}_{selection_key}"):
                     selected_lob_values.append(value)
 
     # Filter the DataFrame based on selected 'lob' values
@@ -130,4 +210,47 @@ if st.session_state.get('select_columns', False):
             'valuation_as_at': valuation_as_at,
             'quarter_bracket': quarter_bracket,
             # ... include other selections as well
-        }
+        }
+
+
+# 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")
+with upload_col2:
+    uploaded_file_claims_recoveries = st.file_uploader("Upload Claims Recoveries Data", key="claims_recoveries")
+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:
+    with button_col1:
+        if uploaded_file_premiums is not None:
+            if st.button(premiumlabel):
+                process_premiums_file(uploaded_file_premiums)
+
+    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
+
+    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')