Update rag.py

rag.py

@@ -31,15 +31,15 @@ GREETINGS = [
 try:
     with open('dataset.json', 'r') as f:
         dataset = json.load(f)
-    if not all(isinstance(item, dict) and 'input' in item and 'response' in item for item in dataset):
+    if not all(isinstance(item, dict) and 'Question' in item and 'Answer' in item for item in dataset):
         raise ValueError("Invalid dataset structure")
 except Exception as e:
     print(f"Error loading dataset: {e}")
     dataset = []
 
 # Precompute embeddings
-dataset_questions = [item.get("input", "").lower().strip() for item in dataset]
-dataset_answers = [item.get("response", "") for item in dataset]
+dataset_questions = [item.get("Question", "").lower().strip() for item in dataset]
+dataset_answers = [item.get("Answer", "") for item in dataset]
 dataset_embeddings = similarity_model.encode(dataset_questions, convert_to_tensor=True)
 
 # Save unmatched queries to Hugging Face
@@ -110,7 +110,7 @@ def get_best_answer(user_input):
 
     if best_score >= 0.65:
         original_answer = dataset_answers[best_match_idx]
-        prompt = f"""You are an official assistant for the University of Education Lahore.
+        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.
@@ -125,7 +125,7 @@ DO NOT add any new or extra information. ONLY rephrase and improve the clarity a
 ### Rephrased Answer:
 """
     else:
-        prompt = f"""As an official assistant for University of Education Lahore, provide a helpful response:
+        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.