Update analysis.py

analysis.py

@@ -46,12 +46,25 @@ async def create_alert(patient_id: str, risk_data: dict):
     }
     
     try:
+        # Insert into database
         await alerts_collection.insert_one(alert_doc)
-        await broadcast_notification(alert_doc["notification"])
+        
+        # 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"])
+            
         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 for patient {patient_id}: {str(e)}")
+        logger.error(f"Failed to create alert: {str(e)}")
         raise
 
 async def analyze_patient_report(