Update app.py

app.py

@@ -8,6 +8,10 @@ import lime.lime_tabular
 import optuna
 import wandb
 import json
+import time
+import psutil
+import shutil
+import ast
 from smolagents import HfApiModel, CodeAgent
 from huggingface_hub import login
 from sklearn.ensemble import RandomForestClassifier
@@ -299,7 +303,8 @@ with gr.Blocks() as demo:
         lime_img = gr.Image(label="LIME Explanation")
 
 
-    agent_btn.click(fn=run_agent, inputs=[], outputs=[insights_output, visual_output])
+    agent_btn.click(fn=analyze_data, inputs=[file_input], outputs=[insights_output, visual_output])
+    visual_output = gr.Gallery(label="Visualizations").style(grid=(2, 3), height="auto")
     train_btn.click(fn=train_model, inputs=[], outputs=[metrics_output, trials_output])
     explain_btn.click(fn=explainability, inputs=[], outputs=[shap_img, lime_img])