updates to chart generation

__init__.py

@@ -1,3 +0,0 @@
-from .app import data_url
-
-__all__ = ["data_url"]

functions/chart_functions.py

@@ -22,13 +22,18 @@ def chart_generation_func(data: List[dict], session_hash: str, layout: Dict[str,
         #Processing data to account for variation from LLM
         data_list = []
         layout_dict = {}        
+
         if isinstance(data, list):
-            data_list = data
+           data_obj = data[0]
+        else:
+           data_obj = data  
+
+        if isinstance(data_obj, str):
+           data_dict = ast.literal_eval(data_obj)
         else:
-            data_list.append(data)
+           data_dict = data_obj
 
-        if isinstance(data[0], str): 
-           data_list[0] = ast.literal_eval(data_list[0])
+        data_list.append(data_dict)
 
         if isinstance(layout, list):
            layout_obj = layout[0]
@@ -53,6 +58,7 @@ def chart_generation_func(data: List[dict], session_hash: str, layout: Dict[str,
     
     except Exception as e:
       print("CHART ERROR")
+      print(e)
       reply = f"""There was an error generating the Plotly Chart from {data} and {layout}
               The error is {e},
               You should probably try again.
@@ -89,6 +95,7 @@ def table_generation_func(data: List[dict], session_hash):
     
     except Exception as e:
       print("TABLE ERROR")
+      print(e)
       reply = f"""There was an error generating the Pandas DataFrame table from {data}
               The error is {e},
               You should probably try again.