updates for link generation

functions/chart_functions.py

@@ -33,7 +33,7 @@ def table_generation_func(data: List[str], session_hash):
     csv_path = f'{dir_path}/data.csv'
     df.to_csv(csv_path)
     download_path = f'gradio_api/file/temp/{session_hash}/data.csv'
-    html_table = df.to_html() + f'<p>Download as a <a href="{download_path}">CSV</a></p>'
+    html_table = df.to_html() + f'<p>Download as a <a href="{download_path}">CSV file</a></p>'
     print(html_table)
 
     return {"reply": html_table}

functions/chat_functions.py

@@ -61,7 +61,9 @@ def chatbot_with_fc(message, history, session_hash):
     else:
         messages = [
             ChatMessage.from_system(
-                "You are a helpful and knowledgeable agent who has access to an SQLite database which has a table called 'data_source'. You also have access to a chart API that uses chart.js dictionaries formatted as a string to generate charts and graphs."
+                """You are a helpful and knowledgeable agent who has access to an SQLite database which has a table called 'data_source'. 
+                You also have access to a chart API that uses chart.js dictionaries formatted as a string to generate charts and graphs.
+                You also have access to a function, called table_generation_func, that builds table formatted html and generates a link to download as CSV."""
             )
         ]
         messages.append(ChatMessage.from_user(message))

tools.py

@@ -37,7 +37,10 @@ def tools_call(session_hash):
         "type": "function",
             "function": {
                 "name": "chart_generation_func",
-                "description": f"This a chart generation tool useful to generate charts and graphs from queried data from our SQL table called 'data_source with the following Columns: {columns}. Returns an iframe string which will be displayed inline in our chat window. Do not edit the string returned from the chart_generation_func function in any way and display it fully to the user in the chat window. You can add your own text supplementary to it for context if desired.",
+                "description": f"""This a chart generation tool useful to generate charts and graphs from queried data from our SQL table called 'data_source 
+                with the following Columns: {columns}. Returns an iframe string which will be displayed inline in our chat window. Do not edit the string returned 
+                from the chart_generation_func function in any way and display it fully to the user in the chat window. You can add your own text supplementary 
+                to it for context if desired.""",
                 "parameters": {
                     "type": "object",
                     "properties": {
@@ -63,8 +66,7 @@ def tools_call(session_hash):
                 '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. 
-                You can add your own text next to the returned string for context if desired.""",
+                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.""",
                 "parameters": {
                     "type": "object",
                     "properties": {