Update functions.py

functions.py

@@ -1,21 +1,13 @@
 # functions.py
 import json
-import requests
 from datasets import load_dataset
-import matplotlib.pyplot as plt
-import time
 
-# Función para cargar modalidades y tareas
+# Cargar modalidades y tareas
 def cargar_modalidades_tareas():
-    try:
-        with open("modalidades_tareas.json", "r") as file:
-            MODALIDAD_TAREAS = json.load(file)
-        return MODALIDAD_TAREAS
-    except FileNotFoundError:
-        raise ValueError("El archivo 'modalidades_tareas.json' no se encontró.")
-    except json.JSONDecodeError:
-        raise ValueError("El archivo 'modalidades_tareas.json' tiene un formato incorrecto.")
-
+    with open("modalidades_tareas.json", "r") as file:
+        MODALIDAD_TAREAS = json.load(file)
+    return MODALIDAD_TAREAS
+    
 # Validar modalidades y tareas
 def validar_modalidades_tareas(modalidades_tareas):
     try:
@@ -92,6 +84,39 @@ def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_p
     except Exception as e:
         raise ValueError(f"Error al generar el dataset: {str(e)}")
 
+# Generar encabezado localmente
+def generar_encabezado(tareas_seleccionadas):
+    MODALIDAD_TAREAS = cargar_modalidades_tareas()
+    
+    if not tareas_seleccionadas:
+        raise ValueError("Selecciona al menos una tarea.")
+    
+    columnas = ["id"]
+    modalidades_usadas = set()
+    tareas_procesadas = set()
+    
+    # Mapear tareas a sus modalidades y columnas
+    for tarea in tareas_seleccionadas:
+        for modalidad, datos in MODALIDAD_TAREAS.items():
+            if tarea in datos["tareas"]:
+                # Añadir columnas de la modalidad una vez
+                if modalidad not in modalidades_usadas:
+                    columnas.extend(datos["columnas"])
+                    modalidades_usadas.add(modalidad)
+                # Añadir columna de la tarea
+                columna_tarea = f"{tarea.lower().replace(' ', '_')}_label"
+                if columna_tarea not in tareas_procesadas:
+                    columnas.append(columna_tarea)
+                    tareas_procesadas.add(columna_tarea)
+    
+    # Optimizar columnas (ej. ordenar y eliminar redundancias)
+    columnas_optimizadas = ["id"] + sorted(
+        columnas[1:],
+        key=lambda x: (x.endswith("_input"), x)  # Priorizar columnas de entrada <button class="citation-flag" data-index="4">
+    )
+    
+    return ",".join(columnas_optimizadas)
+
 def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
     max_reintentos = 3
     espera_entre_reintentos = 2