Update utils.py

utils.py

@@ -7,6 +7,52 @@ from typing import Dict, List, Tuple
 from bson import ObjectId
 import logging
 from config import logger
+# Add to your utils.py
+from fastapi import WebSocket
+import asyncio
+
+class NotificationManager:
+    def __init__(self):
+        self.active_connections = {}
+        self.notification_queue = asyncio.Queue()
+
+    async def connect(self, websocket: WebSocket, user_id: str):
+        await websocket.accept()
+        self.active_connections[user_id] = websocket
+
+    def disconnect(self, user_id: str):
+        if user_id in self.active_connections:
+            del self.active_connections[user_id]
+
+    async def broadcast_notification(self, notification: dict):
+        """Broadcast to all connected clients"""
+        for connection in self.active_connections.values():
+            try:
+                await connection.send_json({
+                    "type": "notification",
+                    "data": notification
+                })
+            except Exception as e:
+                logger.error(f"Error sending notification: {e}")
+
+notification_manager = NotificationManager()
+
+async def broadcast_notification(notification: dict):
+    """Broadcast notification to relevant users"""
+    # Determine recipients based on notification type/priority
+    recipients = []
+    if notification["priority"] == "high":
+        recipients = ["psychiatrist", "emergency_team", "primary_care"]
+    else:
+        recipients = ["primary_care", "case_manager"]
+    
+    # Add to each recipient's notification queue
+    await notification_manager.notification_queue.put({
+        "recipients": recipients,
+        "notification": notification
+    })
+
+
 
 def clean_text_response(text: str) -> str:
     text = re.sub(r'\n\s*\n', '\n\n', text)