Upload app.py

app.py

@@ -46,12 +46,13 @@ def get_recommendations_and_generate(query_text, k=3):
         idx = int(idx_numpy)
         trip_plan = {
             "dest": dataset['test']['dest'][idx],
+            "day": dataset['test']['day'][idx],
             "reference_information": dataset['test']['reference_information'][idx]
         }
         results.append(trip_plan)
         
     while len(results) < 3:
-        results.append({"dest": "No trip plan found", "reference_information": ""})
+        results.append({"dest": "No trip plan found", "day":"", "reference_information": ""})
         
     # 2. Create a prompt for the generative model
     prompt = f"Write a complete travel plan that includes a title and a day-by-day itinerary. The trip must be about: {query_text}."