Tie model and language selection to selected tab

app.py

@@ -101,23 +101,23 @@ with demo:
 
         demo.load(
             core.update_task_groups_and_fewshot,
-            [gr.State(value=0), fewshot],
-            [shown_tasks, fewshot, selected_tab],
+            [gr.State(value=0), model_types, langs_bar,fewshot],
+            [shown_tasks, fewshot, selected_tab, model_types, langs_bar],
         )
         fewshot.change(
             core.update_task_groups_and_fewshot,
-            [selected_tab, fewshot],
-            [shown_tasks, fewshot, selected_tab],
+            [selected_tab, model_types, langs_bar, fewshot],
+            [shown_tasks, fewshot, selected_tab, model_types, langs_bar],
         )
         acc.select(
             core.update_task_groups_and_fewshot,
-            inputs=[gr.State(value=0), fewshot],
-            outputs=[shown_tasks, fewshot, selected_tab],
+            inputs=[gr.State(value=0), model_types, langs_bar, fewshot],
+            outputs=[shown_tasks, fewshot, selected_tab, model_types, langs_bar],
         )
         misc.select(
             core.update_task_groups_and_fewshot,
-            inputs=[gr.State(value=1), fewshot],
-            outputs=[shown_tasks, fewshot, selected_tab],
+            inputs=[gr.State(value=1), model_types, langs_bar, fewshot],
+            outputs=[shown_tasks, fewshot, selected_tab, model_types, langs_bar],
         )
         for comp, fn in [
             (search_bar, "submit"),

core.py

@@ -4,10 +4,10 @@ import os
 import gradio as gr
 import numpy as np
 import pandas as pd
-import plotly.express as px
 from datasets import load_dataset
 
 import style
+from style import T_SYMBOLS, LANG_SYMBOLS
 
 ZERO_SHOT_ONLY = ["BELEBELE"]
 FEW_SHOT_ONLY = ["GSM8K", "TruthfulQA"]
@@ -127,7 +127,7 @@ def update_df(
         return sort_cols(df, fewshot)
 
 
-def update_task_groups_and_fewshot(current_selected_tab: int, is_fewshot_current: bool = False):
+def update_task_groups_and_fewshot(current_selected_tab: int, model_types, langs_bar, is_fewshot_current: bool = False, ):
     selected_task_type = get_selected_task_type(current_selected_tab)
     available_tasks = get_available_task_groups(selected_task_type, is_fewshot_current)
     new_selected_tasks = available_tasks.copy()
@@ -149,7 +149,25 @@ def update_task_groups_and_fewshot(current_selected_tab: int, is_fewshot_current
         interactive=fewshot_available,
     )
 
-    return [tasks_checkbox_group_update, fewshot_radio_update, current_selected_tab]
+    model_types = gr.CheckboxGroup(
+        label="Select model type",
+        choices=[
+            (
+                f"Pretrained {T_SYMBOLS['pretrained']}",
+                T_SYMBOLS["pretrained"],
+            ),
+            (f"Chat {T_SYMBOLS['chat']}", T_SYMBOLS["chat"]),
+        ],
+        value=list(T_SYMBOLS.values()),
+        interactive=True
+    )
+    langs_bar = gr.CheckboxGroup(
+        choices=[(LANG_SYMBOLS.get(l, l), l) for l in languages_list],
+        value=languages_list,
+        interactive=True,
+    )
+
+    return [tasks_checkbox_group_update, fewshot_radio_update, current_selected_tab, model_types, langs_bar]
 
 
 def get_selected_task_type(task_type_id):