Update app.py

app.py

@@ -101,8 +101,8 @@ def chatbot_response(query, chat_history):
     similarity_scores = cosine_similarity(query_embedding.cpu().numpy(), question_embeddings.cpu().numpy())[0]
     best_idx = int(np.argmax(similarity_scores))
     best_score = similarity_scores[best_idx]
-    matched_q = questions[best_idx]
-    matched_a = answers[best_idx]
+    matched_item = dataset[best_idx]  # Changed this to get full entry including month/year
+    matched_a = matched_item.get("answer", "")
 
     if best_score < 0.4:
         response = "Sorry, but the PUP handbook does not contain such information."
@@ -119,11 +119,15 @@ def chatbot_response(query, chat_history):
                 print(f"[ERROR] HF inference failed: {e}")
                 response = f"(Fallback) {matched_a}"
         else:
-            response = f"According to the university handbook, {matched_a}"
+            if "month" in matched_item and "year" in matched_item:
+                response = f"As of {matched_item['month']}, {matched_item['year']}, {matched_a}"
+            else:
+                response = f"According to 2019 Proposed PUP Handbook, {matched_a}"
 
     chat_history.append((query, response.strip()))
     return "", chat_history, gr.update(visible=True)
 
+
 def record_feedback(feedback, chat_history):
     global feedback_embeddings, feedback_questions
     if chat_history: