Update app.py

app.py

@@ -53,13 +53,33 @@ class StopOnTokens(StoppingCriteria):
 # Medical Keywords and Validation
 # =============================
 MEDICAL_KEYWORDS = [
-    "pain", "ache", "symptom", "disease", "condition", "treatment", "medication",
+    # Symptoms
+    "pain", "ache", "symptom", "hurt", "sore", "discomfort", "suffering",
+    # Common conditions
     "fever", "cough", "cold", "flu", "infection", "allergy", "diabetes", "pressure",
-    "heart", "stomach", "head", "back", "chest", "throat", "doctor", "hospital",
-    "sick", "ill", "health", "medical", "medicine", "prescription", "diagnosis",
-    "injury", "wound", "blood", "breathing", "sleep", "tired", "fatigue", "stress",
-    "anxiety", "depression", "mental", "diet", "nutrition", "exercise", "weight",
-    "vitamin", "supplement", "cancer", "tumor", "surgery", "clinic", "emergency"
+    "asthma", "migraine", "nausea", "vomit", "diarrhea", "constipation",
+    # Body parts
+    "heart", "stomach", "head", "back", "chest", "throat", "lung", "kidney",
+    "liver", "brain", "skin", "eye", "ear", "nose", "tooth", "teeth", "joint",
+    "muscle", "bone", "neck", "shoulder", "knee", "ankle", "foot", "hand",
+    # Medical terms
+    "doctor", "hospital", "clinic", "emergency", "ambulance", "medication",
+    "medicine", "prescription", "diagnosis", "treatment", "therapy", "cure",
+    "sick", "ill", "disease", "condition", "disorder", "syndrome",
+    # Injuries
+    "injury", "wound", "cut", "bruise", "fracture", "sprain", "burn", "bleed",
+    # Vitals and tests
+    "blood", "pressure", "temperature", "pulse", "breathing", "test", "scan",
+    # Mental health
+    "stress", "anxiety", "depression", "mental", "sleep", "insomnia", "tired",
+    "fatigue", "exhausted", "mood", "panic", "worry",
+    # Lifestyle/wellness
+    "diet", "nutrition", "exercise", "weight", "vitamin", "supplement", "healthy",
+    "wellness", "fitness", "eating", "appetite",
+    # Serious conditions
+    "cancer", "tumor", "surgery", "stroke", "attack", "seizure", "diabetic",
+    # Questions about health
+    "health", "medical", "feel", "feeling", "comfortable", "uncomfortable"
 ]
 
 CASUAL_ONLY_PATTERNS = [
@@ -77,8 +97,12 @@ def is_medical_query(message):
         if keyword in message_lower:
             return True
     
-    # If message is longer than 10 words, assume it might be medical
-    if len(message.split()) > 10:
+    # Check for question words combined with longer messages (might be medical)
+    question_words = ["what", "how", "why", "when", "where", "can", "should", "is", "are", "do", "does"]
+    has_question = any(q in message_lower.split()[:3] for q in question_words)
+    
+    # If it has a question word and is longer than 5 words, might be medical
+    if has_question and len(message.split()) > 5:
         return True
     
     return False
@@ -104,9 +128,9 @@ def is_only_greeting(message):
 def get_response(user_input, history_context):
     """Generate response from ChatDoctor model"""
     
-    # If it's only a greeting without medical context, redirect immediately
-    if is_only_greeting(user_input) and not is_medical_query(user_input):
-        return "Hello! I'm ChatDoctor, an AI medical assistant. I'm here specifically to help with health and medical concerns. Please tell me about any symptoms, health questions, or medical issues you'd like to discuss."
+    # STRICT FILTERING: Only allow medical queries to reach the model
+    if not is_medical_query(user_input):
+        return "Hello! I'm ChatDoctor, an AI medical assistant specialized in health and medical topics. I can help you with:\n\n• Symptoms and health concerns\n• Medical conditions and treatments\n• General health advice\n• Wellness and prevention\n\nPlease describe any health-related symptoms or medical questions you have, and I'll do my best to assist you."
     
     human_invitation = "Patient: "
     doctor_invitation = "ChatDoctor: "