Update app.py

app.py

@@ -31,30 +31,12 @@ def invoke(openai_api_key, prompt, rag_option):
         os.environ["OPENAI_API_KEY"] = openai_api_key
 
         ###
-        dataset = load_dataset("MongoDB/airbnb_embeddings", streaming=True, split="train")
-        dataset = dataset.take(100)
-        dataset_df = pd.DataFrame(dataset)
-        dataset_df.head(5)
-        print("Columns:", dataset_df.columns)
-        records = dataset_df.to_dict(orient='records')
-        # To handle catch `NaT` values
-        for record in records:
-            for key, value in record.items():
-                # Check if the value is list-like; if so, process each element.
-                if isinstance(value, list):
-                    processed_list = [None if pd.isnull(v) else v for v in value]
-                    record[key] = processed_list
-                # For scalar values, continue as before.
-                else:
-                    if pd.isnull(value):
-                        record[key] = None
-        try:
-          # Convert each dictionary to a Movie instance
-          listings = [Listing(**record).dict() for record in records]
-          # Get an overview of a single datapoint
-          print(listings[0].keys())
-        except ValidationError as e:
-          print(e)
+        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.
+        """
+        handle_user_query(query, db, collection)
         ###
         
         """