Updating task selection to checkbox

src/components/filters.py

@@ -24,6 +24,10 @@ def initialize_session_state(df):
         # Make sure selected_tasks only includes actual tasks from the dataframe
         valid_tasks = [col for col in df.columns if col not in ['Model Type']]
         st.session_state.selected_tasks = [task for task in st.session_state.selected_tasks if task in valid_tasks]
+        
+        # If no tasks are selected, select all tasks
+        if not st.session_state.selected_tasks:
+            st.session_state.selected_tasks = valid_tasks
     
     # Always select all model types
     st.session_state.selected_model_types = list(df['Model Type'].unique())
@@ -73,7 +77,7 @@ def render_metric_selection():
 
 def render_task_selection(df):
     """
-    Render the task selection component
+    Render the task selection component with checkboxes
     
     Args:
         df (pandas.DataFrame): The DataFrame with model data
@@ -86,43 +90,44 @@ def render_task_selection(df):
     # Extract task columns (exclude Model Type and Overall)
     all_tasks = [col for col in df.columns if col not in ['Model Type']]
     
-    # Determine number of columns based on screen width
-    num_cols = 3  # Default for medium screens
+    # Increase number of columns to reduce spacing
+    num_cols = 4  # More columns for tighter spacing
     
-    # Create task buttons in a fixed number of columns with balanced width
+    # Create task groups in a fixed number of columns with balanced width
     task_groups = [all_tasks[i:i+num_cols] for i in range(0, len(all_tasks), num_cols)]
     
-    # Custom CSS for button styling
+    # Clear selected tasks if it's empty or not initialized
+    if 'selected_tasks' not in st.session_state or not st.session_state.selected_tasks:
+        st.session_state.selected_tasks = []
+    
+    # Add custom CSS to reduce spacing between checkboxes
     st.markdown("""
     <style>
-    /* Make buttons same width in their columns and centered */
-    .stButton > button {
-        width: 100%;
-        max-width: 300px;
-        margin: 0 auto;
-        display: block;
+    /* Reduce spacing in checkbox containers */
+    .stCheckbox {
+        padding: 0px !important;
+        margin-bottom: 5px !important;
     }
     </style>
     """, unsafe_allow_html=True)
     
-    # Display buttons in rows
+    # Display checkboxes in rows
     for group in task_groups:
         # Create columns with equal width
         cols = st.columns(num_cols)
         
-        # Add buttons to each column
+        # Add checkboxes to each column
         for i, task in enumerate(group):
             if i < len(cols):  # Ensure we don't exceed available columns
                 with cols[i]:
                     is_selected = task in st.session_state.selected_tasks
-                    button_label = f"✓ {task}" if is_selected else task
-                    button_type = "primary" if is_selected else "secondary"
                     
-                    if st.button(button_label, key=f"task_{task}", type=button_type):
-                        if is_selected:
-                            st.session_state.selected_tasks.remove(task)
-                        else:
+                    # Use checkbox instead of button
+                    if st.checkbox(task, value=is_selected, key=f"task_checkbox_{task}"):
+                        if task not in st.session_state.selected_tasks:
                             st.session_state.selected_tasks.append(task)
-                        st.rerun()  # Force UI update
+                    else:
+                        if task in st.session_state.selected_tasks:
+                            st.session_state.selected_tasks.remove(task)
     
     return st.session_state.selected_tasks