Update functions.py

functions.py

@@ -6,48 +6,41 @@ from datasets import load_dataset
 
 # Cargar modalidades y tareas (ahora devuelve una lista de tareas) <button class="citation-flag" data-index="1">
 def cargar_modalidades_tareas():
-    with open("modalidades_tareas.json", "r") as file:
-        MODALIDAD_TAREAS = json.load(file)
-    return MODALIDAD_TAREAS
+    try:
+        with open("modalidades_tareas.json", "r") as file:
+            return json.load(file)
+    except FileNotFoundError:
+        return {}  # Devolver diccionario vacío si no existe
 
-# Función para actualizar el JSON desde HuggingFace y devolver tareas
 def actualizar_modalidades_tareas_desde_huggingface():
     MODALIDADES = ["text-classification", "image-classification", "speech-recognition"]
     MODALIDAD_TAREAS = cargar_modalidades_tareas()  # Cargar JSON existente
     
     for task in MODALIDADES:
-        url = f"https://huggingface.co/api/datasets?task={task}&full=true&limit=10"
-        response = requests.get(url).json()
+        response = requests.get(f"https://huggingface.co/api/datasets?task={task}&full=true&limit=5").json()
         
         for dataset in response:
             dataset_id = dataset["id"]
             dataset_info = requests.get(f"https://huggingface.co/api/datasets/{dataset_id}").json()
             
-            # Añadir datasets y columnas al JSON
-            if task not in MODALIDAD_TAREAS["NLP"]["tareas"]:
-                MODALIDAD_TAREAS["NLP"]["tareas"][task] = {
-                    "nombre": "Clasificación de texto",
-                    "columnas": ["text", "label"],
+            # Añadir datos al JSON
+            if task not in MODALIDAD_TAREAS:
+                MODALIDAD_TAREAS[task] = {
+                    "nombre": task.replace("-", " ").capitalize(),
+                    "columnas": list(dataset_info.get("features", {}).keys()),
                     "datasets": {}
                 }
             
-            MODALIDAD_TAREAS["NLP"]["tareas"][task]["datasets"][dataset_id] = {
+            MODALIDAD_TAREAS[task]["datasets"][dataset_id] = {
                 "columnas": list(dataset_info.get("features", {}).keys()),
                 "licencia": dataset.get("license", "unknown")
             }
     
-    # Guardar JSON con indentación para evitar errores <button class="citation-flag" data-index="1">
+    # Guardar JSON con permisos explícitos <button class="citation-flag" data-index="2">
     with open("modalidades_tareas.json", "w") as file:
         json.dump(MODALIDAD_TAREAS, file, indent=4)
     
-    return "Metadatos actualizados: columnas y datasets almacenados."
-
-    
-# Cargar modalidades y tareas
-def cargar_modalidades_tareas():
-    with open("modalidades_tareas.json", "r") as file:
-        MODALIDAD_TAREAS = json.load(file)
-    return MODALIDAD_TAREAS
+    return list(MODALIDAD_TAREAS.keys())  # Devolver lista de tareas actualizadas
 
 # Generar gráfica de barras
 def generar_grafica_barras(tareas_seleccionadas, MODALIDAD_TAREAS):