Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 from functions import (
     cargar_modalidades_tareas,
@@ -6,31 +5,46 @@ from functions import (
     buscar_datasets,
     generar_dataset,
     generar_encabezado,
-    actualizar_modalidades_tareas_desde_huggingface  # Nueva función
+    actualizar_modalidades_tareas_desde_huggingface
 )
 
-MODALIDAD_TAREAS = cargar_modalidades_tareas()
+# Estado para almacenar las tareas disponibles
+tareas_disponibles_state = gr.State([])
 
-# Cargar/actualizar modalidades
-try:
-    MODALIDAD_TAREAS = cargar_modalidades_tareas()
-except json.JSONDecodeError:
-    # Si el JSON está corrupto, actualizar desde HuggingFace <button class="citation-flag" data-index="1">
-    actualizar_modalidades_tareas_desde_huggingface()
-    MODALIDAD_TAREAS = cargar_modalidades_tareas()
+# Cargar inicialmente las tareas
+MODALIDAD_TAREAS = cargar_modalidades_tareas()
+tareas_iniciales = [t for datos in MODALIDAD_TAREAS.values() for t in datos["tareas"].keys()]
 
 with gr.Blocks() as demo:
     gr.Markdown("# Diseñador de Modelos de Redes Neuronales y Generador de Datasets Multimodales")
-
-    # Botón para actualizar el JSON
+    
+    # Panel 1: Tareas Disponibles (ahora vinculado al State)
     with gr.Row():
-        btn_actualizar_json = gr.Button("Actualizar Modalidades desde HuggingFace")
-        mensaje_actualizacion = gr.Textbox(label="Estado de Actualización", interactive=False)
-        btn_actualizar_json.click(
-            fn=actualizar_modalidades_tareas_desde_huggingface,
-            inputs=[],
-            outputs=mensaje_actualizacion
+        tareas_disponibles = gr.CheckboxGroup(
+            choices=tareas_iniciales,
+            label="Tareas Disponibles",
+            interactive=True
         )
+        tareas_disponibles_state.value = tareas_iniciales  # Inicializar State
+    
+    # Panel 2: Botón de Actualización
+    with gr.Row():
+        btn_actualizar_json = gr.Button("Actualizar Modalidades desde HuggingFace")
+        mensaje_actualizacion = gr.Textbox(label="Estado", interactive=False)
+    
+    # Al actualizar el JSON, refrescar las tareas
+    def actualizar_tareas_handler():
+        nuevas_tareas = actualizar_modalidades_tareas_desde_huggingface()
+        return {
+            tareas_disponibles: gr.CheckboxGroup.update(choices=nuevas_tareas),
+            mensaje_actualizacion: "Tareas actualizadas correctamente."
+        }
+    
+    btn_actualizar_json.click(
+        fn=actualizar_tareas_handler,
+        inputs=[],
+        outputs=[tareas_disponibles, mensaje_actualizacion]
+    )
 
     # Panel 1: Seleccionar Tareas
     with gr.Row():