Add sample test cases knowledge base for demo validation

app.py

@@ -80,6 +80,91 @@ SYMPTOM_LABELS = {
     "chills": "Chills"
 }
 
+# ============================================================================
+# Sample Test Cases for Knowledge Base
+# ============================================================================
+
+SAMPLE_CASES = [
+    {
+        "name": "🚨 Case 1: Cardiac Emergency",
+        "symptoms": ["chest_pain", "shortness_of_breath", "dizziness"],
+        "expected": "Emergency - Cardiac",
+        "rationale": "Classic cardiac triad: chest pain + breathlessness + dizziness indicates possible MI/ACS",
+        "confidence": "~85-95%"
+    },
+    {
+        "name": "⚠️ Case 2: Respiratory Distress",
+        "symptoms": ["shortness_of_breath", "cough", "fever", "chills"],
+        "expected": "Urgent - Respiratory",
+        "rationale": "Fever + chills with respiratory symptoms suggests pneumonia or severe bronchitis",
+        "confidence": "~75-85%"
+    },
+    {
+        "name": "⚠️ Case 3: Systemic Infection",
+        "symptoms": ["fever", "chills", "fatigue", "muscle_ache"],
+        "expected": "Urgent - Infection",
+        "rationale": "High fever with systemic symptoms (chills, myalgia) indicates significant infection",
+        "confidence": "~80-90%"
+    },
+    {
+        "name": "📋 Case 4: Common Flu",
+        "symptoms": ["fever", "cough", "sore_throat", "runny_nose", "muscle_ache"],
+        "expected": "Standard - Flu-like",
+        "rationale": "Classic influenza presentation: upper respiratory + systemic symptoms",
+        "confidence": "~85-95%"
+    },
+    {
+        "name": "📋 Case 5: GI Complaint",
+        "symptoms": ["nausea", "abdominal_pain", "fever"],
+        "expected": "Standard - GI",
+        "rationale": "GI symptoms with low-grade fever suggests gastroenteritis or food poisoning",
+        "confidence": "~70-80%"
+    },
+    {
+        "name": "📋 Case 6: Musculoskeletal",
+        "symptoms": ["back_pain", "joint_pain", "muscle_ache"],
+        "expected": "Standard - Musculoskeletal",
+        "rationale": "Localized pain without systemic symptoms - routine orthopedic/rheumatologic eval",
+        "confidence": "~75-85%"
+    },
+    {
+        "name": "✅ Case 7: Minor Illness",
+        "symptoms": ["headache", "fatigue", "runny_nose"],
+        "expected": "Low Priority - Minor",
+        "rationale": "Mild symptoms without fever or respiratory distress - likely common cold",
+        "confidence": "~70-80%"
+    },
+]
+
+def get_sample_cases_markdown():
+    """Generate markdown table of sample cases."""
+    md = """
+## 📚 Sample Test Cases
+
+Use these cases to validate the system is working correctly. Click "Try This Case" to auto-fill symptoms.
+
+| Case | Symptoms | Expected Result | Confidence |
+|------|----------|-----------------|------------|
+"""
+    for case in SAMPLE_CASES:
+        symptoms_str = ", ".join([SYMPTOM_LABELS[s] for s in case["symptoms"]])
+        md += f"| {case['name']} | {symptoms_str} | **{case['expected']}** | {case['confidence']} |\n"
+    
+    md += """
+
+### How to Use
+1. Go to the **Clinical Triage** tab
+2. Select the symptoms listed for a case
+3. Click **Classify Patient**
+4. Verify the result matches the expected classification
+
+### Understanding Results
+- **Confidence** is the cosine similarity between patient vector and category prototype
+- Higher values (>80%) indicate strong match
+- Multiple categories may have similar scores for overlapping symptoms
+"""
+    return md
+
 # ============================================================================
 # UI Functions
 # ============================================================================
@@ -229,7 +314,29 @@ with gr.Blocks(title="Atomic VSA", theme=gr.themes.Soft()) as demo:
                 outputs=[result_output, scores_output, tech_output]
             )
         
-        # Tab 2: About
+        # Tab 2: Sample Cases (Knowledge Base)
+        with gr.TabItem("📚 Sample Cases"):
+            gr.Markdown(get_sample_cases_markdown())
+            
+            gr.Markdown("---\n### 🧪 Detailed Case Explanations")
+            
+            for case in SAMPLE_CASES:
+                with gr.Accordion(case["name"], open=False):
+                    gr.Markdown(f"""
+**Symptoms to Select:**
+{chr(10).join(['- ' + SYMPTOM_LABELS[s] for s in case['symptoms']])}
+
+**Expected Classification:** `{case['expected']}`
+
+**Clinical Rationale:** {case['rationale']}
+
+**Expected Confidence Range:** {case['confidence']}
+
+---
+*This pattern is encoded in the VSA prototype vectors built from clinical knowledge.*
+""")
+        
+        # Tab 3: About
         with gr.TabItem("📄 About"):
             gr.Markdown("""
             ## Atomic Vector Symbolic Architecture