Update weaviate_utils.py

weaviate_utils.py

@@ -27,13 +27,17 @@ def hybrid_search_weaviate(client, selected_class, query):
     return results
 
 
-
 def convert_to_tapas_format(data):
     """
     Convert the list of dictionaries (from Weaviate) into the format TAPAS expects.
     Return the table as a list of lists.
     """
-    df = pd.DataFrame(data)
+    # Extract the data objects from the results
+    data_objects = data.get('data', {}).get('Get', {}).get('Things', [])
+    
+    # Convert the data objects into a DataFrame
+    df = pd.DataFrame([obj['thing'] for obj in data_objects])
+    
     table = [df.columns.tolist()] + df.values.tolist()
     return table