Atualização do PLUE

app.py

@@ -238,6 +238,7 @@ def create_leaderboard_component(dataframe, displayed_cols, hidden_cols=None, ca
 
 # --- Definição do Grupo PLUE ---
 PLUE_GROUP_AREAS = ["Área Médica", "Área do Direito", "Provas Militares", "Computação", "Multidisciplinar"]
+PLUE_GENERAL_VIEW_NAME = "Conhecimentos Gerais para Língua Portuguesa"
 # -------
 
 
@@ -280,12 +281,10 @@ with demo:
         plue_tab_created = False
         plue_components = {}
 
-        # Ordenar áreas para consistência (PLUE será tratada separadamente)
         all_area_names = sorted(AREA_DEFINITIONS.keys())
 
         for area_name in all_area_names:
             if area_name not in PLUE_GROUP_AREAS:
-                # Criar abas normais para áreas fora do grupo
                 with gr.TabItem(f"🎓 {area_name}", id=tab_index):
                     tasks_in_area = AREA_DEFINITIONS[area_name]
                     area_cols_to_display = [
@@ -315,59 +314,79 @@ with demo:
                     )
                 tab_index += 1
             elif not plue_tab_created:
-                # Criar a aba PLUE apenas uma vez
                 with gr.TabItem("🎓 PLUE", id=tab_index) as plue_tab:
                     plue_tab_created = True
-                    gr.Markdown("## Selecione a área PLUE para visualizar:")
+                    gr.Markdown("## Selecione a visualização PLUE:")
                     plue_dropdown = gr.Dropdown(
-                        choices=PLUE_GROUP_AREAS,
-                        label="Área PLUE",
-                        value=PLUE_GROUP_AREAS[0] # Valor inicial
+                        choices=[PLUE_GENERAL_VIEW_NAME] + PLUE_GROUP_AREAS,
+                        label="Visualização PLUE",
+                        value=PLUE_GENERAL_VIEW_NAME
                     )
-                    plue_leaderboard_output = gr.Group() # Container para o leaderboard da sub-área
+                    plue_leaderboard_output = gr.Group()
                     
-                    # Função para atualizar o conteúdo da aba PLUE
-                    def update_plue_tab(selected_area):
-                        print(f"Atualizando aba PLUE para: {selected_area}")
-                        tasks_in_area = AREA_DEFINITIONS[selected_area]
-                        area_cols_to_display = [
-                            AutoEvalColumn.model_type_symbol.name,
-                            AutoEvalColumn.model.name,
-                        ] + [task.name for task in tasks_in_area]
+                    def update_plue_tab(selected_option):
+                        print(f"Atualizando aba PLUE para: {selected_option}")
                         
-                        area_hidden_cols = list(AREA_AVG_COLUMN_MAP.values()) + [
-                            task.name for task in Tasks if task not in tasks_in_area
-                        ] + [
-                            AutoEvalColumn.model_type.name,
-                            AutoEvalColumn.architecture.name,
-                            AutoEvalColumn.weight_type.name,
-                            AutoEvalColumn.precision.name,
-                            AutoEvalColumn.license.name,
-                            AutoEvalColumn.params.name,
-                            AutoEvalColumn.likes.name,
-                            AutoEvalColumn.still_on_hub.name,
-                            AutoEvalColumn.revision.name
-                        ] 
-                        
-                        # Criar o componente leaderboard para a área selecionada
+                        if selected_option == PLUE_GENERAL_VIEW_NAME:
+                            displayed_cols = [
+                                AutoEvalColumn.model_type_symbol.name,
+                                AutoEvalColumn.model.name,
+                            ] + [AREA_AVG_COLUMN_MAP[area] for area in PLUE_GROUP_AREAS if area in AREA_AVG_COLUMN_MAP]
+                            
+                            hidden_cols = [task.name for task in Tasks] + [
+                                avg_col for area, avg_col in AREA_AVG_COLUMN_MAP.items() if area not in PLUE_GROUP_AREAS
+                            ] + [
+                                AutoEvalColumn.average.name
+                            ] + [
+                                AutoEvalColumn.model_type.name,
+                                AutoEvalColumn.architecture.name,
+                                AutoEvalColumn.weight_type.name,
+                                AutoEvalColumn.precision.name,
+                                AutoEvalColumn.license.name,
+                                AutoEvalColumn.params.name,
+                                AutoEvalColumn.likes.name,
+                                AutoEvalColumn.still_on_hub.name,
+                                AutoEvalColumn.revision.name
+                            ]
+                            title = PLUE_GENERAL_VIEW_NAME
+                        else:
+                            selected_area = selected_option
+                            tasks_in_area = AREA_DEFINITIONS[selected_area]
+                            displayed_cols = [
+                                AutoEvalColumn.model_type_symbol.name,
+                                AutoEvalColumn.model.name,
+                            ] + [task.name for task in tasks_in_area]
+                            
+                            hidden_cols = list(AREA_AVG_COLUMN_MAP.values()) + [
+                                task.name for task in Tasks if task not in tasks_in_area
+                            ] + [
+                                AutoEvalColumn.model_type.name,
+                                AutoEvalColumn.architecture.name,
+                                AutoEvalColumn.weight_type.name,
+                                AutoEvalColumn.precision.name,
+                                AutoEvalColumn.license.name,
+                                AutoEvalColumn.params.name,
+                                AutoEvalColumn.likes.name,
+                                AutoEvalColumn.still_on_hub.name,
+                                AutoEvalColumn.revision.name
+                            ] 
+                            title = selected_area
                         plue_leaderboard = create_leaderboard_component(
                             LEADERBOARD_DF, 
-                            displayed_cols=area_cols_to_display,
-                            hidden_cols=[col for col in area_hidden_cols if col != AutoEvalColumn.average.name],
-                            title=selected_area
+                            displayed_cols=displayed_cols,
+                            hidden_cols=[col for col in hidden_cols if col in LEADERBOARD_DF.columns],
+                            title=title
                         )
                         return plue_leaderboard
                         
-                    # Registrar o evento de mudança do dropdown
                     plue_dropdown.change(
                         fn=update_plue_tab,
                         inputs=[plue_dropdown],
                         outputs=[plue_leaderboard_output]
                     )
                     
-                    # Carregar conteúdo inicial (para a primeira área do dropdown)
                     with plue_leaderboard_output:
-                         update_plue_tab(PLUE_GROUP_AREAS[0])
+                         update_plue_tab(PLUE_GENERAL_VIEW_NAME)
                          
                 tab_index += 1