Update app.py

app.py

@@ -1,12 +1,19 @@
+# app.py
 import gradio as gr
+import matplotlib.pyplot as plt
 from functions import (
-    MODALIDAD_TAREAS,
+    cargar_modalidades_tareas,
+    validar_modalidades_tareas,
     generar_grafica_barras,
-    generar_encabezado,
     buscar_datasets,
-    generar_dataset
+    generar_dataset,
+    generar_encabezado_inteligente
 )
 
+# Cargar modalidades y tareas
+MODALIDAD_TAREAS = cargar_modalidades_tareas()
+validar_modalidades_tareas(MODALIDAD_TAREAS)
+
 # Interfaz Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Diseñador de Modelos de Redes Neuronales y Generador de Datasets Multimodales")
@@ -23,17 +30,17 @@ with gr.Blocks() as demo:
         grafica_barras = gr.Plot(label="Gráfica de Barras")
         btn_actualizar_grafica = gr.Button("Actualizar Gráfica")
         btn_actualizar_grafica.click(
-            fn=lambda tareas: generar_grafica_barras(tareas),
+            fn=lambda tareas: generar_grafica_barras(tareas, MODALIDAD_TAREAS),
             inputs=tareas_seleccionadas,
             outputs=grafica_barras
         )
     
-    # Panel 3: Generar Encabezado del CSV
+    # Panel 3: Generar Encabezado del CSV con IA
     with gr.Row():
         btn_generar_encabezado = gr.Button("Generar Encabezado del CSV")
         encabezado_generado = gr.Textbox(label="Encabezado Generado", interactive=False)
         btn_generar_encabezado.click(
-            fn=generar_encabezado,
+            fn=generar_encabezado_inteligente,
             inputs=tareas_seleccionadas,
             outputs=encabezado_generado
         )