Update app.py

app.py

@@ -20,20 +20,35 @@ def validar_modalidades_tareas(modalidades_tareas):
 validar_modalidades_tareas(MODALIDAD_TAREAS)
 
 # Función para generar la gráfica de barras
-def generar_grafica_barras():
-    modalidades = list(MODALIDAD_TAREAS.keys())
-    cantidades = [len(MODALIDAD_TAREAS[modalidad]) for modalidad in modalidades]
+def generar_grafica_barras(tareas_seleccionadas):
+    # Contar la cantidad de tareas seleccionadas por modalidad
+    conteo_modalidades = {}
+    for modalidad, tareas in MODALIDAD_TAREAS.items():
+        conteo_modalidades[modalidad] = len([tarea for tarea in tareas if tarea in tareas_seleccionadas])
+    
+    modalidades = list(conteo_modalidades.keys())
+    cantidades = [conteo_modalidades[modalidad] for modalidad in modalidades]
     
     # Crear la gráfica de barras horizontal
     fig, ax = plt.subplots(figsize=(10, 6))
     ax.barh(modalidades, cantidades, color='skyblue', edgecolor='black')
-    ax.set_xlabel('Cantidad de Tareas')
+    ax.set_xlabel('Cantidad de Tareas Seleccionadas')
     ax.set_ylabel('Modalidades')
-    ax.set_title('Cantidad de Tareas por Modalidad')
+    ax.set_title('Distribución de Tareas Seleccionadas por Modalidad')
     ax.invert_yaxis()  # Invertir el eje Y para que las modalidades aparezcan de arriba hacia abajo
     
     return fig
 
+# Función para generar el encabezado del CSV
+def generar_encabezado(tareas_seleccionadas):
+    if not tareas_seleccionadas:
+        raise ValueError("Debes seleccionar al menos una tarea.")
+    
+    columnas = ["id"]  # Añadimos 'id' como primer elemento
+    for tarea in tareas_seleccionadas:
+        columnas.append(f"{tarea.lower().replace(' ', '_')}_label")
+    return ",".join(columnas)
+
 # Función para buscar datasets compatibles en HuggingFace
 def buscar_datasets(tareas_seleccionadas, filtro_tamaño=None, filtro_licencia=None):
     query = "+".join(tareas_seleccionadas)
@@ -81,17 +96,32 @@ with gr.Blocks() as demo:
     
     # Panel 1: Lista de Tareas Disponibles con Checkboxes
     with gr.Row():
-        #gr.Markdown("### Selecciona Tareas")
+        gr.Markdown("### Selecciona Tareas")
         todas_las_tareas = [tarea for tareas in MODALIDAD_TAREAS.values() for tarea in tareas]
         tareas_seleccionadas = gr.CheckboxGroup(choices=todas_las_tareas, label="Tareas Disponibles", interactive=True)
     
     # Panel 2: Gráfica de Barras
     with gr.Row():
-        gr.Markdown("### Distribución de Tareas por Modalidad")
+        gr.Markdown("### Distribución de Tareas Seleccionadas por Modalidad")
         grafica_barras = gr.Plot(label="Gráfica de Barras")
-        grafica_barras.value = generar_grafica_barras()
+        btn_actualizar_grafica = gr.Button("Actualizar Gráfica")
+        btn_actualizar_grafica.click(
+            fn=lambda tareas: generar_grafica_barras(tareas),
+            inputs=tareas_seleccionadas,
+            outputs=grafica_barras
+        )
+    
+    # Panel 3: Generar Encabezado del CSV
+    with gr.Row():
+        btn_generar_encabezado = gr.Button("Generar Encabezado del CSV")
+        encabezado_generado = gr.Textbox(label="Encabezado Generado", interactive=False)
+        btn_generar_encabezado.click(
+            fn=generar_encabezado,
+            inputs=tareas_seleccionadas,
+            outputs=encabezado_generado
+        )
 
-    # Panel 3: Buscar Datasets Compatibles con Filtros Adicionales
+    # Panel 4: Buscar Datasets Compatibles con Filtros Adicionales
     with gr.Row():
         with gr.Column():
             filtro_tamaño = gr.Dropdown(choices=["small", "medium", "large"], label="Filtrar por Tamaño")
@@ -104,7 +134,7 @@ with gr.Blocks() as demo:
             outputs=datasets_encontrados
         )
 
-    # Panel 4: Generar Dataset con Paginación
+    # Panel 5: Generar Dataset con Paginación
     with gr.Row():
         pagina_actual = gr.Number(value=1, label="Página Actual")
         filas_por_pagina = gr.Number(value=5, label="Filas por Página")
@@ -113,7 +143,7 @@ with gr.Blocks() as demo:
         btn_descargar_csv = gr.File(label="Descargar CSV")
         btn_generar_dataset.click(
             generar_dataset,
-            inputs=[datasets_encontrados, pagina_actual, filas_por_pagina],
+            inputs=[encabezado_generado, datasets_encontrados, pagina_actual, filas_por_pagina],
             outputs=vista_previa_csv
         )