Update app.py

app.py

@@ -157,6 +157,7 @@ def load_from_dir(
     selected_frameworks: List[str] | None = None,
     selected_model_types: List[str] | None = None,
     selected_precisions: List[str] | None = None,
+    search_keyword: str = "",
     force_refresh: bool = False,
 ):
     try:
@@ -209,6 +210,13 @@ def load_from_dir(
         lower_selected = [str(x).lower() for x in selected_precisions]
         df = df[df["Precision"].astype(str).str.lower().isin(lower_selected)]
 
+    # Search keyword filter - search across all columns
+    if search_keyword and search_keyword.strip():
+        keyword_lower = search_keyword.strip().lower()
+        # Create a mask that checks if the keyword appears in any column
+        mask = df.astype(str).apply(lambda row: row.str.lower().str.contains(keyword_lower).any(), axis=1)
+        df = df[mask]
+
     if df.empty:
         empty_html = "<p>No records found.</p>"
         return empty_html
@@ -246,6 +254,7 @@ def auto_refresh_from_dir(
     selected_frameworks: List[str] | None = None,
     selected_model_types: List[str] | None = None,
     selected_precisions: List[str] | None = None,
+    search_keyword: str = "",
 ):
     return load_from_dir(
         dir_path,
@@ -253,6 +262,7 @@ def auto_refresh_from_dir(
         selected_frameworks=selected_frameworks,
         selected_model_types=selected_model_types,
         selected_precisions=selected_precisions,
+        search_keyword=search_keyword,
         force_refresh=True,
     )
 
@@ -280,12 +290,13 @@ def build_app() -> gr.Blocks:
         white-space: nowrap;
         font-size: 13px;
         text-align: left;
+        color: #000000 !important;
     }
     
     .gradio-container table.metrics-table th {
         background: linear-gradient(to bottom, #f9fafb, #f3f4f6);
         font-weight: 600;
-        color: #374151;
+        color: #000000 !important;
         position: sticky;
         top: 0;
         z-index: 10;
@@ -307,12 +318,12 @@ def build_app() -> gr.Blocks:
     }
     
     .gradio-container table.metrics-table a {
-        color: #2563eb;
+        color: #000000 !important;
         text-decoration: none;
     }
     
     .gradio-container table.metrics-table a:hover {
-        color: #1d4ed8;
+        color: #000000 !important;
         text-decoration: underline;
     }
     
@@ -336,16 +347,39 @@ def build_app() -> gr.Blocks:
         box-shadow: 0 1px 3px rgba(0, 0, 0, 0.1);
     }
     
+    /* Accordion styling */
+    .gradio-container .accordion {
+        background: white !important;
+    }
+    
     /* Info section */
     .info-section {
         max-height: 400px;
         overflow-y: auto;
         padding-right: 10px;
+        background: white !important;
+    }
+    
+    /* Override any dark backgrounds in groups and accordions */
+    .gradio-container .group, 
+    .gradio-container .accordion,
+    .gradio-container .panel {
+        background-color: white !important;
+    }
+    
+    /* Search box styling */
+    .search-box {
+        background: white !important;
+        padding: 16px;
+        border-radius: 8px;
+        border: 1px solid #d1d5db;
+        box-shadow: 0 1px 3px rgba(0, 0, 0, 0.1);
+        margin-bottom: 16px;
     }
     
     /* Heading styling */
     .gradio-container h1 {
-        color: #111827;
+        color: #000000 !important;
         font-weight: 700;
         margin-bottom: 24px;
     }
@@ -364,6 +398,13 @@ def build_app() -> gr.Blocks:
         with gr.Row():
             # Left side - Filters (narrower)
             with gr.Column(scale=2):
+                with gr.Group(elem_classes="search-box"):
+                    search_input = gr.Textbox(
+                        label="🔍 Search",
+                        placeholder="Search across all columns...",
+                        lines=1
+                    )
+                
                 with gr.Group(elem_classes="filter-section"):
                     gr.Markdown("### 🎛️ Filters")
                     
@@ -429,35 +470,41 @@ def build_app() -> gr.Blocks:
 
         demo.load(
             fn=auto_refresh_from_dir,
-            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter],
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
             outputs=[leaderboard_output],
         )
 
+        search_input.change(
+            fn=load_from_dir,
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
+            outputs=[leaderboard_output],
+        )
+        
         task_filter.change(
             fn=load_from_dir,
-            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter],
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
             outputs=[leaderboard_output],
         )
         framework_filter.change(
             fn=load_from_dir,
-            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter],
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
             outputs=[leaderboard_output],
         )
         model_type_filter.change(
             fn=load_from_dir,
-            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter],
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
             outputs=[leaderboard_output],
         )
         precision_filter.change(
             fn=load_from_dir,
-            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter],
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
             outputs=[leaderboard_output],
         )
 
         timer = gr.Timer(60.0)
         timer.tick(
             fn=auto_refresh_from_dir,
-            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter],
+            inputs=[dir_path, task_filter, framework_filter, model_type_filter, precision_filter, search_input],
             outputs=[leaderboard_output],
         )