Create graph_tool.py

graph_tool.py

@@ -0,0 +1,48 @@
+import base64
+from io import BytesIO
+import matplotlib.pyplot as plt
+import numpy as np
+
+def generate_plot(data, labels, plot_type, title, x_label="", y_label=""):
+    """
+    Generates a plot (bar, line, or pie) and returns it as a Base64-encoded string.
+    
+    Args:
+        data (dict): A dictionary where keys are labels and values are numerical data.
+        labels (list): The labels for the data points.
+        plot_type (str): The type of plot to generate ('bar', 'line', or 'pie').
+        title (str): The title of the plot.
+        x_label (str): The label for the x-axis.
+        y_label (str): The label for the y-axis.
+        
+    Returns:
+        str: A Base64-encoded string of the plot image.
+    """
+    fig, ax = plt.subplots()
+
+    # Extract keys and values from the data dictionary
+    x_data = list(data.keys())
+    y_data = list(data.values())
+
+    if plot_type == 'bar':
+        ax.bar(x_data, y_data, tick_label=labels)
+    elif plot_type == 'line':
+        ax.plot(x_data, y_data)
+    elif plot_type == 'pie':
+        ax.pie(y_data, labels=x_data, autopct='%1.1f%%', startangle=90)
+        ax.axis('equal') # Equal aspect ratio ensures that pie is drawn as a circle.
+    else:
+        raise ValueError("Invalid plot_type. Choose 'bar', 'line', or 'pie'.")
+
+    ax.set_title(title)
+    ax.set_xlabel(x_label)
+    ax.set_ylabel(y_label)
+
+    # Save plot to a BytesIO buffer in memory instead of to a file
+    buf = BytesIO()
+    plt.savefig(buf, format='png', bbox_inches='tight')
+    plt.close(fig)  # Close the plot to free up memory
+    
+    # Encode the image data to a Base64 string
+    img_base64 = base64.b64encode(buf.getvalue()).decode('utf-8')
+    return f'<img src="data:image/png;base64,{img_base64}"/>'