Update endpoints.py

endpoints.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, Depends, HTTPException, UploadFile, File, Query, Form
+from fastapi import APIRouter, Depends, HTTPException, UploadFile, File, Query, Form, Path
 from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.encoders import jsonable_encoder
 from typing import Optional, List
@@ -35,7 +35,7 @@ class RiskLevel(BaseModel):
     score: float
     factors: Optional[List[str]] = None
 
-def create_router(agent, logger, patients_collection, analysis_collection, users_collection, chats_collection):
+def create_router(agent, logger, patients_collection, analysis_collection, users_collection, chats_collection, notifications_collection):
     router = APIRouter()
 
     @router.get("/status")
@@ -181,15 +181,16 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
     ):
         logger.info(f"Fetching notifications for {current_user['email']}")
         try:
-            # Fetch notifications for the current user (e.g., alerts related to their patients)
+            # Fetch notifications for the current user
             notifications = await notifications_collection.find({"user_id": current_user["email"]}).sort("timestamp", -1).to_list(length=10)
             return [
                 {
                     "id": str(notification["_id"]),
+                    "title": f"Alert for Patient {notification.get('patient_id', 'Unknown')}",
                     "message": notification.get("message", "No message"),
                     "timestamp": notification.get("timestamp", datetime.utcnow()).isoformat(),
-                    "patient_id": notification.get("patient_id"),
-                    "severity": notification.get("severity", "info")
+                    "severity": notification.get("severity", "info"),
+                    "read": notification.get("read", False)
                 }
                 for notification in notifications
             ]
@@ -197,6 +198,43 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
             logger.error(f"Error fetching notifications: {e}")
             raise HTTPException(status_code=500, detail="Failed to retrieve notifications")
 
+    @router.post("/notifications/{notification_id}/read")
+    async def mark_notification_as_read(
+        notification_id: str = Path(..., description="The ID of the notification to mark as read"),
+        current_user: dict = Depends(get_current_user)
+    ):
+        logger.info(f"Marking notification {notification_id} as read for {current_user['email']}")
+        try:
+            result = await notifications_collection.update_one(
+                {"_id": ObjectId(notification_id), "user_id": current_user["email"]},
+                {"$set": {"read": True}}
+            )
+            if result.matched_count == 0:
+                raise HTTPException(status_code=404, detail="Notification not found or not authorized")
+            return {"status": "success", "message": "Notification marked as read"}
+        except InvalidId:
+            raise HTTPException(status_code=400, detail="Invalid notification ID format")
+        except Exception as e:
+            logger.error(f"Error marking notification as read: {e}")
+            raise HTTPException(status_code=500, detail="Failed to mark notification as read")
+
+    @router.post("/notifications/read-all")
+    async def mark_all_notifications_as_read(
+        current_user: dict = Depends(get_current_user)
+    ):
+        logger.info(f"Marking all notifications as read for {current_user['email']}")
+        try:
+            result = await notifications_collection.update_many(
+                {"user_id": current_user["email"], "read": False},
+                {"$set": {"read": True}}
+            )
+            if result.matched_count == 0:
+                logger.info("No unread notifications to mark as read")
+            return {"status": "success", "message": f"Marked {result.modified_count} notifications as read"}
+        except Exception as e:
+            logger.error(f"Error marking all notifications as read: {e}")
+            raise HTTPException(status_code=500, detail="Failed to mark all notifications as read")
+
     @router.post("/voice/transcribe")
     async def transcribe_voice(
         audio: UploadFile = File(...),
@@ -352,7 +390,8 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                     "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"
+                    "severity": "high" if analysis["suicide_risk"]["level"] in ["moderate", "severe"] else "medium",
+                    "read": False
                 }
                 await notifications_collection.insert_one(notification)
                 logger.info(f"✅ Created notification for suicide risk alert: {notification}")