Update functions.py

functions.py

@@ -92,11 +92,25 @@ 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)}")
 
-# Función inteligente para generar el encabezado usando IA de HuggingFace
 def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
     max_reintentos = 3
     espera_entre_reintentos = 2
     
+    # Prompt mejorado con instrucciones estrictas <button class="citation-flag" data-index="3">
+    prompt = f"""
+    Genera SOLO el encabezado CSV para las tareas: {', '.join(tareas_seleccionadas)}.
+    Formato: id,tarea1_label,tarea2_label,...
+    Ejemplo para 'Clasificación de texto' y 'Segmentación':
+    id,clasificación_de_texto_label,segmentación_label
+    """
+    
+    # Usar modelo especializado en generación de texto estructurado <button class="citation-flag" data-index="2">
+    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-code-3b"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    
     for intento in range(max_reintentos + 1):
         try:
             if not tareas_seleccionadas:
@@ -105,33 +119,17 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
             if not api_key:
                 raise ValueError("API Key no proporcionada. Ingresa una clave válida.")
             
-            # Prompt optimizado
-            prompt = f"""
-            Eres un experto en datasets. Genera un encabezado CSV para las tareas: {', '.join(tareas_seleccionadas)}.
-            Ejemplo de salida para 'Clasificación de texto' y 'Detección de objetos':
-            id,clasificación_de_texto_label,detección_de_objetos_label
-            
-            Reglas:
-            - Solo responde con el encabezado, sin texto adicional.
-            - El primer campo debe ser 'id'.
-            - Las columnas deben seguir el formato 'tarea_label' en minúsculas y sin espacios.
-            - Separa las columnas con comas.
-            """
-            
-            # Usar un modelo alternativo (distilgpt2) <button class="citation-flag" data-index="2">
-            API_URL = "https://api-inference.huggingface.co/models/distilgpt2"
-            headers = {
-                "Authorization": f"Bearer {api_key}",
-                "Content-Type": "application/json"
-            }
-            
-            # Enviar solicitud a la API
+            # Enviar solicitud con parámetros ajustados <button class="citation-flag" data-index="2">
             response = requests.post(
                 API_URL,
                 headers=headers,
                 json={
                     "inputs": prompt,
-                    "parameters": {"max_length": 128, "temperature": 0.1}
+                    "parameters": {
+                        "max_length": 64,
+                        "temperature": 0.1,
+                        "stop": ["\n\n"]  # Detener la generación al finalizar el encabezado <button class="citation-flag" data-index="3">
+                    }
                 }
             )
             response.raise_for_status()
@@ -140,14 +138,12 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
             if response.status_code == 200:
                 encabezado = response.json()[0]["generated_text"].strip()
                 
-                # Validar formato
+                # Validar formato estricto
                 if not encabezado.startswith("id,"):
                     raise ValueError("Formato inválido. Respuesta: " + encabezado)
                 
                 return encabezado
-            else:
-                raise ValueError(f"Error en la API: {response.text}")
-        
+            
         except requests.exceptions.HTTPError as e:
             if e.response.status_code == 503 and intento < max_reintentos:
                 time.sleep(espera_entre_reintentos)