Modified lang_list and task columns

app.py

@@ -31,11 +31,10 @@ request_folder = request_file.parent
 #     "CO₂ cost (kg)",
 # ]
 # languages
-lang_list = ["Spain", "Costa Rica", "Mexico", "Peru", "Uruguay"]
+lang_list = ["Spain", "Portuguese", "English", "Spanish", "Costa Rica", "Mexico", "Peru", "Uruguay", "Basque", "Catalan", "Galician"]
 
 # column order
 model_columns = ["model_name", "url", "type"]
-task_columns = [f"tass_{lang.lower().replace(' ', '_')}" for lang in lang_list]
 
 scheduler = CommitScheduler(
     repo_id="iberbench/ivace-user-request",
@@ -82,6 +81,7 @@ def load_data(lang) -> pd.DataFrame:
             .to_pandas()
         )
         # filter lang columns
+        task_columns = [col for col in data.columns if col not in model_columns]
         task_lang_columns = get_lang_columns(task_columns, lang)
         data = data[model_columns + task_lang_columns]
 
@@ -143,6 +143,7 @@ def create_search_per_language(lang: str, search_dict: dict):
             ] = True
 
         # select columns to display
+        task_columns = [col for col in st.session_state[f"leaderboard_data_{lang}"].columns if col not in model_columns]
         task_lang_columns = get_lang_columns(task_columns, lang)
         columns = model_columns + task_lang_columns