UPDATE: charts

functions.py

@@ -29,6 +29,7 @@ from urllib.parse import urlparse, urljoin
 from supabase import create_client
 from dotenv import load_dotenv
 import os
+import base64
 import time
 import requests
 
@@ -322,4 +323,9 @@ def analyzeData(query, dataframe):
     llm = ChatGroq(name = "gemma2-9b-it")
     df = SmartDataframe(dataframe, config = {"llm": llm, "verbose": False})
     response = df.chat(query)
-    return response
+    if os.path.isfile(response):
+        with open(response, "rb") as file:
+            b64string = base64.b64encode(file.read()).decode("utf-8")
+        return f"data:image/png;base64,{b64string}"
+    else:
+        return response