From 897d7f46d115af2b22766af8e9a97b6da82f4fcd Mon Sep 17 00:00:00 2001
From: Gabriele Venturi <lele.venturi@gmail.com>
Date: Fri, 15 Dec 2023 00:32:53 +0100
Subject: [PATCH] fix: return path when plotting a chart (#816)

---
 pandasai/responses/response_parser.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pandasai/responses/response_parser.py b/pandasai/responses/response_parser.py
index a60ec3436..8f7d1802b 100644
--- a/pandasai/responses/response_parser.py
+++ b/pandasai/responses/response_parser.py
@@ -95,6 +95,8 @@ def format_plot(self, result: dict) -> Any:
         with Image.open(result["value"]) as img:
             img.show()
 
+        return result["value"]
+
     def format_other(self, result) -> Any:
         """
         Returns the result generated against a user query other than dataframes