Update app.py

app.py

@@ -11,6 +11,10 @@ import psutil
 from datetime import datetime
 from huggingface_hub import login
 from dotenv import load_dotenv
+import aiohttp
+import asyncio
+from googlesearch import search
+from apscheduler.schedulers.background import BackgroundScheduler
 from transformers import (
     AutoTokenizer, AutoModelForCausalLM,
     BitsAndBytesConfig
@@ -53,6 +57,7 @@ class MedicalTriageBot:
         self.num_relevant_chunks = 5
         self.last_interaction_time = time.time()
         self.interaction_cooldown = 1.0
+        self.nhs_api_url = "https://api.nhs.uk/conditions/"
         self.safety_phrases = [
             "999", "111", "emergency", "GP", "NHS",
             "consult a doctor", "seek medical attention"
@@ -87,6 +92,75 @@ class MedicalTriageBot:
             ("emergency_signs", "Any difficulty breathing, chest pain, or confusion?")
         ]
 
+    async def query_nhs_api(self, symptom: str):
+        """Dynamic API query for latest NHS guidelines"""
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.get(f"{self.nhs_api_url}{symptom}") as response:
+                    if response.status == 200:
+                        return await response.json()
+                    return None
+        except Exception as e:
+            logger.error(f"NHS API Error: {e}")
+            return None
+
+    def web_fallback(self, query: str):
+        """Google search fallback for NHS resources"""
+        try:
+            NHS_SITES = ["site:nhs.uk", "site:gov.uk"]
+            return [j for j in search(f"{query} {' '.join(NHS_SITES)}", num=3, stop=3)]
+        except Exception as e:
+            logger.error(f"Web search failed: {e}")
+            return []
+
+    def get_medical_context(self, query):
+        """Hybrid context retrieval system"""
+        try:
+            # 1. Try local knowledge base
+            local_context = self.vector_store.similarity_search(query, k=2)
+            if len(local_context) < 1:
+                raise ValueError("No local results")
+                
+            # 2. Fallback to NHS API
+            api_data = asyncio.run(self.query_nhs_api(query))
+            if api_data:
+                return self._parse_api_response(api_data)
+                
+            # 3. Final fallback to web search
+            web_results = self.web_fallback(query)
+            return "\n".join(web_results[:2])
+            
+        except Exception as e:
+            logger.error(f"Context retrieval failed: {e}")
+            return ""
+
+    def _parse_api_response(self, api_data):
+        """Structure NHS API response for LLM consumption"""
+        return f"""
+        [NHS Direct Guidelines]
+        Condition: {api_data.get('name', '')}
+        Symptoms: {', '.join(api_data.get('symptoms', []))}
+        Treatment: {api_data.get('treatment', '')}
+        Last Updated: {api_data.get('dateModified', '')}
+        """
+
+    def schedule_knowledge_updates(self):
+        """Weekly index rebuild with fresh data"""
+        scheduler = BackgroundScheduler()
+        scheduler.add_job(self.build_medical_index, 'interval', weeks=1)
+        scheduler.start()
+
+    @torch.inference_mode()
+    def generate_safe_response(self, message, history):
+        """Enhanced with dynamic crisis handling"""
+        # Emergency detection
+        if any(kw in message.lower() for kw in ['suicid', 'end it all', 'kill myself']):
+            return self._mental_health_crisis_response()
+            
+        try:
+            # Dynamic context handling
+            context = self.get_medical_context(message)
+
     def _determine_triage_level(self, context):
         context_lower = context.lower()
         for level, keywords in triage_levels.items():
@@ -396,6 +470,15 @@ class MedicalTriageBot:
         )
         vector_store.save_local("medical_index")
 
+    def _mental_health_crisis_response(self):
+        """Immediate crisis intervention"""
+        return """🚨 **Emergency Mental Health Support**
+1. Call 999 immediately
+2. Text SHOUT to 85258 (24/7 crisis text line)
+3. Stay on chat - I'll help you connect to services
+
+You're not alone. Let's get through this together."""
+
     def get_medical_context(self, query):
         try:
             docs = self.vector_store.similarity_search(query, k=2)