Update analysis.py

analysis.py

@@ -1,6 +1,6 @@
 from typing import Optional, Tuple, List
 from enum import Enum
-from config import agent, patients_collection, analysis_collection, alerts_collection, logger
+from config import agent, patients_collection, analysis_collection, alerts_collection, notifications_collection, users_collection, logger
 from models import RiskLevel
 from utils import (
     structure_medical_response, 
@@ -26,6 +26,10 @@ class NotificationStatus(str, Enum):
 
 async def create_alert(patient_id: str, risk_data: dict):
     try:
+        # Get patient information for better notification message
+        patient = await patients_collection.find_one({"fhir_id": patient_id})
+        patient_name = patient.get("full_name", "Unknown Patient") if patient else "Unknown Patient"
+        
         alert_doc = {
             "patient_id": patient_id,
             "type": "suicide_risk",
@@ -38,7 +42,7 @@ async def create_alert(patient_id: str, risk_data: dict):
                 "type": "risk_alert",
                 "status": "unread",
                 "title": f"Suicide Risk: {risk_data['level'].capitalize()}",
-                "message": f"Patient {patient_id} shows {risk_data['level']} risk factors",
+                "message": f"Patient {patient_name} ({patient_id}) shows {risk_data['level']} risk factors",
                 "icon": "⚠️",
                 "action_url": f"/patient/{patient_id}/risk-assessment",
                 "priority": "high" if risk_data["level"] in ["high", "severe"] else "medium"
@@ -47,10 +51,43 @@ async def create_alert(patient_id: str, risk_data: dict):
         
         await alerts_collection.insert_one(alert_doc)
         
+        # Create notifications for all doctors and admins when risk is high or moderate
+        if risk_data["level"] in ["high", "moderate", "severe"]:
+            # Get all users with doctor or admin roles
+            doctors_and_admins = await users_collection.find({
+                "roles": {"$in": ["doctor", "admin"]}
+            }).to_list(length=None)
+            
+            logger.info(f"📧 Creating notifications for {len(doctors_and_admins)} doctors/admins")
+            
+            # Create individual notifications for each doctor/admin
+            notifications = []
+            for user in doctors_and_admins:
+                notification = {
+                    "user_id": user["email"],
+                    "patient_id": patient_id,
+                    "message": f"🚨 {risk_data['level'].upper()} suicide risk detected for patient {patient_name}",
+                    "timestamp": datetime.utcnow(),
+                    "severity": "high" if risk_data["level"] in ["high", "severe"] else "medium",
+                    "read": False,
+                    "type": "suicide_risk_alert",
+                    "risk_level": risk_data["level"],
+                    "risk_score": risk_data["score"],
+                    "patient_name": patient_name
+                }
+                notifications.append(notification)
+            
+            if notifications:
+                # Insert all notifications at once
+                result = await notifications_collection.insert_many(notifications)
+                logger.info(f"✅ Created {len(result.inserted_ids)} notifications for risk alert")
+            else:
+                logger.warning("⚠️ No doctors or admins found to notify")
+        
         # Simplified WebSocket notification - remove Hugging Face specific code
         await broadcast_notification(alert_doc["notification"])
         
-        logger.warning(f"⚠️ Created suicide risk alert for patient {patient_id}")
+        logger.warning(f"⚠️ Created suicide risk alert for patient {patient_id} ({patient_name}) - Level: {risk_data['level']}")
         return alert_doc
     except Exception as e:
         logger.error(f"Failed to create alert: {str(e)}")