Update rag.py

rag.py

@@ -6,6 +6,7 @@ import os
 import pandas as pd
 from datasets import load_dataset, Dataset
 from dotenv import load_dotenv
+import random
 
 # Load environment variables
 load_dotenv()
@@ -27,6 +28,14 @@ GREETINGS = [
     "hey there", "greetings"
 ]
 
+# Fixed rephrased unmatched query responses
+UNMATCHED_RESPONSES = [
+    "Thank you for your query. We’ve forwarded it to our support team and it will be added soon. In the meantime, you can visit the University of Education official website or reach out via the contact details below.\n\n📞 +92-42-99262231-33\n✉️ [email protected]\n🌐 https://ue.edu.pk",
+    "We’ve noted your question and it’s in queue for inclusion. For now, please check the University of Education website or contact the administration directly.\n\n📞 +92-42-99262231-33\n✉️ [email protected]\n🌐 https://ue.edu.pk",
+    "Your query has been recorded. We’ll update the system with relevant information shortly. Meanwhile, you can visit UE's official site or reach out using the details below:\n\n📞 +92-42-99262231-33\n✉️ [email protected]\n🌐 https://ue.edu.pk",
+    "We appreciate your question. It has been forwarded for further processing. Until it’s available here, feel free to visit the official UE website or use the contact options:\n\n📞 +92-42-99262231-33\n✉️ [email protected]\n🌐 https://ue.edu.pk"
+]
+
 # Load local dataset
 try:
     with open('dataset.json', 'r') as f:
@@ -100,32 +109,18 @@ def get_best_answer(user_input):
 
     if best_score < 0.65:
         manage_unmatched_queries(user_input)
+        return random.choice(UNMATCHED_RESPONSES)
 
-    if best_score >= 0.65:
-        original_answer = dataset_answers[best_match_idx]
-        prompt = f"""Name is UOE AI Assistant! You are an official assistant for the University of Education Lahore.
-
+    original_answer = dataset_answers[best_match_idx]
+    prompt = f"""Name is UOE AI Assistant! You are an official assistant for the University of Education Lahore.
 Rephrase the following official answer clearly and professionally. 
 Use structured formatting (like headings, bullet points, or numbered lists) where appropriate.
 DO NOT add any new or extra information. ONLY rephrase and improve the clarity and formatting of the original answer.
-
 ### Question:
 {user_input}
-
 ### Original Answer:
 {original_answer}
-
 ### Rephrased Answer:
-"""
-    else:
-        prompt = f"""Name is UOE AI Assistant! As an official assistant for University of Education Lahore, provide a helpful response:
-Include relevant details about university policies.
-If unsure, direct to official channels.
-
-### Question:
-{user_input}
-
-### Official Answer:
 """
 
     llm_response = query_groq_llm(prompt)
@@ -136,9 +131,4 @@ If unsure, direct to official channels.
                 return llm_response.split(marker)[-1].strip()
         return llm_response
     else:
-        return dataset_answers[best_match_idx] if best_score >= 0.65 else (
-            "For official information:\n"
-            "📞 +92-42-99262231-33\n"
-            "✉️ [email protected]\n"
-            "🌐 https://ue.edu.pk"
-        )
+        return dataset_answers[best_match_idx]