Update functions.py

functions.py

@@ -101,27 +101,32 @@ 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 detallados
+        # Prompt mejorado
         prompt = f"""
-        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:
+        Genera un encabezado de CSV para las siguientes tareas: {', '.join(tareas_seleccionadas)}.
+        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
+        2. Las columnas deben seguir el formato 'tarea_label' (ej. 'clasificación_de_texto_label').
+        3. No incluir espacios ni texto adicional.
+        Ejemplo: id,traducción_label,clasificación_de_imágenes_label
         """
         
-        # 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">
+        # Usar un modelo de generación de texto compatible (ej. GPT-2) <button class="citation-flag" data-index="2">
+        API_URL = "https://api-inference.huggingface.co/models/gpt2"
         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, "parameters": {"max_length": 256}})
+        response = requests.post(
+            API_URL,
+            headers=headers,
+            json={
+                "inputs": prompt,
+                "parameters": {"max_length": 256, "temperature": 0.5}  # Ajustar parámetros <button class="citation-flag" data-index="2">
+            }
+        )
         response.raise_for_status()
         
         # Procesar respuesta
@@ -130,7 +135,7 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
             
             # Validar formato básico
             if not encabezado.startswith("id,"):
-                raise ValueError("El encabezado no inicia con 'id,'. Respuesta del modelo: " + encabezado)
+                raise ValueError(f"Formato inválido. Respuesta del modelo: {encabezado}")
             
             return encabezado
         else:
@@ -141,6 +146,8 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
             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.")
+        elif e.response.status_code == 503:
+            raise ValueError("El modelo está temporalmente no disponible. Intenta más tarde o prueba con otro modelo.") <button class="citation-flag" data-index="1">
         else:
             raise ValueError(f"Error HTTP: {str(e)}")
     except KeyError: