Update functions.py

functions.py

@@ -101,38 +101,41 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
         if not api_key:
             raise ValueError("API Key no proporcionada. Ingresa una clave válida.")
         
-        # Usar un modelo público compatible (ej. gpt2 o flan-t5-base)
-        API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-base"  # Modelo alternativo <button class="citation-flag" data-index="2">
+        # Prompt mejorado con ejemplos <button class="citation-flag" data-index="3">
+        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,...
+        """
+        
+        # Usar un modelo público compatible
+        API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-base"
         headers = {
             "Authorization": f"Bearer {api_key}",
             "Content-Type": "application/json"
         }
         
-        # Construir el prompt
-        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. "
-            "Formato esperado: id,tarea1_label,tarea2_label,..."
-        )
-        
         # Enviar solicitud a la API
         response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-        response.raise_for_status()  # Lanza error si hay un problema HTTP
+        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
+            if not encabezado.startswith("id,"):
+                raise ValueError("El modelo no generó un encabezado válido. Inténtalo de nuevo o prueba otro modelo.")
+            
             return encabezado
         else:
             raise ValueError(f"Error en la API: {response.text}")
     
     except requests.exceptions.HTTPError as e:
         if e.response.status_code == 401:
-            raise ValueError("API Key inválida o sin permisos. Verifica tu clave en HuggingFace.")
+            raise ValueError("API Key inválida o sin permisos.")
         elif e.response.status_code == 403:
-            raise ValueError("Acceso denegado al modelo. Prueba con otro modelo público como 'google/flan-t5-base'.")
-        elif e.response.status_code == 404:
-            raise ValueError("El modelo no existe en HuggingFace. Asegúrate de usar un modelo válido.")
+            raise ValueError("Acceso denegado al modelo. Prueba con otro modelo público.")
         else:
             raise ValueError(f"Error HTTP: {str(e)}")
     except Exception as e: