Update functions.py

functions.py

@@ -1,5 +1,3 @@
-from datasets import load_dataset
-import requests
 import json
 import matplotlib.pyplot as plt
 
@@ -38,14 +36,24 @@ def generar_grafica_barras(tareas_seleccionadas):
     
     return fig
 
-# Función para generar el encabezado del CSV
+# Función para generar el encabezado del CSV (sistema inteligente)
 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
+    
+    # Generar columnas dinámicamente basadas en las tareas seleccionadas
+    for tarea in tareas_seleccionadas:
+        # Columna de entrada para la tarea
+        columnas.append(f"{tarea.lower().replace(' ', '_')}_input")
+        # Columna de salida para la tarea
+        columnas.append(f"{tarea.lower().replace(' ', '_')}_output")
+    
+    # Añadir columnas adicionales para etiquetas si es necesario
     for tarea in tareas_seleccionadas:
         columnas.append(f"{tarea.lower().replace(' ', '_')}_label")
+    
     return ",".join(columnas)
 
 # Función para buscar datasets compatibles en HuggingFace
@@ -87,4 +95,4 @@ def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_p
             filas.append(f"Error cargando dataset {dataset_id}: {str(e)}")
     
     contenido_csv = "\n".join([encabezado] + filas)
-    return contenido_csv
+    return contenido_csv