First pass at table filtering

app.py

@@ -1,10 +1,20 @@
 import os
+import pandas as pd
 import requests
 import huggingface_hub
 import gradio as gr
 
+data = pd.read_csv("data.csv")
 webhook_url = os.environ.get("WEBHOOK_URL")
 
+def filter_table(name, type, arch, lcns):
+    tmp = data
+    tmp = tmp[tmp["Name"].str.contains(name)]
+    tmp = tmp[tmp["Type"].isin(type)]
+    tmp = tmp[tmp["Architecture"].isin(arch)]
+    tmp = tmp[tmp["License"].isin(lcns)]
+    return tmp
+
 def submit_model(name):
     try:
         huggingface_hub.hf_hub_download(repo_id=name, filename="config.json") # sanity check input
@@ -33,8 +43,16 @@ with gr.Blocks() as demo:
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.Tab("🏅 LLM Benchmark"):
-            gr.Markdown("Table filters under construction")
-            gr.Dataframe("data.csv")
+            with gr.Row():
+                namefilter = model_name = gr.Textbox(max_lines=1, placeholder="Search by model name...", show_label=False)
+                typefilter = gr.Dropdown(label="Filter by model type", multiselect=True, choices=list(set(data["Type"])), value=list(set(data["Type"])))
+                archfilter = gr.Dropdown(label="Filter by model architecture", multiselect=True, choices=list(set(data["Architecture"])), value=list(set(data["Architecture"])))
+                lcnsfilter = gr.Dropdown(label="Filter by model license", multiselect=True, choices=list(set(data["License"])), value=list(set(data["License"])))
+                filter = gr.Button("Filter")
+                
+            table = gr.Dataframe(data)
+
+            filter.click(fn=filter_table, inputs=[namefilter,typefilter,archfilter,lcnsfilter], outputs=table)
         
         with gr.Tab("📝 About"):
             gr.Markdown("""
@@ -50,6 +68,7 @@ with gr.Blocks() as demo:
                     submit = gr.Button("Submit", variant="primary", scale=0)
             
             output = gr.Markdown("Enter a public HF repo id, then hit Submit to add it to the evaluation queue.")
+            
             submit.click(fn=submit_model, inputs=model_name, outputs=output)