Update modules/graph_plotter.py

modules/graph_plotter.py

@@ -33,8 +33,8 @@ def plot_equation(equation):
     image.save(img_name)
     # Returning the image
     output_dict = {
-        "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}
+        "llm_output": f"Done. The graph is plotted and displayed.",
+        "display": {"html": [f'<img src="{img_name}" alt="Plotted graph">'], "metadata": equation}
         "type": "image",
     }
     return output_dict