Create statistical_graph_generator.py

tools/statistical_graph_generator.py

@@ -0,0 +1,90 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from smolagents.tools import tool
+import time
+from typing import List, Dict
+
+@tool
+def plot_bar_chart(labels: List[str], values: List[float], title: str = "Gráfico de Barras", x_label: str = "Categorias", y_label: str = "Valores") -> str:
+    """
+    Gera e salva um gráfico de barras.
+    Args:
+        labels: Uma lista de strings para os rótulos de cada barra.
+        values: Uma lista de números (int ou float) correspondendo ao valor de cada barra.
+        title: O título do gráfico.
+        x_label: O rótulo do eixo X.
+        y_label: O rótulo do eixo Y.
+    Returns:
+        O caminho do arquivo da imagem PNG gerada.
+    """
+    try:
+        plt.figure(figsize=(10, 6))
+        plt.bar(labels, values)
+        plt.title(title)
+        plt.xlabel(x_label)
+        plt.ylabel(y_label)
+        plt.xticks(rotation=45, ha="right")
+        plt.grid(axis='y', linestyle='--', alpha=0.7)
+        plt.tight_layout()
+
+        file_path = f"bar_chart_{int(time.time())}.png"
+        plt.savefig(file_path)
+        plt.close()
+
+        return f"Gráfico de barras gerado e salvo em: {file_path}"
+    except Exception as e:
+        return f"Erro ao gerar o gráfico de barras: {str(e)}"
+
+@tool
+def plot_pie_chart(labels: List[str], sizes: List[float], title: str = "Gráfico de Setores") -> str:
+    """
+    Gera e salva um gráfico de pizza (setores).
+    Args:
+        labels: Uma lista de strings para os rótulos de cada fatia.
+        sizes: Uma lista de números (int ou float) correspondendo ao tamanho de cada fatia.
+        title: O título do gráfico.
+    Returns:
+        O caminho do arquivo da imagem PNG gerada.
+    """
+    try:
+        plt.figure(figsize=(8, 8))
+        plt.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140, shadow=True)
+        plt.title(title)
+        plt.axis('equal')  # Assegura que o gráfico seja um círculo.
+
+        file_path = f"pie_chart_{int(time.time())}.png"
+        plt.savefig(file_path)
+        plt.close()
+
+        return f"Gráfico de setores gerado e salvo em: {file_path}"
+    except Exception as e:
+        return f"Erro ao gerar o gráfico de setores: {str(e)}"
+
+@tool
+def plot_histogram(data: List[float], bins: int = 10, title: str = "Histograma", x_label: str = "Valores", y_label: str = "Frequência") -> str:
+    """
+    Gera e salva um histograma a partir de um conjunto de dados.
+    Args:
+        data: Uma lista de números (int ou float) para compor o histograma.
+        bins: O número de 'caixas' (colunas) no histograma.
+        title: O título do gráfico.
+        x_label: O rótulo do eixo X.
+        y_label: O rótulo do eixo Y.
+    Returns:
+        O caminho do arquivo da imagem PNG gerada.
+    """
+    try:
+        plt.figure(figsize=(10, 6))
+        plt.hist(data, bins=bins, edgecolor='black')
+        plt.title(title)
+        plt.xlabel(x_label)
+        plt.ylabel(y_label)
+        plt.grid(axis='y', linestyle='--', alpha=0.7)
+
+        file_path = f"histogram_{int(time.time())}.png"
+        plt.savefig(file_path)
+        plt.close()
+
+        return f"Histograma gerado e salvo em: {file_path}"
+    except Exception as e:
+        return f"Erro ao gerar o histograma: {str(e)}"