Update app.py

app.py

@@ -70,12 +70,16 @@ with button_col3:
 if st.session_state.get('select_columns', False):
     st.write("Please choose the 'lob' column and the values to filter by:")
 
+    # Determine columns with only string values
+    string_columns = [col for col in st.session_state['premiums_df'].columns 
+                      if st.session_state['premiums_df'][col].dropna().apply(lambda x: isinstance(x, str)).all()]
+
     # Create two columns: one for the 'lob' column dropdown, one for the checkboxes
     col1, col2 = st.columns(2)
     
-    # Dropdown to select the 'lob' column
+    # Dropdown to select the 'lob' column, filtered to only string columns
     with col1:
-        lob_column = st.selectbox('Choose the column for "lob":', st.session_state['premiums_df'].columns, key='lob_column')
+        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 = []
@@ -83,8 +87,13 @@ if st.session_state.get('select_columns', False):
     # 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:
-            for value in st.session_state['premiums_df'][st.session_state['lob_column']].unique():
+            # 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)