Upload app.py

app.py

@@ -46,13 +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],
+            "days": dataset['test']['days'][idx],
             "reference_information": dataset['test']['reference_information'][idx]
         }
         results.append(trip_plan)
         
     while len(results) < 3:
-        results.append({"dest": "No trip plan found", "day":"", "reference_information": ""})
+        results.append({"dest": "No trip plan found", "days":"", "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}."
@@ -89,7 +89,7 @@ def format_trip_plan(trip):
     # Formats the recommended trips with markdown
     if not trip:
         return "### No similar trip plan found."
-    return f"### {trip['day']}-days trip to {trip['dest'].upper()}\n**Suggested Plan:**\n{trip['reference_information']}"
+    return f"### {trip['days']}-days trip to {trip['dest'].upper()}\n**Suggested Plan:**\n{trip['reference_information']}"
 
 def format_generated_trip(trip_text):
     return trip_text