debug

app.py

@@ -131,28 +131,40 @@ from dataclasses import fields
 
 
 
+from gradio_leaderboard import SelectColumns, Leaderboard
+import pandas as pd
+from typing import List, Dict, Optional
+from dataclasses import fields
+
 class SmartSelectColumns(SelectColumns):
     """
-    Enhanced SelectColumns component following the gradio_leaderboard docs pattern.
+    Enhanced SelectColumns component matching exact original parameters.
     """
     def __init__(
         self,
         benchmark_keywords: Optional[List[str]] = None,
         model_keywords: Optional[List[str]] = None,
         initial_selected: Optional[List[str]] = None,
-        label: str = "Select Results:",
-        **kwargs
+        label: Optional[str] = None,
+        show_label: bool = True,
+        info: Optional[str] = None,
+        allow: bool = True
     ):
-        # Initialize with basic SelectColumns parameters
-        default_selection = initial_selected or []
+        # Match exact parameters from working SelectColumns
         super().__init__(
-            default_selection=default_selection,
+            default_selection=initial_selected or [],
+            cant_deselect=[],
+            allow=allow,
             label=label,
-            **kwargs
+            show_label=show_label,
+            info=info
         )
         
         self.benchmark_keywords = benchmark_keywords or []
         self.model_keywords = model_keywords or []
+        
+        # Store groups for later use
+        self._groups = {}
     
     def get_filtered_groups(self, columns: List[str]) -> Dict[str, List[str]]:
         """Get column groups based on keywords."""
@@ -176,6 +188,7 @@ class SmartSelectColumns(SelectColumns):
             if matching_cols:
                 filtered_groups[f"Model group for {model}"] = matching_cols
                 
+        self._groups = filtered_groups
         return filtered_groups
 
 
@@ -187,6 +200,7 @@ class SmartSelectColumns(SelectColumns):
 
 
 
+
 def restart_space():
     API.restart_space(repo_id=REPO_ID)
 
@@ -457,30 +471,36 @@ def init_leaderboard_mib_subgraph(dataframe, track):
     benchmark_keywords = ["ioi", "mcqa", "arithmetic", "arc"]
     model_keywords = ["qwen2_5", "gpt2", "gemma2", "llama3"]
 
-    # Create SmartSelectColumns instance
+    # Create SmartSelectColumns instance with exact same parameters as working version
     smart_columns = SmartSelectColumns(
         benchmark_keywords=benchmark_keywords,
         model_keywords=model_keywords,
         initial_selected=["Method", "Average"],
-        label="Select Results:",
-        cant_deselect=["Method"]  # Ensure Method column is always shown
+        allow=True,
+        label=None,
+        show_label=True,
+        info=None
     )
 
-
-    # Create Leaderboard
-    print("\nCreating leaderboard...")
-    leaderboard = Leaderboard(
-        value=dataframe,
-        datatype=[c.type for c in fields(AutoEvalColumn_mib_subgraph)],
-        select_columns=smart_columns,
-        search_columns=["Method"],
-        hide_columns=[],
-        interactive=False
-    )
-    
-    working_select = SelectColumns(default_selection=["Method", "Average"])
-    print("working_select is ", working_select)
-    return leaderboard
+    try:
+        print("\nCreating leaderboard...")
+        # Get groups before creating leaderboard
+        smart_columns.get_filtered_groups(dataframe.columns)
+        
+        leaderboard = Leaderboard(
+            value=dataframe,
+            datatype=[c.type for c in fields(AutoEvalColumn_mib_subgraph)],
+            select_columns=smart_columns,
+            search_columns=["Method"],
+            hide_columns=[],
+            interactive=False
+        )
+        print("Leaderboard created successfully")
+        return leaderboard
+        
+    except Exception as e:
+        print("Error creating leaderboard:", str(e))
+        raise