update app

app.py

@@ -812,32 +812,46 @@ def _auto_suggest_datasets(task_description: str, benchmark_description: str) ->
     return [[d] for d in ds_ids]
 
 
+def _rows(table) -> List[List[str]]:
+    """Normalize Gradio dataframe output (which may be pandas DataFrame) to list of rows."""
+    if table is None:
+        return []
+    if hasattr(table, "to_numpy"):
+        try:
+            return table.to_numpy().tolist()
+        except Exception:
+            return []
+    return table or []
+
+
 def _add_selected_suggestion(
-    current: List[List[str]],
-    suggestions: List[List[str]],
+    current,
+    suggestions,
     idx: Optional[int],
 ) -> List[List[str]]:
-    if idx is None or suggestions is None:
-        return current
-    if idx < 0 or idx >= len(suggestions):
-        return current
-    val = str(suggestions[idx][0]).strip()
+    rows_current = _rows(current)
+    rows_suggestions = _rows(suggestions)
+    if idx is None or idx < 0 or idx >= len(rows_suggestions):
+        return rows_current
+    val = str(rows_suggestions[idx][0]).strip() if rows_suggestions[idx] else ""
     if not val:
-        return current
-    existing = [str(r[0]).strip() for r in (current or []) if r]
+        return rows_current
+    existing = [str(r[0]).strip() for r in rows_current if r and len(r) > 0]
     if val in existing:
-        return current
-    return (current or []) + [[val]]
+        return rows_current
+    return rows_current + [[val]]
 
 
 def _add_all_suggestions(
-    current: List[List[str]],
-    suggestions: List[List[str]],
+    current,
+    suggestions,
 ) -> List[List[str]]:
-    existing = [str(r[0]).strip() for r in (current or []) if r]
-    merged = list(current or [])
-    for row in suggestions or []:
-        val = str(row[0]).strip()
+    rows_current = _rows(current)
+    rows_suggestions = _rows(suggestions)
+    existing = [str(r[0]).strip() for r in rows_current if r and len(r) > 0]
+    merged = list(rows_current)
+    for row in rows_suggestions:
+        val = str(row[0]).strip() if row else ""
         if val and val not in existing:
             existing.append(val)
             merged.append([val])