Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from datasets import load_dataset
 import requests
 import json
+import matplotlib.pyplot as plt
 
 # Cargar modalidades y tareas desde un archivo JSON
 with open("modalidades_tareas.json", "r") as file:
@@ -18,33 +19,24 @@ def validar_modalidades_tareas(modalidades_tareas):
 # Validar el diccionario
 validar_modalidades_tareas(MODALIDAD_TAREAS)
 
-# Función para actualizar las tareas según las modalidades seleccionadas
-def actualizar_tareas(modalidades_seleccionadas):
-    tareas_disponibles = []
-    for modalidad in modalidades_seleccionadas:
-        if modalidad in MODALIDAD_TAREAS:
-            tareas_disponibles.extend(MODALIDAD_TAREAS[modalidad])
-    # Eliminar duplicados y devolver una lista de tareas disponibles
-    return list(set(tareas_disponibles))
-
-# Función para generar el encabezado del CSV (ahora incluye 'id')
-def generar_encabezado(modalidades_seleccionadas, tareas_seleccionadas):
-    if not modalidades_seleccionadas or not tareas_seleccionadas:
-        raise ValueError("Debes seleccionar al menos una modalidad y una tarea.")
+# 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]
+    
+    # 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_ylabel('Modalidades')
+    ax.set_title('Cantidad de Tareas por Modalidad')
+    ax.invert_yaxis()  # Invertir el eje Y para que las modalidades aparezcan de arriba hacia abajo
     
-    columnas = ["id"]  # Añadimos 'id' como primer elemento
-    for modalidad in modalidades_seleccionadas:
-        columnas.append(f"{modalidad.lower()}_input")
-        columnas.append(f"{modalidad.lower()}_output")
-        if modalidad == "Educación":  # Caso especial para una nueva modalidad
-            columnas.append("metodología_educativa")
-    for tarea in tareas_seleccionadas:
-        columnas.append(f"{tarea.lower().replace(' ', '_')}_label")
-    return ",".join(columnas)
+    return fig
 
 # Función para buscar datasets compatibles en HuggingFace
-def buscar_datasets(modalidades_seleccionadas, tareas_seleccionadas, filtro_tamaño=None, filtro_licencia=None):
-    query = "+".join(modalidades_seleccionadas + tareas_seleccionadas)
+def buscar_datasets(tareas_seleccionadas, filtro_tamaño=None, filtro_licencia=None):
+    query = "+".join(tareas_seleccionadas)
     url = f"https://huggingface.co/api/datasets?search={query}"
     response = requests.get(url)
     datasets = response.json()
@@ -56,9 +48,6 @@ def buscar_datasets(modalidades_seleccionadas, tareas_seleccionadas, filtro_tama
             continue
         if filtro_licencia and dataset.get("license") != filtro_licencia:
             continue
-        # Filtrar por modalidad específica
-        if "Educación" in modalidades_seleccionadas and "educación" not in dataset["tags"]:
-            continue
         resultados.append(f"- {dataset['id']}: {dataset['description']}")
     return "\n".join(resultados)
 
@@ -90,28 +79,22 @@ def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_p
 with gr.Blocks() as demo:
     gr.Markdown("# Diseñador de Modelos de Redes Neuronales y Generador de Datasets Multimodales")
     
-    # Panel 1: Selección de Modalidades
-    with gr.Row():
-        modalidades = list(MODALIDAD_TAREAS.keys())
-        modalidades_seleccionadas = gr.CheckboxGroup(choices=modalidades, label="Selecciona Modalidades")
-    
-    # Panel 2: Configuración de Tareas (Corregido)
+    # Panel 1: Lista de Tareas Disponibles
     with gr.Row():
-        gr.Markdown("### Selecciona Tareas")
-        tareas_disponibles = gr.CheckboxGroup(choices=[], label="Tareas Disponibles", interactive=True, elem_classes=["expanded-checkbox"])
-        modalidades_seleccionadas.change(
-            fn=lambda modalidades: actualizar_tareas(modalidades),  # Usamos una lambda para simplificar
-            inputs=modalidades_seleccionadas,
-            outputs=tareas_disponibles
+        gr.Markdown("### Tareas Disponibles")
+        tareas_disponibles = gr.Textbox(
+            label="Tareas",
+            value="\n".join([tarea for tareas in MODALIDAD_TAREAS.values() for tarea in tareas]),
+            interactive=False
         )
-
-    # Panel 3: Generar Encabezado del CSV
+    
+    # Panel 2: Gráfica de Barras
     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(generar_encabezado, inputs=[modalidades_seleccionadas, tareas_disponibles], outputs=encabezado_generado)
+        gr.Markdown("### Distribución de Tareas por Modalidad")
+        grafica_barras = gr.Plot(label="Gráfica de Barras")
+        grafica_barras.update(generar_grafica_barras())
 
-    # Panel 4: Buscar Datasets Compatibles con Filtros Adicionales
+    # Panel 3: 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")
@@ -120,11 +103,11 @@ with gr.Blocks() as demo:
         datasets_encontrados = gr.Textbox(label="Datasets Encontrados", interactive=False)
         btn_buscar_datasets.click(
             buscar_datasets,
-            inputs=[modalidades_seleccionadas, tareas_disponibles, filtro_tamaño, filtro_licencia],
+            inputs=[tareas_disponibles, filtro_tamaño, filtro_licencia],
             outputs=datasets_encontrados
         )
 
-    # Panel 5: Generar Dataset con Paginación
+    # Panel 4: 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")
@@ -133,19 +116,9 @@ with gr.Blocks() as demo:
         btn_descargar_csv = gr.File(label="Descargar CSV")
         btn_generar_dataset.click(
             generar_dataset,
-            inputs=[encabezado_generado, datasets_encontrados, pagina_actual, filas_por_pagina],
+            inputs=[datasets_encontrados, pagina_actual, filas_por_pagina],
             outputs=vista_previa_csv
         )
 
-# Estilo CSS para expandir el área de selección de tareas
-demo.css = """
-.expanded-checkbox {
-    max-height: 200px; /* Altura máxima */
-    overflow-y: auto;  /* Scroll vertical si es necesario */
-    border: 1px solid #ccc; /* Borde para mejorar la visibilidad */
-    padding: 10px; /* Espaciado interno */
-}
-"""
-
 # Ejecutar la aplicación
 demo.launch()