Update analysis.py

analysis.py

@@ -15,11 +15,25 @@ async def create_alert(patient_id: str, risk_data: dict):
         "score": risk_data["score"],
         "factors": risk_data["factors"],
         "timestamp": datetime.utcnow(),
-        "acknowledged": False
+        "acknowledged": False,
+        # Facebook-like notification fields
+        "notification": {
+            "type": NotificationType.RISK_ALERT,
+            "status": NotificationStatus.UNREAD,
+            "title": f"Suicide Risk: {risk_data['level'].capitalize()}",
+            "message": f"Patient {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"
+        }
     }
     await alerts_collection.insert_one(alert_doc)
+    
+    # Trigger real-time notification
+    await broadcast_notification(alert_doc["notification"])
+    
     logger.warning(f"⚠️ Created suicide risk alert for patient {patient_id}")
-
+    return alert_doc
 async def analyze_patient_report(patient_id: Optional[str], report_content: str, file_type: str, file_content: bytes):
     identifier = patient_id if patient_id else compute_file_content_hash(file_content)
     report_data = {"identifier": identifier, "content": report_content, "file_type": file_type}