Add search

app.py

@@ -47,6 +47,17 @@ def refresh():
     return get_leaderboard_df()
 
 
+# Function to update the table based on search query
+def update_table(search_query):
+    df = get_leaderboard_df()
+    if search_query:
+        search_terms = search_query.split(";")
+        search_terms = [term.strip() for term in search_terms]
+        pattern = "|".join(search_terms)
+        df = df[df["Model"].str.contains(pattern, regex=True)]
+    return df
+
+
 leaderboard_df = get_leaderboard_df()
 
 demo = gr.Blocks()
@@ -55,10 +66,14 @@ with demo:
     gr.HTML(TITLE)
     with gr.Column():
         gr.Markdown(DESCRIPTION, elem_classes="markdown-text")
+        with gr.Row():
+            search_bar = gr.Textbox(placeholder="Search for your model...", show_label=False)
         with gr.Group():
             leaderboard_table = gr.Dataframe(value=leaderboard_df, wrap=True, height=1000)
         with gr.Row():
             refresh_button = gr.Button("Refresh")
-            refresh_button.click(refresh, inputs=[], outputs=[leaderboard_table])
+
+    search_bar.submit(update_table, inputs=[search_bar], outputs=[leaderboard_table])
+    refresh_button.click(refresh, inputs=[], outputs=[leaderboard_table])
 
 demo.launch()