Update app.py

app.py

@@ -4,7 +4,7 @@ import logging, os, re, sys, threading
 
 from datasets import load_dataset
 from dotenv import load_dotenv, find_dotenv
-from custom_utils import process_records, connect_to_database, setup_vector_search_index, handle_user_query
+from custom_utils import process_records, connect_to_database, handle_user_query
 
 from pydantic import BaseModel
 from typing import Optional
@@ -39,6 +39,7 @@ def invoke(openai_api_key, prompt, rag_option):
         Include a reason as to why you've chosen your selection.
         """
 
+        """
         print("111")
         dataset = load_dataset("MongoDB/airbnb_embeddings", streaming=True, split="train")
         #dataset = dataset.take(100)
@@ -61,7 +62,9 @@ def invoke(openai_api_key, prompt, rag_option):
         print("666")
         # Not available in free tier
         #setup_vector_search_index(collection=collection)
+        """
 
+        """
         print("777")
         search_path = "address.country"
 
@@ -76,11 +79,11 @@ def invoke(openai_api_key, prompt, rag_option):
 
         print("999")
         additional_stages = [match_stage]
+        """
 
         print("000")
-        result = handle_user_query(openai_api_key, query, db, collection, additional_stages)
-
-        return result
+        #result = handle_user_query(openai_api_key, query, db, collection, additional_stages)
+        return handle_user_query(openai_api_key, query, db, collection)
 
 gr.close_all()