1. Changed "Link" to "Model_Link"

2. Added type hints
3. Added a subfunction
4. Removed unnecessary variables

core.py

@@ -75,7 +75,7 @@ def filter_type(df: pd.DataFrame, model_types: list[str]) -> pd.DataFrame:
     return df[df["Type"].isin(model_types)]
 
 
-def filter_model_size(df: pd.DataFrame, model_sizes, lookup: dict):
+def filter_model_size(df: pd.DataFrame, model_sizes: list, lookup: dict):
     filtered_model_size = [model_name for model_name, model_size in lookup.items() if
                            model_sizes[0] <= model_size <= model_sizes[1]]
     filtered_df = df[df['Model_Name'].isin(filtered_model_size)]
@@ -154,17 +154,7 @@ def update_df(
         return sort_cols(df, fewshot)
 
 
-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()
-
-    tasks_checkbox_group_update = gr.CheckboxGroup(
-        choices=available_tasks,
-        value=new_selected_tasks,
-    )
-
+def update_fewshot_status(current_selected_tab, is_fewshot_current):
     if current_selected_tab == 0:
         is_fewshot_new = is_fewshot_current
         fewshot_available = True
@@ -177,11 +167,27 @@ def update_task_groups_and_fewshot(current_selected_tab: int, model_types, langs
     else:
         raise ValueError(f"Unknown tab id {current_selected_tab}")
 
-    fewshot_radio_update = gr.Radio(
-        value=is_fewshot_new,
-        interactive=fewshot_available,
+    return is_fewshot_new, fewshot_available
+
+
+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()
+
+    tasks_checkbox_group_update = gr.CheckboxGroup(
+        choices=available_tasks,
+        value=new_selected_tasks,
     )
 
+    is_fewshot_new, fewshot_available = update_fewshot_status(current_selected_tab, is_fewshot_current)
+
+    # fewshot_radio_update = gr.Radio(
+    #     value=is_fewshot_new,
+    #     interactive=fewshot_available,
+    # )
+
     if current_selected_tab == 2:
         model_types = gr.CheckboxGroup(
             value=[T_SYMBOLS['chat']],

utils.py

@@ -1,4 +1,6 @@
-def add_model_hyperlink(df, lookup):
+import pandas as pd
+
+def add_model_hyperlink(df: pd.DataFrame, lookup: dict):
     df["Model_Name"] = df["Model_Name"].apply(
         lambda
             x: f'<a target="_blank" href="{lookup[x]}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;"> {x} </a>' if