fix search_data data type

Pinyu Su · Pinyu Su · commit c86a171331da · 2024-09-19T16:36:11.000+08:00
diff --git a/application/utils/llm.py b/application/utils/llm.py
@@ -355,6 +355,7 @@ def data_visualization(model_id, search_box, search_data, prompt_map):
     else:
         all_columns_data = search_data
         columns = all_columns_data[0]
+        search_data = pd.DataFrame(search_data[1:], columns=search_data[0])
     all_columns_data = convert_timestamps_to_str(all_columns_data)
     try:
         if len(all_columns_data) < 1:
