Update app.py

app.py

@@ -5,8 +5,11 @@ from Gradio_UI import GradioUI
 import os
 from tools.graph_plotter_tool import plot_function_graph
 from tools.statistical_graph_analyzer import analyze_statistical_graph
+
+
+from tools.statistical_graph_generator import plot_bar_chart, plot_pie_chart, plot_histogram, plot_line_chart
+from tools.statistical_graph_analyzer import analyze_dataset
 from tools.media_graph_critic import critique_media_graph
-from tools.statistical_graph_generator import generate_statistical_graph
 
 #from openinference.instrumentation.smolagents import SmolAgentsInstrumentor
 
@@ -46,8 +49,17 @@ with open("prompts.yaml", 'r', encoding='utf-8') as stream:
 # O agente usará nosso modelo e nosso prompt, com a única ferramenta de resposta final.
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, plot_function_graph, analyze_statistical_graph, critique_media_graph, generate_statistical_graph], # Adicione as novas ferramentas aqui
-    max_steps=4, 
+    tools=[
+        final_answer,
+        plot_function_graph,
+        plot_bar_chart,
+        plot_pie_chart,
+        plot_histogram,
+        plot_line_chart,
+        analyze_dataset,
+        critique_media_graph
+    ],  # Adicione as novas ferramentas aqui
+    max_steps=5, 
     verbosity_level=1,
     prompt_templates=prompt_templates
 )