Update chart generation data processing

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]