Update config.py

config.py

@@ -4,6 +4,7 @@ from fastapi import FastAPI
 from txagent.txagent import TxAgent
 from db.mongo import get_mongo_client
 import asyncio
+from pyfcm import FCMNotification
 
 # Logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -36,9 +37,33 @@ users_collection = db["users"]
 patients_collection = db["patients"]
 analysis_collection = db["patient_analysis_results"]
 alerts_collection = db["clinical_alerts"]
-notifications_collection = db["notifications"]  # New collection for notifications
+notifications_collection = db["notifications"]
 logger.info("📡 Connected to MongoDB synchronously")
 
+# FCM settings
+FCM_API_KEY = os.getenv("FCM_API_KEY", "your-fcm-server-key")
+push_service = FCMNotification(api_key=FCM_API_KEY)
+
+async def send_push_notification(recipient_email, message):
+    try:
+        # Fetch the user's device token
+        user = await users_collection.find_one({"email": recipient_email})
+        device_token = user.get("device_token") if user and "device_token" in user else None
+        if not device_token:
+            logger.warning(f"No device token found for {recipient_email}")
+            return
+
+        # Send push notification
+        result = push_service.notify_single_device(
+            registration_id=device_token,
+            message_title="Risk Alert",
+            message_body=message,
+            sound="default"  # Optional: Plays a sound on the device
+        )
+        logger.info(f"Push notification sent to {recipient_email}: {result}")
+    except Exception as e:
+        logger.error(f"Failed to send push notification to {recipient_email}: {str(e)}")
+
 # JWT settings
 SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key")
 ALGORITHM = "HS256"