Update functions.py

functions.py

@@ -97,32 +97,38 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
         if not tareas_seleccionadas:
             raise ValueError("Debes seleccionar al menos una tarea.")
         
-        # Validar que se haya proporcionado una API Key
+        # Validar API Key
         if not api_key:
             raise ValueError("Debes proporcionar una API Key válida.")
         
-        # Usar un modelo de HuggingFace para generar el encabezado
-        API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-large"
-        headers = {"Authorization": f"Bearer {api_key}"}
+        # Usar un modelo compatible con la API de HuggingFace
+        API_URL = "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-72B"  # Modelo alternativo <button class="citation-flag" data-index="3">
+        headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
         
-        # Construir el prompt para el modelo
+        # Construir el prompt
         prompt = (
             f"Genera un encabezado de CSV para las siguientes tareas: {', '.join(tareas_seleccionadas)}. "
-            "El encabezado debe incluir 'id' como primer elemento y etiquetas relevantes para cada tarea."
+            "Incluye 'id' como primer elemento y etiquetas relevantes para cada tarea. "
+            "Formato esperado: id,tarea1_label,tarea2_label,..."
         )
         
-        # Enviar la solicitud al modelo
+        # Enviar solicitud a la API
         response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-        response.raise_for_status()  # Lanzar una excepción si hay un error HTTP
+        response.raise_for_status()  # Lanzar error si hay un problema HTTP
         
+        # Procesar respuesta
         if response.status_code == 200:
-            encabezado = response.json()[0]["generated_text"]
+            encabezado = response.json()[0]["generated_text"].strip()
             return encabezado
         else:
-            raise ValueError(f"Error al generar el encabezado: {response.text}")
-    except requests.exceptions.RequestException as e:
-        raise ValueError(f"Error en la solicitud a la API de HuggingFace: {str(e)}")
-    except KeyError:
-        raise ValueError("La respuesta de la API no contiene los datos esperados.")
+            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.") <button class="citation-flag" data-index="1">
+        elif e.response.status_code == 404:
+            raise ValueError("El modelo no existe en HuggingFace. Prueba otro modelo.") <button class="citation-flag" data-index="2">
+        else:
+            raise ValueError(f"Error HTTP: {str(e)}")
     except Exception as e:
-        raise ValueError(f"Error inesperado al generar el encabezado: {str(e)}")
+        raise ValueError(f"Error inesperado: {str(e)}")