init

app.py

@@ -41,20 +41,30 @@ def get_restaurants(search, location, meters):
         model="gpt-3.5-turbo",
         messages=[
             {"role": "system", "content": "You are a helpful restaurant assistant."},
-            { "role": "user", "content": f"Find me the 2 best restaurant and why based on {search} and  {restaurant_docs}. explain trades offs and why I should go to each one."}
+            { "role": "user", "content": f"Find me the 2 best restaurant and why based on {search} and  {restaurant_docs}. explain trades offs and why I should go to each one. You can mention the third option as a possible alternative."}
         ]
         )
 
     trips_collection.delete_many({"searchTrip": newTrip})
 
-    return chat_response.choices[0].message.content
+    first_restaurant = restaurant_docs[0]['restaurant_id']
+    second_restaurant = restaurant_docs[1]['restaurant_id']
+    third_restaurant = restaurant_docs[2]['restaurant_id']
+
+    if (first_restaurant or second_restaurant or third_restaurant) is None:
+        return "No restaurants found", '<iframe style="background: #FFFFFF;border: none;border-radius: 2px;box-shadow: 0 2px 10px 0 rgba(70, 76, 79, .2);" width="640" height="480" src="https://charts.mongodb.com/charts-paveldev-wiumf/embed/charts?id=65c24b0c-2215-4e6f-829c-f484dfd8a90c&filter={\'restaurant_id\':\'\'}&maxDataAge=3600&theme=light&autoRefresh=true"></iframe>'
+    else:
+        restaurant_string = f"\'{first_restaurant}\', \'{second_restaurant}\', \'{third_restaurant}\'"
+    iframe = '<iframe style="background: #FFFFFF;border: none;border-radius: 2px;box-shadow: 0 2px 10px 0 rgba(70, 76, 79, .2);" width="640" height="480" src="https://charts.mongodb.com/charts-paveldev-wiumf/embed/charts?id=65c24b0c-2215-4e6f-829c-f484dfd8a90c&filter={\'restaurant_id\':{$in:['  + restaurant_string  + ']}}&maxDataAge=3600&theme=light&autoRefresh=true"></iframe>'
+
+    return chat_response.choices[0].message.content, iframe
     
 
 def pre_aggregate_meters(location, meters):
 
     tripId = ObjectId()
 
-    restaurants_collection.aggregate([
+    result = restaurants_collection.aggregate([
         {
             "$geoNear": {
             "near": location,
@@ -76,6 +86,8 @@ def pre_aggregate_meters(location, meters):
         }   
     ]);
 
+    print(result)
+    print(trips_collection.count_documents({"searchTrip": tripId}));
     sleep(10)
 
     return tripId
@@ -87,7 +99,7 @@ with gr.Blocks() as demo:
     # MongoDB's Vector Restaurant planner 
     Start typing below to see the results
     """)
-    gr.HTML(value='<iframe style="background: #FFFFFF;border: none;border-radius: 2px;box-shadow: 0 2px 10px 0 rgba(70, 76, 79, .2);" width="640" height="480" src="https://charts.mongodb.com/charts-paveldev-wiumf/embed/charts?id=65c24b0c-2215-4e6f-829c-f484dfd8a90c&maxDataAge=3600&theme=light&autoRefresh=true"></iframe>')
+    #gr.HTML(value='<iframe style="background: #FFFFFF;border: none;border-radius: 2px;box-shadow: 0 2px 10px 0 rgba(70, 76, 79, .2);" width="640" height="480" src="https://charts.mongodb.com/charts-paveldev-wiumf/embed/charts?id=65c24b0c-2215-4e6f-829c-f484dfd8a90c&maxDataAge=3600&theme=light&autoRefresh=true"></iframe>')
 #    
     gr.Interface(
         get_restaurants,
@@ -111,7 +123,7 @@ with gr.Blocks() as demo:
             ]
         })], label="Location", info="What location you need?"),
         gr.Slider(minimum=500, maximum=10000, randomize=False, step=5, label="Radius in meters")],
-       gr.Textbox(label="MongoDB Vector Recommendations", placeholder="Results will be displayed here"),
+       [gr.Textbox(label="MongoDB Vector Recommendations", placeholder="Results will be displayed here"), "html"],
         
     )
     #radio.change(location_searched, loc, out)

iframe.html

@@ -0,0 +1 @@
+<iframe style="background: #FFFFFF;border: none;border-radius: 2px;box-shadow: 0 2px 10px 0 rgba(70, 76, 79, .2);" width="640" height="480" src="https://charts.mongodb.com/charts-paveldev-wiumf/embed/charts?id=65c24b0c-2215-4e6f-829c-f484dfd8a90c&filter={'restaurant_id':{$in:['50005104', '41166347', '41314543']}}&maxDataAge=3600&theme=light&autoRefresh=true"></iframe>