Render board correctly for model link

app.py

@@ -217,41 +217,51 @@ def init_leaderboard(df):
     else:
         logger.info(f"Initializing leaderboard with {len(df)} rows")
 
-    # Ensure "Security Score ⬆️" and "Safetensors" columns are present
-    if "Security Score ⬆️" not in df.columns:
-        logger.warning("Security Score column not found in DataFrame")
-        df["Security Score ⬆️"] = None
-    if "Safetensors" not in df.columns:
-        logger.warning("Safetensors column not found in DataFrame")
-        df["Safetensors"] = None
+    # Ensure all required columns exist
+    for col in COLS:
+        if col not in df.columns:
+            logger.warning(f"Column {col} not found in DataFrame, adding with None values")
+            df[col] = None
+
+    # Map dataset columns to display columns
+    column_mapping = {
+        "model_id": "Model",
+        "security_score": "Security Score ⬆️",
+        "safetensors_compliant": "Safetensors",
+        "precision": "Precision"
+    }
+    
+    for src, dst in column_mapping.items():
+        if src in df.columns:
+            df[dst] = df[src]
+            logger.debug(f"Mapped column {src} to {dst}")
 
     # Sort by Security Score if available
     if "Security Score ⬆️" in df.columns:
         df = df.sort_values(by="Security Score ⬆️", ascending=False)
         logger.info("Sorted leaderboard by Security Score")
 
-    # Ensure all required columns are present
-    for col in COLS:
-        if col not in df.columns:
-            logger.warning(f"Column {col} not found in DataFrame, adding with None values")
-            df[col] = None
-
     # Select only the columns we want to display
     df = df[COLS]
 
     logger.info(f"Final leaderboard columns: {df.columns.tolist()}")
     logger.debug(f"Leaderboard data:\n{df}")
 
-    # Create the leaderboard
-    return gr.Dataframe(
-        headers=COLS,
-        datatype=["str"] * len(COLS),
-        row_count=10,
-        col_count=(len(COLS), "fixed"),
+    # Create the leaderboard using gradio_leaderboard
+    return Leaderboard(
         value=df,
-        wrap=True,
-        column_widths=[50] + [None] * (len(COLS) - 1),
-        type="pandas",
+        datatype=["html" if col == "Model" else "number" if col == "Security Score ⬆️" else "bool" if col == "Safetensors" else "str" for col in COLS],
+        select_columns=SelectColumns(
+            default_selection=COLS,
+            cant_deselect=["Model", "Security Score ⬆️", "Safetensors"],
+            label="Select Columns to Display:",
+        ),
+        search_columns=["Model"],
+        filter_columns=[
+            ColumnFilter("Safetensors", type="boolean", label="Show only Safetensors models"),
+            ColumnFilter("Security Score ⬆️", type="slider", min=0, max=1, step=0.1, label="Minimum Security Score"),
+        ],
+        interactive=False,
     )