Update modules/graph_plotter.py

modules/graph_plotter.py

@@ -22,19 +22,19 @@ def plot_equation(equation):
     plt.xlim(-5, 5)
     plt.ylim(-5, 5)
     plt.gca().set_aspect('equal', adjustable='box')
-    random_num = random.randint(1, 100)
-    # Save plot to a PIL image
+    random_num = random.randint(1, 1000)
     fig = plt.gcf()
     fig.canvas.draw()
     image = Image.frombytes('RGB', fig.canvas.get_width_height(), fig.canvas.tostring_rgb())
 
     # Close the plot
     plt.close()
-
+    img_name = f"graph{random_num}.jpg"
+    image.save(img_name)
     # Returning the image
     output_dict = {
-        "llm_output": f"Done. The graph is plotted and displayed. The graph is stored as variable {random_num}",
-        "real_output": {"display": image, "name": random_num, "metadata": equation},
+        "llm_output": f"Done. The graph is plotted, displayed, and saved. The graph is stored as pil image variable {random_num}. It is also saved as {img_name}",
+        "real_output": {"display": image, "name": random_num, "metadata": equation, "file_name": img_name, "saved": True}
         "type": "image",
     }
     return output_dict