Update functions.py

functions.py

@@ -101,31 +101,36 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
         if not api_key:
             raise ValueError("API Key no proporcionada. Ingresa una clave válida.")
         
-        # Prompt mejorado con ejemplos <button class="citation-flag" data-index="3">
+        # Prompt mejorado con ejemplos detallados
         prompt = f"""
-        Genera un encabezado de CSV para las siguientes tareas: {', '.join(tareas_seleccionadas)}.
-        Incluye 'id' como primer elemento y etiquetas relevantes para cada tarea.
-        Ejemplo de formato esperado: id,tarea1_label,tarea2_label,...
+        Eres un asistente especializado en generar encabezados de CSV para datasets multimodales.
+        Tareas seleccionadas: {', '.join(tareas_seleccionadas)}.
+        Genera un encabezado de CSV con las siguientes reglas:
+        1. El primer campo debe ser 'id'.
+        2. Cada tarea debe tener una columna con el formato 'tarea_label' (ej. 'clasificación_de_texto_label').
+        3. Evita texto adicional, solo devuelve el encabezado.
+        Ejemplo para tareas 'Clasificación de texto' y 'Detección de objetos':
+        id,clasificación_de_texto_label,detección_de_objetos_label
         """
         
-        # Usar un modelo público compatible
-        API_URL = "https://api-inference.huggingface.co/models/gpt2"
+        # Usar un modelo más adecuado para prompts en español
+        API_URL = "https://api-inference.huggingface.co/models/dccuchile/bert-base-spanish-wwm-cased"  # Modelo en español <button class="citation-flag" data-index="1">
         headers = {
             "Authorization": f"Bearer {api_key}",
             "Content-Type": "application/json"
         }
         
         # Enviar solicitud a la API
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+        response = requests.post(API_URL, headers=headers, json={"inputs": prompt, "parameters": {"max_length": 256}})
         response.raise_for_status()
         
         # Procesar respuesta
         if response.status_code == 200:
             encabezado = response.json()[0]["generated_text"].strip()
             
-            # Validar que el encabezado comience con 'id' y contenga comas
+            # Validar formato básico
             if not encabezado.startswith("id,"):
-                raise ValueError("El modelo no generó un encabezado válido. Inténtalo de nuevo o prueba otro modelo.")
+                raise ValueError("El encabezado no inicia con 'id,'. Respuesta del modelo: " + encabezado)
             
             return encabezado
         else:
@@ -138,5 +143,7 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
             raise ValueError("Acceso denegado al modelo. Prueba con otro modelo público.")
         else:
             raise ValueError(f"Error HTTP: {str(e)}")
+    except KeyError:
+        raise ValueError("La respuesta de la API no contiene datos válidos. Asegúrate de usar un modelo compatible.")
     except Exception as e:
         raise ValueError(f"Error inesperado: {str(e)}")