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 'Question' in item and 'Answer' in item for item in dataset):
+    if not all(isinstance(item, dict) and 'input' in item and 'response' 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("Question", "").lower().strip() for item in dataset]
-dataset_answers = [item.get("Answer", "") for item in dataset]
+dataset_questions = [item.get("input", "").lower().strip() for item in dataset]
+dataset_answers = [item.get("response", "") for item in dataset]
 dataset_embeddings = similarity_model.encode(dataset_questions, convert_to_tensor=True)
 
 # Save unmatched queries to Hugging Face