Update app.py

app.py

@@ -41,20 +41,26 @@ def invoke(openai_api_key, prompt, rag_option):
         and not too far from resturants, can you recommend a place? 
         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)
+        print("222")
         # Convert the dataset to a pandas dataframe
         dataset_df = pd.DataFrame(dataset)
         dataset_df.head(5)
         print("Columns:", dataset_df.columns)
         
         listings = process_records(dataset_df)
-        
+        print("333")
+
+        print("444")        
         db, collection = connect_to_database()
         collection.delete_many({})
         collection.insert_many(listings)
         print("Data ingestion into MongoDB completed")
-
+        print("555")
+        
+        """
         setup_vector_search_index(collection=collection)
 
         search_path = "address.country"
@@ -69,12 +75,9 @@ def invoke(openai_api_key, prompt, rag_option):
         
         additional_stages = [match_stage]
 
-        query = """
-        I want to stay in a place that's warm and friendly, 
-        and not too far from resturants, can you recommend a place? 
-        Include a reason as to why you've chosen your selection"
-        """
         result = handle_user_query(openai_api_key, query, db, collection, additional_stages)
+        """
+        result = "TODO"
         ###
 
         return result