Update app.py

app.py

@@ -51,7 +51,7 @@ def buscar_datasets(modalidades_seleccionadas, tareas_seleccionadas, filtro_tama
     return "\n".join(resultados)
 
 # Función para generar el dataset
-def generar_dataset(encabezado, datasets_seleccionados):
+def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_por_pagina=5):
     if not datasets_seleccionados:
         raise ValueError("Debes seleccionar al menos un dataset.")
     
@@ -63,11 +63,11 @@ def generar_dataset(encabezado, datasets_seleccionados):
         dataset_id = dataset_id.strip("- ").split(":")[0]  # Extraer ID del dataset
         try:
             dataset = load_dataset(dataset_id, split="train")
-            for i, fila in enumerate(dataset):
-                if i >= 5:  # Limitar a 5 filas por simplicidad
-                    break
+            inicio = (pagina_actual - 1) * filas_por_pagina
+            fin = pagina_actual * filas_por_pagina
+            for i, fila in enumerate(dataset[inicio:fin]):
                 valores = [str(fila.get(col, "valor_default")) for col in columnas[1:]]  # Ignorar 'id'
-                filas.append(f"id_{i}," + ",".join(valores))
+                filas.append(f"id_{inicio + i}," + ",".join(valores))
         except Exception as e:
             filas.append(f"Error cargando dataset {dataset_id}: {str(e)}")
     
@@ -84,8 +84,9 @@ with gr.Blocks() as demo:
         modalidades_seleccionadas = gr.CheckboxGroup(choices=modalidades, label="Selecciona Modalidades")
     
     # Panel 2: Configuración de Tareas (CheckboxGroup expandido)
-    with gr.Row():
-        tareas_disponibles = gr.CheckboxGroup(choices=[], label="Selecciona Tareas", interactive=True, elem_classes=["expanded-checkbox"])
+    with gr.Column():
+        gr.Markdown("### Selecciona Tareas")
+        tareas_disponibles = gr.CheckboxGroup(choices=[], label="Tareas Disponibles", interactive=True, elem_classes=["expanded-checkbox"])
         modalidades_seleccionadas.change(actualizar_tareas, inputs=modalidades_seleccionadas, outputs=tareas_disponibles)
 
     # Panel 3: Generar Encabezado del CSV
@@ -107,18 +108,26 @@ with gr.Blocks() as demo:
             outputs=datasets_encontrados
         )
 
-    # Panel 5: Generar Dataset
+    # 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")
         btn_generar_dataset = gr.Button("Generar Dataset")
         vista_previa_csv = gr.Textbox(label="Vista Previa del CSV", interactive=False)
         btn_descargar_csv = gr.File(label="Descargar CSV")
-        btn_generar_dataset.click(generar_dataset, inputs=[encabezado_generado, datasets_encontrados], outputs=vista_previa_csv)
+        btn_generar_dataset.click(
+            generar_dataset,
+            inputs=[encabezado_generado, 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 */
 }
 """