[data] fixing code

app.py

@@ -18,6 +18,7 @@ from src.display.css_html_js import custom_css
 
 # Load with multi-header and set the first column ('Model') as index
 LEADERBOARD_DF = pd.read_csv("leaderboard_data.csv", header=[0, 1], index_col=0)
+LEADERBOARD_DF.index.name = "Model" # Explicitly name the index
 # No need to astype(str) globally here, Gradio DataFrame handles it well.
 # If specific styling or type issues arise, it can be done selectively.
 
@@ -52,12 +53,16 @@ with demo:
                 )
 
                 def update_table(selected_models_from_filter):
-                    # Filter based on index (Model names)
-                    if selected_models_from_filter:
-                        filtered_df = LEADERBOARD_DF.loc[LEADERBOARD_DF.index.isin(selected_models_from_filter)]
+                    if not selected_models_from_filter:
+                        # Return an empty DataFrame with the same index and columns
+                        filtered_df = LEADERBOARD_DF.iloc[0:0]
                     else:
-                        # If no models are selected, show an empty DataFrame with the same columns
-                        filtered_df = pd.DataFrame(columns=LEADERBOARD_DF.columns)
+                        # Filter out any selected models not actually in the index to prevent errors
+                        valid_selected_models = [model for model in selected_models_from_filter if model in LEADERBOARD_DF.index]
+                        if not valid_selected_models:
+                             filtered_df = LEADERBOARD_DF.iloc[0:0]
+                        else:
+                             filtered_df = LEADERBOARD_DF.loc[valid_selected_models]
                     return gr.DataFrame.update(value=filtered_df)
 
                 # Event listener (only model_selector now)