Use Root URL

functions/chart_functions.py

@@ -2,6 +2,12 @@ from typing import List
 from quickchart import QuickChart
 import pandas as pd
 from utils import TEMP_DIR
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+root_url = os.getenv("ROOT_URL")
 
 def chart_generation_func(queries: List[str], session_hash):
     print("CHART GENERATION")
@@ -32,7 +38,7 @@ def table_generation_func(data: List[str], session_hash):
     df = pd.DataFrame(data)
     csv_path = f'{dir_path}/data.csv'
     df.to_csv(csv_path)
-    download_path = f'gradio_api/file/temp/{session_hash}/data.csv'
+    download_path = f'{root_url}/gradio_api/file/temp/{session_hash}/data.csv'
     html_table = df.to_html() + f'<p>Download as a <a target="_blank" href="{download_path}">CSV file</a></p>'
     print(html_table)
 

tools.py

@@ -65,8 +65,8 @@ def tools_call(session_hash):
                 "description": f"""This an table generation tool useful to format data as a table from queried data from our SQL table called 
                 'data_source with the following Columns: {columns}. Returns an html string generated from the pandas library and pandas.to_html() 
                 function which will be displayed inline in our chat window. There will also be a link to download the CSV included in the HTML string. 
-                Do not change or edit this link in any way. Do not edit the HTML returned by the function in any way.
-                Do not edit the string returned from the table_generation_func function in any way and display it fully to the user in the chat window.""",
+                This link should open in a new window. Do not edit the string returned by the function in any way when displaying to the user, as the user needs 
+                all of the information returned by the function in it's exact state.""",
                 "parameters": {
                     "type": "object",
                     "properties": {