Update weaviate_utils.py

weaviate_utils.py

@@ -4,6 +4,31 @@ from weaviate.embedded import EmbeddedOptions
 from weaviate import Client
 import pandas as pd  # <-- Add this import
 from io import StringIO  # <-- Add this import
+import pandas as pd
+
+def hybrid_search_weaviate(client, selected_class, query):
+    """
+    Perform a hybrid search on Weaviate using the provided class and query.
+    Return the results as a list of dictionaries.
+    """
+    # Perform the hybrid search
+    results = client.data_object.get_by_search(
+        className=selected_class,
+        query=query,
+        filters=None,  # No additional filters for now
+        limit=100  # Limit to 100 results for now
+    )
+    
+    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)
+    table = [df.columns.tolist()] + df.values.tolist()
+    return table
 
 def initialize_weaviate_client():
     return weaviate.Client(embedded_options=EmbeddedOptions())