Update app.py

app.py

@@ -70,12 +70,24 @@ with button_col3:
 if st.session_state.get('select_columns', False):
     st.write("Please choose the 'lob' column and the values to filter by:")
 
+    # Create two columns: one for the 'lob' column dropdown, one for the checkboxes
+    col1, col2 = st.columns(2)
+    
     # Dropdown to select the 'lob' column
-    lob_column = st.selectbox('Choose the column for "lob":', st.session_state['premiums_df'].columns)
+    with col1:
+        lob_column = st.selectbox('Choose the column for "lob":', st.session_state['premiums_df'].columns, key='lob_column')
+
+    # Initialize an empty list to hold the selected values
+    selected_lob_values = []
 
-    # Multiselect to choose values from the 'lob' column to filter the DataFrame
-    selected_lob_values = st.multiselect('Choose values from the "lob" column:', 
-                                         st.session_state['premiums_df'][lob_column].unique())
+    # Generate checkboxes for each unique value in the 'lob' column
+    with col2:
+        st.write("Select values for filtering:")
+        if 'lob_column' in st.session_state and st.session_state['lob_column'] in st.session_state['premiums_df'].columns:
+            for value in st.session_state['premiums_df'][st.session_state['lob_column']].unique():
+                # 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:
@@ -84,8 +96,8 @@ if st.session_state.get('select_columns', False):
 
         # 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)
-        quarter_bracket = st.selectbox('Quarter Bracket:', filtered_df.columns)
+        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