Update data processing for chart object

functions/chart_functions.py

@@ -21,19 +21,20 @@ 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 = {}        
+        layout_dict = {}       
 
         if isinstance(data, list):
-           data_obj = data[0]
-        else:
-           data_obj = data  
-
-        if isinstance(data_obj, str):
-           data_dict = ast.literal_eval(data_obj)
+           data_list = data
         else:
-           data_dict = data_obj
+           data_list.append(data)
 
-        data_list.append(data_dict)
+        data_dict_list = [] 
+        for data_obj in data_list:   
+         if isinstance(data_obj, str):
+            data_dict = ast.literal_eval(data_obj)
+         else:
+            data_dict = data_obj
+         data_dict_list.append(data_dict)
 
         if isinstance(layout, list):
            layout_obj = layout[0]
@@ -46,7 +47,7 @@ def chart_generation_func(data: List[dict], session_hash: str, layout: Dict[str,
            layout_dict = layout_obj
            
 
-        fig = dict({"data": data_list,
+        fig = dict({"data": data_dict_list,
                     "layout": layout_dict})
         pio.write_html(fig, chart_path, full_html=False)