Update app.py

app.py

@@ -21,16 +21,16 @@ def get_restaurants(search, location, meters):
         return None, None, None, None
 
     try:
-        with st.status("Geo search data..."):
+        with st.status("Search data..."):
             newTrip, pre_agg = pre_aggregate_meters(restaurants_collection, location, meters)
 
-        response = openai_client.embeddings.create(
+            response = openai_client.embeddings.create(
             input=search,
             model="text-embedding-3-small",
             dimensions=256
-        )
+           )
 
-        vectorQuery = {
+            vectorQuery = {
             "$vectorSearch": {
                 "index": "vector_index",
                 "queryVector": response.data[0].embedding,
@@ -40,10 +40,11 @@ def get_restaurants(search, location, meters):
                 "filter": {"searchTrip": newTrip}
             }
         }
+             st.write("Vector query")
+             restaurant_docs = list(trips_collection.aggregate([vectorQuery, {"$project": {"_id": 0, "embedding": 0}}]))
 
-        restaurant_docs = list(trips_collection.aggregate([vectorQuery, {"$project": {"_id": 0, "embedding": 0}}]))
-
-        stream_response = openai_client.chat.completions.create(
+             st.write("RAG...")
+             stream_response = openai_client.chat.completions.create(
             model="gpt-4o",
             messages=[
                 {"role": "system", "content": "You are a helpful restaurant assistant. Answer shortly and quickly. You will get a context if the context is not relevant to the user query please address that and not provide by default the restaurants as is."},