Update analysis.py

analysis.py

@@ -25,48 +25,36 @@ class NotificationStatus(str, Enum):
     ARCHIVED = "archived"
 
 async def create_alert(patient_id: str, risk_data: dict):
-    """Create a new risk alert with notification metadata"""
-    alert_doc = {
-        "patient_id": patient_id,
-        "type": "suicide_risk",
-        "level": risk_data["level"],
-        "score": risk_data["score"],
-        "factors": risk_data["factors"],
-        "timestamp": datetime.utcnow(),
-        "acknowledged": False,
-        "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"
-        }
-    }
-    
     try:
-        # Insert into database
+        alert_doc = {
+            "patient_id": patient_id,
+            "type": "suicide_risk",
+            "level": risk_data["level"],
+            "score": risk_data["score"],
+            "factors": risk_data["factors"],
+            "timestamp": datetime.utcnow(),
+            "acknowledged": False,
+            "notification": {
+                "type": "risk_alert",
+                "status": "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)
         
-        # For Hugging Face Spaces, use their event system instead of raw WebSocket
-        if os.getenv('SPACES') == '1':
-            from gradio import EventSource
-            EventSource.emit(
-                event="notification",
-                data=json.dumps(alert_doc["notification"]),
-                broadcast=True
-            )
-        else:
-            # Local development with regular WebSocket
-            await broadcast_notification(alert_doc["notification"])
-            
+        # Simplified WebSocket notification - remove Hugging Face specific code
+        await broadcast_notification(alert_doc["notification"])
+        
         logger.warning(f"⚠️ Created suicide risk alert for patient {patient_id}")
         return alert_doc
     except Exception as e:
         logger.error(f"Failed to create alert: {str(e)}")
         raise
-
 async def analyze_patient_report(
     patient_id: Optional[str], 
     report_content: str,