Revert UI changes

app.py

@@ -7,15 +7,23 @@ import gradio as gr
 data = pd.read_csv("data.csv")
 webhook_url = os.environ.get("WEBHOOK_URL")
 
-def filter_table(name, type, arch, license):
+def filter_table(cols, name, type, arch, license):
     tmp = data
+    # filter
     tmp = tmp[tmp["Name"].str.contains(name)]
     tmp = tmp[tmp["Type"].isin(type)]
     tmp = tmp[tmp["Architecture"].isin(arch)]
     tmp = tmp[tmp["License"].isin(license)]
-    tmp = tmp.rename({"Type": "T"}, axis=1)
-    tmp["T"] = tmp["T"].apply(lambda x: x[0])
-    tmp["Name"] = tmp["Name"].apply(lambda x: f'<a target="_blank" href="https://huggingface.co/{x}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{x}</a>')
+    # show/hide
+    tmp = tmp.drop(cols, axis=1)
+    # prettify
+    if "Type" in tmp:
+        tmp["Type"] = tmp["Type"].apply(lambda x: x[0])
+        tmp = tmp.rename({"Type": "T"}, axis=1)
+
+    if "Name" in tmp:
+        tmp["Name"] = tmp["Name"].apply(lambda x: f'<a target="_blank" href="https://huggingface.co/{x}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{x}</a>')
+    
     return tmp
 
 def submit_model(name):
@@ -47,25 +55,25 @@ with gr.Blocks() as demo:
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.Tab("🏅 LLM Benchmark"):
             with gr.Row():
-                with gr.Column(variant="panel"):
+                with gr.Column():
                     namefilter = gr.Textbox(max_lines=1, label="Search by model name and hit Enter")
-                    gr.Markdown("Show/hide columns: coming soon")
+                    colfilter = gr.CheckboxGroup(label="Hide columns", choices=list(data.columns))
                     
-                with gr.Column(variant="panel"):
+                with gr.Column():
                     typefilter = gr.CheckboxGroup(label="Filter by model type", choices=list(data["Type"].unique()), value=[n for n in data["Type"].unique() if n not in ["Pending"]])
                 
-                with gr.Column(variant="panel"):
+                with gr.Column():
                     archfilter = gr.CheckboxGroup(label="Filter by model architecture", choices=list(data["Architecture"].unique()), value=list(data["Architecture"].unique()))
                     lcnsfilter = gr.CheckboxGroup(label="Filter by model license", choices=list(data["License"].unique()), value=list(data["License"].unique()))
                 
             table = gr.Dataframe(datatype="markdown")
 
-            namefilter.submit(filter_table, [namefilter,typefilter,archfilter,lcnsfilter], table)
+            namefilter.submit(filter_table, [colfilter,namefilter,typefilter,archfilter,lcnsfilter], table)
             
-            for filter in [typefilter, archfilter, lcnsfilter]:
-                filter.input(filter_table, [namefilter,typefilter,archfilter,lcnsfilter], table)
+            for filter in [colfilter,typefilter,archfilter,lcnsfilter]:
+                filter.input(filter_table, [colfilter,namefilter,typefilter,archfilter,lcnsfilter], table)
 
-            demo.load(fn=filter_table, inputs=[namefilter,typefilter,archfilter,lcnsfilter], outputs=table)
+            demo.load(fn=filter_table, inputs=[colfilter,namefilter,typefilter,archfilter,lcnsfilter], outputs=table)
         
         with gr.Tab("📝 About"):
             gr.Markdown("""