Update functions.py

functions.py

@@ -82,4 +82,36 @@ def generar_encabezado(tareas_seleccionadas):
         )
     )
     
-    return ",".join(columnas_ordenadas)
+    return ",".join(columnas_ordenadas)
+
+def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_por_pagina=5):
+    try:
+        if not datasets_seleccionados:
+            raise ValueError("Debes seleccionar al menos un dataset.")
+        
+        columnas = encabezado.split(",")
+        filas = []
+        
+        # Procesar cada dataset seleccionado
+        for dataset_linea in datasets_seleccionados.split("\n"):
+            dataset_id = dataset_linea.strip("- ").split(":")[0]  # Extraer ID
+            
+            try:
+                dataset = load_dataset(dataset_id, split="train")
+                inicio = (pagina_actual - 1) * filas_por_pagina
+                fin = pagina_actual * filas_por_pagina
+                
+                for i, fila in enumerate(dataset[inicio:fin]):
+                    valores = []
+                    for col in columnas[1:]:  # Ignorar 'id'
+                        valores.append(str(fila.get(col, "valor_default")))
+                    filas.append(f"id_{inicio + i}," + ",".join(valores))
+            
+            except Exception as e:
+                filas.append(f"Error en dataset {dataset_id}: {str(e)}")
+        
+        contenido_csv = "\n".join([encabezado] + filas)
+        return contenido_csv
+    
+    except Exception as e:
+        raise ValueError(f"Error al generar el dataset: {str(e)}")