multi select models

src/streamlit_app.py

@@ -414,16 +414,49 @@ def main():
     # Filter data
     filtered_df = df[df['dataset'] == selected_dataset]
     
-    # Model filter (optional)
-    models = ['All'] + sorted(filtered_df['model'].unique())
-    selected_model = st.sidebar.selectbox("Select Model", models)
+    # Model filter - changed to multiselect for selective visualization
+    st.sidebar.subheader("Select Models to Display")
+    all_models = sorted(filtered_df['model'].unique())
     
-    if selected_model != 'All':
-        filtered_df = filtered_df[filtered_df['model'] == selected_model]
+    # Add "Select All" / "Deselect All" buttons
+    col1, col2 = st.sidebar.columns(2)
+    with col1:
+        if st.button("Select All", key="select_all"):
+            st.session_state['selected_models'] = all_models
+    with col2:
+        if st.button("Deselect All", key="deselect_all"):
+            st.session_state['selected_models'] = []
+    
+    # Initialize selected models if not in session state
+    if 'selected_models' not in st.session_state:
+        st.session_state['selected_models'] = all_models
+    
+    # Multi-select widget for models
+    selected_models = st.sidebar.multiselect(
+        "Models to visualize:",
+        options=all_models,
+        default=st.session_state.get('selected_models', all_models),
+        key='model_multiselect'
+    )
+    
+    # Update session state
+    st.session_state['selected_models'] = selected_models
+    
+    # Filter dataframe based on selected models
+    if selected_models:
+        filtered_df = filtered_df[filtered_df['model'].isin(selected_models)]
+    else:
+        # If no models selected, show empty dataframe
+        filtered_df = pd.DataFrame()
     
     # Main content
     st.header(f"Results for {selected_dataset}")
     
+    # Check if any models are selected
+    if filtered_df.empty:
+        st.warning("No models selected. Please select at least one model from the sidebar to visualize results.")
+        return
+    
     # Overall metrics
     col1, col2, col3 = st.columns(3)
     with col1: