Update app.py

app.py

@@ -36,12 +36,17 @@ index = faiss.IndexFlatL2(embedding_dim)
 # Function to Check Query Category
 def preprocess_query(query):
     tokens = query.lower().split()
-    epilepsy_keywords = ["seizure", "epilepsy", "convulsion", "neurology", "brain activity", "headache related to epilepsy"] # Added specific phrase
-    healthcare_keywords = ["headache", "fever", "blood pressure", "diabetes", "cough", "flu", "nutrition", "mental health", "pain", "legs", "body pain", "health", "medical", "symptoms", "treatment", "disease", "illness"] # Extended healthcare keywords - added "illness"
+    epilepsy_keywords = ["seizure", "epilepsy", "convulsion", "neurology", "brain activity", "headache related to epilepsy"]
+    healthcare_keywords = ["headache", "fever", "blood pressure", "diabetes", "cough", "flu", "nutrition", "mental health", "pain", "legs", "body pain", "health", "medical", "symptoms", "treatment", "disease", "illness"]
 
-    is_epilepsy_related = any(k in tokens for k in epilepsy_keywords)
-    is_healthcare_related = any(k in tokens for k in healthcare_keywords) and not is_epilepsy_related # Healthcare but NOT epilepsy
-    is_general = not is_epilepsy_related and not is_healthcare_related # Truly general queries
+    is_healthcare_related = False  # Initialize to False
+    for keyword in healthcare_keywords: # Explicitly check for healthcare keywords first
+        if keyword in tokens:
+            is_healthcare_related = True
+            break # Exit loop once a healthcare keyword is found
+
+    is_epilepsy_related = any(k in tokens for k in epilepsy_keywords) and not is_healthcare_related # Epilepsy check AFTER healthcare
+    is_general = not is_epilepsy_related and not is_healthcare_related
 
     print(f"Query: {query}") # Debugging prints
     print(f"Tokens: {tokens}")