Update endpoints.py

endpoints.py

@@ -183,6 +183,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
         try:
             # Fetch notifications for the current user
             notifications = await notifications_collection.find({"user_id": current_user["email"]}).sort("timestamp", -1).to_list(length=10)
+            logger.info(f"Retrieved {len(notifications)} notifications for {current_user['email']}")
             return [
                 {
                     "id": str(notification["_id"]),
@@ -382,19 +383,24 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                 file_type=content_type,
                 file_content=file_content
             )
+            logger.info(f"Analysis result for patient {patient_id}: {analysis}")
 
             # Create a notification if suicide risk is detected
-            if analysis.get("suicide_risk", {}).get("level") != "none":
+            suicide_risk = analysis.get("suicide_risk", {})
+            logger.info(f"Suicide risk detected: {suicide_risk}")
+            if 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']})",
+                    "message": f"Suicide risk alert for patient {patient_id}: {suicide_risk['level'].upper()} (Score: {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 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}")
+            else:
+                logger.warning(f"No suicide risk detected for patient {patient_id}, no notification created")
 
             if "_id" in analysis and isinstance(analysis["_id"], ObjectId):
                 analysis["_id"] = str(analysis["_id"])