Update app.py

app.py

@@ -36,7 +36,7 @@ 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"]
+    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"] # Extended healthcare keywords
 
     is_epilepsy_related = any(k in tokens for k in epilepsy_keywords)
@@ -112,27 +112,40 @@ def generate_response(user_query, chat_history):
 
         return f"**NeuroGuard:** ✅ **Analysis:**\n{pubmedbert_insights}\n\n**Response:**\n{model_response}"
 
-    # If Healthcare Related but Not Epilepsy - Provide General Wellness Tips
+    # If Healthcare Related but Not Epilepsy - Provide General Wellness Tips and Basic Remedies
     elif is_healthcare_related:
+        remedy_suggestions = {
+            "headache": "For headaches, you can try resting in a quiet, dark room, staying hydrated, and applying a cold compress to your forehead. Over-the-counter pain relievers like ibuprofen or acetaminophen may also help.",
+            "pain": "For general pain, rest, gentle stretching, and applying heat or ice to the affected area can provide relief. Over-the-counter pain relievers might also be beneficial. ",
+            "legs pain": "For leg pain, try elevating your legs, applying heat or ice, and gentle stretching exercises. Staying hydrated and ensuring good circulation can also help. "
+            # Add more specific remedies as needed
+        }
+        suggested_remedy = "For general wellness:\n" # Default if no specific remedy keyword is found
+        for keyword, remedy in remedy_suggestions.items():
+            if keyword in corrected_query: # Check corrected query for keywords
+                suggested_remedy = remedy + "\n\n" # Use specific remedy if keyword found
+                break # Use only the first matching remedy for now
+
         general_health_tips = (
-            "For general health and well-being:\n"
             "- 💧 Stay hydrated by drinking plenty of water throughout the day.\n"
             "- 🍎 Maintain a balanced diet rich in fruits, vegetables, and whole grains.\n"
             "- 🚶‍♀️ Incorporate regular physical activity into your daily routine.\n"
             "- 😴 Ensure you get adequate sleep to allow your body to rest and recover.\n"
             "- 🧘 Practice stress-reducing activities such as deep breathing or meditation.\n"
-            "- 🩺 **Important:** These tips are for general wellness. Always consult a healthcare professional for any specific health concerns or before making significant changes to your health regimen."
-        )
-        return (
-            f"**NeuroGuard:** 🩺 It sounds like your question '{user_query}' is about general health. While I specialize in epilepsy, here are some general wellness tips that might be helpful:\n\n" # Use original user_query in response
-            f"{general_health_tips}"
+            "- 🩺 **Important:** These tips are for general wellness and informational purposes only, and are not medical advice. Always consult a healthcare professional for any specific health concerns or before making significant changes to your health regimen."
         )
 
+        response_content = f"**NeuroGuard:** 🩺 It seems your question '{user_query}' is about general health. While I focus on epilepsy, here's some information that might help:\n\n" # Use original user_query
+        response_content += suggested_remedy
+        response_content += general_health_tips # Always include general tips
+
+        return response_content
+
 
     # If General Query (Not Healthcare or Epilepsy Related) - Provide a different response
     elif is_general:
         return (
-            f"**NeuroGuard:** 💡  My expertise is focused on epilepsy and general health.  For topics outside of these areas, like '{user_query}', I may not be the best resource. \n\n"
+            f"**NeuroGuard:** 💡  My expertise is focused on epilepsy and general health.  For topics outside of these areas, like '{user_query}', I may not be the best resource. \n\n" # Use original user_query
             "**Tip:** For general knowledge questions, you might find better answers using a general search engine or a chatbot trained on a broader range of topics."
         )