Update endpoints.py

endpoints.py

@@ -175,6 +175,28 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
             logger.error(f"Error fetching chats: {e}")
             raise HTTPException(status_code=500, detail="Failed to retrieve chats")
 
+    @router.get("/notifications")
+    async def get_notifications(
+        current_user: dict = Depends(get_current_user)
+    ):
+        logger.info(f"Fetching notifications for {current_user['email']}")
+        try:
+            # Fetch notifications for the current user (e.g., alerts related to their patients)
+            notifications = await notifications_collection.find({"user_id": current_user["email"]}).sort("timestamp", -1).to_list(length=10)
+            return [
+                {
+                    "id": str(notification["_id"]),
+                    "message": notification.get("message", "No message"),
+                    "timestamp": notification.get("timestamp", datetime.utcnow()).isoformat(),
+                    "patient_id": notification.get("patient_id"),
+                    "severity": notification.get("severity", "info")
+                }
+                for notification in notifications
+            ]
+        except Exception as e:
+            logger.error(f"Error fetching notifications: {e}")
+            raise HTTPException(status_code=500, detail="Failed to retrieve notifications")
+
     @router.post("/voice/transcribe")
     async def transcribe_voice(
         audio: UploadFile = File(...),
@@ -323,6 +345,18 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                 file_content=file_content
             )
 
+            # Create a notification if suicide risk is detected
+            if analysis.get("suicide_risk", {}).get("level") != "none":
+                notification = {
+                    "user_id": current_user["email"],
+                    "message": f"Suicide risk alert for patient {patient_id}: {analysis['suicide_risk']['level'].upper()} (Score: {analysis['suicide_risk']['score']})",
+                    "patient_id": patient_id,
+                    "timestamp": datetime.utcnow(),
+                    "severity": "high" if analysis["suicide_risk"]["level"] in ["moderate", "severe"] else "medium"
+                }
+                await notifications_collection.insert_one(notification)
+                logger.info(f"✅ Created notification for suicide risk alert: {notification}")
+
             if "_id" in analysis and isinstance(analysis["_id"], ObjectId):
                 analysis["_id"] = str(analysis["_id"])
             if "timestamp" in analysis and isinstance(analysis["timestamp"], datetime):