Update app.py

app.py

@@ -93,7 +93,7 @@ def generate_scatter_plot_with_labels(x_values: list[float], y_values: list[floa
         y_label: Label for the y-axis (default is 'Y-Axis').
 
     Returns:
-        PILImage Object
+        an image of the scatter plot.
     """
     try:
         if len(x_values) != len(y_values) or len(x_values) != len(labels):
@@ -132,7 +132,7 @@ def generate_scatter_plot_with_labels(x_values: list[float], y_values: list[floa
 
 @tool
 def generate_line_plot(x_values: list[float], y_values: list[float], title: str = 'Line Plot', x_label: str = 'X-Axis', y_label: str = 'Y-Axis') -> PILImage:
-    """Generates a line plot.
+    """Generates a line plot from the provided x and y values.
     
     Args:
         x_values: A list of numerical values for the x-axis.
@@ -141,7 +141,7 @@ def generate_line_plot(x_values: list[float], y_values: list[float], title: str
         x_label: Label for the x-axis.
         y_label: Label for the y-axis.
     Returns:
-        An Image object containing the line plot.
+        an image of the line plot.
     """
     try:
         plt.plot(x_values, y_values, marker='o', linestyle='-', color='b')
@@ -161,6 +161,90 @@ def generate_line_plot(x_values: list[float], y_values: list[float], title: str
     except Exception as e:
         print(f"Error generating line plot: {str(e)}")
 
+@tool
+def generate_histogram(data: list[float], bins: int = 10, title: str = 'Histogram', x_label: str = 'Values', y_label: str = 'Frequency') -> PILImage:
+    """Generates a histogram from the provided data and bins values.
+    
+    Args:
+        data: A list of numerical values.
+        bins: Number of bins for the histogram.
+        title: Title for the histogram.
+        x_label: Label for the x-axis.
+        y_label: Label for the y-axis.
+    Returns:
+        an image of the histogram plot.
+    """
+    try:
+        plt.hist(data, bins=bins, color='purple', edgecolor='black', alpha=0.7)
+        plt.title(title)
+        plt.xlabel(x_label)
+        plt.ylabel(y_label)
+
+        img_buffer = io.BytesIO()
+        plt.savefig(img_buffer, format='png', dpi=300, bbox_inches='tight')
+        plt.close()
+
+        img_buffer.seek(0)
+        img = PILImage.open(img_buffer)
+        return img
+
+    except Exception as e:
+        print(f"Error generating histogram: {str(e)}")
+
+@tool
+def generate_pie_chart(labels: list[str], values: list[float], title: str = 'Pie Chart') -> PILImage:
+    """Generates a pie chart from the provided labels and values.
+    
+    Args:
+        labels: A list of category labels.
+        values: A list of numerical values for each category.
+        title: Title for the pie chart.
+    Returns:
+        an image of the pie chart.
+    """
+    try:
+        plt.pie(values, labels=labels, autopct='%1.1f%%', startangle=140, colors=['blue', 'red', 'green', 'yellow'])
+        plt.title(title)
+        plt.axis('equal')
+
+        img_buffer = io.BytesIO()
+        plt.savefig(img_buffer, format='png', dpi=300, bbox_inches='tight')
+        plt.close()
+
+        img_buffer.seek(0)
+        img = PILImage.open(img_buffer)
+        return img
+
+    except Exception as e:
+        print(f"Error generating pie chart: {str(e)}")
+
+@tool
+def generate_box_plot(data: list[list[float]], title: str = 'Box Plot', y_label: str = 'Values') -> PILImage:
+    """Generates a box plot from the provided data.
+    
+    Args:
+        data: A list of numerical lists representing different categories.
+        title: Title for the box plot.
+        y_label: Label for the y-axis.
+    Returns:
+        Aan image of the box plot.
+    """
+    try:
+        plt.boxplot(data)
+        plt.title(title)
+        plt.ylabel(y_label)
+
+        img_buffer = io.BytesIO()
+        plt.savefig(img_buffer, format='png', dpi=300, bbox_inches='tight')
+        plt.close()
+
+        img_buffer.seek(0)
+        img = PILImage.open(img_buffer)
+        return img
+
+    except Exception as e:
+        print(f"Error generating box plot: {str(e)}")
+
 
 
 
@@ -185,7 +269,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,generate_bar_chart,generate_scatter_plot_with_labels,generate_line_plot], ## add your tools here (don't remove final answer)
+    tools=[final_answer,generate_bar_chart,generate_scatter_plot_with_labels,generate_line_plot,generate_histogram,generate_pie_chart,generate_box_plot], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,