Update app.py

app.py

@@ -176,280 +176,281 @@ class PearlyBot:
             raise
             # Load your knowledge base content
     def _load_knowledge_base(self):
-    # Add validation and error handling for knowledge base loading
-        return {
+        """Load and validate knowledge base content"""
+        try:
+            knowledge_base = { {
             "triage_scenarios.txt": """Medical Triage Scenarios and Responses:
 
-EMERGENCY (999) SCENARIOS:
-1. Cardiovascular:
-- Chest pain/pressure
-- Heart attack symptoms
-- Irregular heartbeat with dizziness
-Response: Immediate 999 call, sit/lie down, chew aspirin if available
-
-2. Respiratory:
-- Severe breathing difficulty
-- Choking
-- Unable to speak full sentences
-Response: 999, sitting position, clear airway
-
-3. Neurological:
-- Stroke symptoms (FAST)
-- Seizures
-- Unconsciousness
-Response: 999, recovery position if unconscious
-
-4. Trauma:
-- Severe bleeding
-- Head injuries with confusion
-- Major burns
-Response: 999, apply direct pressure to bleeding
-
-URGENT CARE (111) SCENARIOS:
-1. Moderate Symptoms:
-- Persistent fever
-- Non-severe infections
-- Minor injuries
-Response: 111 contact, monitor symptoms
-
-2. Minor Emergencies:
-- Small cuts needing stitches
-- Sprains and strains
-- Mild allergic reactions
-Response: 111 or urgent care visit
-
-GP APPOINTMENT SCENARIOS:
-1. Routine Care:
-- Chronic condition review
-- Medication reviews
-- Non-urgent symptoms
-Response: Book routine GP appointment
-
-2. Preventive Care:
-- Vaccinations
-- Health screenings
-- Regular check-ups
-Response: Schedule with GP reception""",
-                "emergency_detection.txt": """Enhanced Emergency Detection Criteria:
-
-IMMEDIATE LIFE THREATS:
-1. Cardiac Symptoms:
-- Chest pain/pressure/tightness
-- Pain spreading to arms/jaw/neck
-- Sweating with nausea
-- Shortness of breath
-
-2. Breathing Problems:
-- Severe shortness of breath
-- Blue lips or face
-- Unable to complete sentences
-- Choking/airway blockage
-
-3. Neurological:
-- FAST (Face, Arms, Speech, Time)
-- Sudden confusion
-- Severe headache
-- Seizures
-- Loss of consciousness
-
-4. Severe Trauma:
-- Heavy bleeding
-- Deep wounds
-- Head injury with confusion
-- Severe burns
-- Broken bones with deformity
-
-5. Anaphylaxis:
-- Sudden swelling
-- Difficulty breathing
-- Rapid onset rash
-- Light-headedness
-
-URGENT BUT NOT IMMEDIATE:
-1. Moderate Symptoms:
-- Persistent fever
-- Dehydration
-- Non-severe infections
-- Minor injuries
-
-2. Worsening Conditions:
-- Increasing pain
-- Progressive symptoms
-- Medication reactions
-
-RESPONSE PROTOCOLS:
-1. For Life Threats:
-- Immediate 999 call
-- Clear first aid instructions
-- Stay on line until help arrives
-
-2. For Urgent Care:
-- 111 contact
-- Monitor for worsening
-- Document symptoms""",
-                "gp_booking.txt": """GP Appointment Booking Templates:
-
-APPOINTMENT TYPES:
-1. Routine Appointments:
-Template: "I need to book a routine appointment for [condition]. My availability is [times/dates]. My GP is Dr. [name] if available."
-
-2. Follow-up Appointments:
-Template: "I need a follow-up appointment regarding [condition] discussed on [date]. My previous appointment was with Dr. [name]."
-
-3. Medication Reviews:
-Template: "I need a medication review for [medication]. My last review was [date]."
-
-BOOKING INFORMATION NEEDED:
-1. Patient Details:
-- Full name
-- Date of birth
-- NHS number (if known)
-- Registered GP practice
-
-2. Appointment Details:
-- Nature of appointment
-- Preferred times/dates
-- Urgency level
-- Special requirements
-
-3. Contact Information:
-- Phone number
-- Alternative contact
-- Preferred contact method
-
-BOOKING PROCESS:
-1. Online Booking:
-- NHS app instructions
-- Practice website guidance
-- System navigation help
-
-2. Phone Booking:
-- Best times to call
-- Required information
-- Queue management tips
-
-3. Special Circumstances:
-- Interpreter needs
-- Accessibility requirements
-- Transport arrangements""",
-                "cultural_sensitivity.txt": """Cultural Sensitivity Guidelines:
-
-CULTURAL AWARENESS:
-1. Religious Considerations:
-- Prayer times
-- Religious observations
-- Dietary restrictions
-- Gender preferences for care
-- Religious festivals/fasting periods
-
-2. Language Support:
-- Interpreter services
-- Multi-language resources
-- Clear communication methods
-- Family involvement preferences
-
-3. Cultural Beliefs:
-- Traditional medicine practices
-- Cultural health beliefs
-- Family decision-making
-- Privacy customs
-
-COMMUNICATION APPROACHES:
-1. Respectful Interaction:
-- Use preferred names/titles
-- Appropriate greetings
-- Non-judgmental responses
-- Active listening
-
-2. Language Usage:
-- Clear, simple terms
-- Avoid medical jargon
-- Confirm understanding
-- Respect silence/pauses
-
-3. Non-verbal Communication:
-- Eye contact customs
-- Personal space
-- Body language awareness
-- Gesture sensitivity
-
-SPECIFIC CONSIDERATIONS:
-1. South Asian Communities:
-- Family involvement
-- Gender sensitivity
-- Traditional medicine
-- Language diversity
-
-2. Middle Eastern Communities:
-- Gender-specific care
-- Religious observations
-- Family hierarchies
-- Privacy concerns
-
-3. African/Caribbean Communities:
-- Traditional healers
-- Community involvement
-- Historical medical mistrust
-- Cultural specific conditions
-
-4. Eastern European Communities:
-- Direct communication
-- Family involvement
-- Medical documentation
-- Language support
-
-INCLUSIVE PRACTICES:
-1. Appointment Scheduling:
-- Religious holidays
-- Prayer times
-- Family availability
-- Interpreter needs
-
-2. Treatment Planning:
-- Cultural preferences
-- Traditional practices
-- Family involvement
-- Dietary requirements
-
-3. Support Services:
-- Community resources
-- Cultural organizations
-- Language services
-- Social support""",
-                "service_boundaries.txt": """Service Limitations and Professional Boundaries:
-
-CLEAR BOUNDARIES:
-1. Medical Advice:
-- No diagnoses
-- No prescriptions
-- No treatment recommendations
-- No medical procedures
-- No second opinions
-
-2. Emergency Services:
-- Clear referral criteria
-- Documented responses
-- Follow-up protocols
-- Handover procedures
-
-3. Information Sharing:
-- Confidentiality limits
-- Data protection
-- Record keeping
-- Information governance
-
-PROFESSIONAL CONDUCT:
-1. Communication:
-- Professional language
-- Emotional boundaries
-- Personal distance
-- Service scope
-
-2. Service Delivery:
-- No financial transactions
-- No personal relationships
-- Clear role definition
-- Professional limits"""
+                EMERGENCY (999) SCENARIOS:
+                1. Cardiovascular:
+                - Chest pain/pressure
+                - Heart attack symptoms
+                - Irregular heartbeat with dizziness
+                Response: Immediate 999 call, sit/lie down, chew aspirin if available
+                
+                2. Respiratory:
+                - Severe breathing difficulty
+                - Choking
+                - Unable to speak full sentences
+                Response: 999, sitting position, clear airway
+                
+                3. Neurological:
+                - Stroke symptoms (FAST)
+                - Seizures
+                - Unconsciousness
+                Response: 999, recovery position if unconscious
+                
+                4. Trauma:
+                - Severe bleeding
+                - Head injuries with confusion
+                - Major burns
+                Response: 999, apply direct pressure to bleeding
+                
+                URGENT CARE (111) SCENARIOS:
+                1. Moderate Symptoms:
+                - Persistent fever
+                - Non-severe infections
+                - Minor injuries
+                Response: 111 contact, monitor symptoms
+                
+                2. Minor Emergencies:
+                - Small cuts needing stitches
+                - Sprains and strains
+                - Mild allergic reactions
+                Response: 111 or urgent care visit
+                
+                GP APPOINTMENT SCENARIOS:
+                1. Routine Care:
+                - Chronic condition review
+                - Medication reviews
+                - Non-urgent symptoms
+                Response: Book routine GP appointment
+                
+                2. Preventive Care:
+                - Vaccinations
+                - Health screenings
+                - Regular check-ups
+                Response: Schedule with GP reception""",
+                                "emergency_detection.txt": """Enhanced Emergency Detection Criteria:
+                
+                IMMEDIATE LIFE THREATS:
+                1. Cardiac Symptoms:
+                - Chest pain/pressure/tightness
+                - Pain spreading to arms/jaw/neck
+                - Sweating with nausea
+                - Shortness of breath
+                
+                2. Breathing Problems:
+                - Severe shortness of breath
+                - Blue lips or face
+                - Unable to complete sentences
+                - Choking/airway blockage
+                
+                3. Neurological:
+                - FAST (Face, Arms, Speech, Time)
+                - Sudden confusion
+                - Severe headache
+                - Seizures
+                - Loss of consciousness
+                
+                4. Severe Trauma:
+                - Heavy bleeding
+                - Deep wounds
+                - Head injury with confusion
+                - Severe burns
+                - Broken bones with deformity
+                
+                5. Anaphylaxis:
+                - Sudden swelling
+                - Difficulty breathing
+                - Rapid onset rash
+                - Light-headedness
+                
+                URGENT BUT NOT IMMEDIATE:
+                1. Moderate Symptoms:
+                - Persistent fever
+                - Dehydration
+                - Non-severe infections
+                - Minor injuries
+                
+                2. Worsening Conditions:
+                - Increasing pain
+                - Progressive symptoms
+                - Medication reactions
+                
+                RESPONSE PROTOCOLS:
+                1. For Life Threats:
+                - Immediate 999 call
+                - Clear first aid instructions
+                - Stay on line until help arrives
+                
+                2. For Urgent Care:
+                - 111 contact
+                - Monitor for worsening
+                - Document symptoms""",
+                                "gp_booking.txt": """GP Appointment Booking Templates:
+                
+                APPOINTMENT TYPES:
+                1. Routine Appointments:
+                Template: "I need to book a routine appointment for [condition]. My availability is [times/dates]. My GP is Dr. [name] if available."
+                
+                2. Follow-up Appointments:
+                Template: "I need a follow-up appointment regarding [condition] discussed on [date]. My previous appointment was with Dr. [name]."
+                
+                3. Medication Reviews:
+                Template: "I need a medication review for [medication]. My last review was [date]."
+                
+                BOOKING INFORMATION NEEDED:
+                1. Patient Details:
+                - Full name
+                - Date of birth
+                - NHS number (if known)
+                - Registered GP practice
+                
+                2. Appointment Details:
+                - Nature of appointment
+                - Preferred times/dates
+                - Urgency level
+                - Special requirements
+                
+                3. Contact Information:
+                - Phone number
+                - Alternative contact
+                - Preferred contact method
+                
+                BOOKING PROCESS:
+                1. Online Booking:
+                - NHS app instructions
+                - Practice website guidance
+                - System navigation help
+                
+                2. Phone Booking:
+                - Best times to call
+                - Required information
+                - Queue management tips
+                
+                3. Special Circumstances:
+                - Interpreter needs
+                - Accessibility requirements
+                - Transport arrangements""",
+                                "cultural_sensitivity.txt": """Cultural Sensitivity Guidelines:
+                
+                CULTURAL AWARENESS:
+                1. Religious Considerations:
+                - Prayer times
+                - Religious observations
+                - Dietary restrictions
+                - Gender preferences for care
+                - Religious festivals/fasting periods
+                
+                2. Language Support:
+                - Interpreter services
+                - Multi-language resources
+                - Clear communication methods
+                - Family involvement preferences
+                
+                3. Cultural Beliefs:
+                - Traditional medicine practices
+                - Cultural health beliefs
+                - Family decision-making
+                - Privacy customs
+                
+                COMMUNICATION APPROACHES:
+                1. Respectful Interaction:
+                - Use preferred names/titles
+                - Appropriate greetings
+                - Non-judgmental responses
+                - Active listening
+                
+                2. Language Usage:
+                - Clear, simple terms
+                - Avoid medical jargon
+                - Confirm understanding
+                - Respect silence/pauses
+                
+                3. Non-verbal Communication:
+                - Eye contact customs
+                - Personal space
+                - Body language awareness
+                - Gesture sensitivity
+                
+                SPECIFIC CONSIDERATIONS:
+                1. South Asian Communities:
+                - Family involvement
+                - Gender sensitivity
+                - Traditional medicine
+                - Language diversity
+                
+                2. Middle Eastern Communities:
+                - Gender-specific care
+                - Religious observations
+                - Family hierarchies
+                - Privacy concerns
+                
+                3. African/Caribbean Communities:
+                - Traditional healers
+                - Community involvement
+                - Historical medical mistrust
+                - Cultural specific conditions
+                
+                4. Eastern European Communities:
+                - Direct communication
+                - Family involvement
+                - Medical documentation
+                - Language support
+                
+                INCLUSIVE PRACTICES:
+                1. Appointment Scheduling:
+                - Religious holidays
+                - Prayer times
+                - Family availability
+                - Interpreter needs
+                
+                2. Treatment Planning:
+                - Cultural preferences
+                - Traditional practices
+                - Family involvement
+                - Dietary requirements
+                
+                3. Support Services:
+                - Community resources
+                - Cultural organizations
+                - Language services
+                - Social support""",
+                                "service_boundaries.txt": """Service Limitations and Professional Boundaries:
+                
+                CLEAR BOUNDARIES:
+                1. Medical Advice:
+                - No diagnoses
+                - No prescriptions
+                - No treatment recommendations
+                - No medical procedures
+                - No second opinions
+                
+                2. Emergency Services:
+                - Clear referral criteria
+                - Documented responses
+                - Follow-up protocols
+                - Handover procedures
+                
+                3. Information Sharing:
+                - Confidentiality limits
+                - Data protection
+                - Record keeping
+                - Information governance
+                
+                PROFESSIONAL CONDUCT:
+                1. Communication:
+                - Professional language
+                - Emotional boundaries
+                - Personal distance
+                - Service scope
+                
+                2. Service Delivery:
+                - No financial transactions
+                - No personal relationships
+                - Clear role definition
+                - Professional limits"""
             }
             
         os.makedirs("knowledge_base", exist_ok=True)
@@ -474,10 +475,11 @@ PROFESSIONAL CONDUCT:
         texts = text_splitter.split_text("\n\n".join(documents))
         self.vector_store = FAISS.from_texts(texts, self.embeddings)
         logger.info("RAG system setup complete")
-        
-        except Exception as e:
-            logger.error(f"Error setting up RAG: {str(e)}")
-            raise
+
+        return knowledge_base
+    except Exception as e:
+        logger.error(f"Error setting up RAG: {str(e)}")
+        raise
 
     def _validate_rag_setup(self):
         """Validate RAG system setup"""