Update functions.py

functions.py

@@ -3,6 +3,7 @@ import json
 import requests
 from datasets import load_dataset
 import matplotlib.pyplot as plt
+import time
 
 # Función para cargar modalidades y tareas
 def cargar_modalidades_tareas():
@@ -93,8 +94,8 @@ def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_p
 
 # Función inteligente para generar el encabezado usando IA de HuggingFace
 def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
-    max_reintentos = 3  # Número máximo de reintentos <button class="citation-flag" data-index="1">
-    espera_entre_reintentos = 2  # Segundos entre reintentos
+    max_reintentos = 3
+    espera_entre_reintentos = 2
     
     for intento in range(max_reintentos + 1):
         try:
@@ -117,8 +118,8 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
             - Separa las columnas con comas.
             """
             
-            # Usar un modelo ligero y estable (t5-small) <button class="citation-flag" data-index="2">
-            API_URL = "https://api-inference.huggingface.co/models/t5-small"
+            # 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"
@@ -130,7 +131,7 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
                 headers=headers,
                 json={
                     "inputs": prompt,
-                    "parameters": {"max_length": 128, "temperature": 0.1}  # Parámetros ajustados <button class="citation-flag" data-index="2">
+                    "parameters": {"max_length": 128, "temperature": 0.1}
                 }
             )
             response.raise_for_status()
@@ -149,7 +150,7 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
         
         except requests.exceptions.HTTPError as e:
             if e.response.status_code == 503 and intento < max_reintentos:
-                time.sleep(espera_entre_reintentos)  # Esperar antes de reintentar <button class="citation-flag" data-index="1">
+                time.sleep(espera_entre_reintentos)
                 continue
             elif e.response.status_code == 401:
                 raise ValueError("API Key inválida o sin permisos.")
@@ -159,7 +160,7 @@ def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
         except Exception as e:
             raise ValueError(f"Error inesperado: {str(e)}")
     
-    # Si fallan todos los reintentos, usar un fallback local <button class="citation-flag" data-index="3">
+    # Fallback local si fallan todos los reintentos
     return generar_encabezado_local(tareas_seleccionadas)
 
 # Fallback local en caso de fallo de la API