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