Update config.py

config.py

@@ -1,10 +1,11 @@
-import os
 import logging
 from fastapi import FastAPI
 from txagent.txagent import TxAgent
 from db.mongo import get_mongo_client
 import asyncio
 from pyfcm import FCMNotification
+from huggingface_hub import HfApi, get_secret
+from datetime import datetime
 
 # Logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -38,10 +39,17 @@ patients_collection = db["patients"]
 analysis_collection = db["patient_analysis_results"]
 alerts_collection = db["clinical_alerts"]
 notifications_collection = db["notifications"]
-logger.info("📡 Connected to MongoDB synchronously")
+logger.info("📡 Connected to MongoDB synchronously at %s", datetime.utcnow().isoformat())
+
+# Retrieve secrets from Hugging Face
+FCM_API_KEY = get_secret("FCM_API_KEY")
+SECRET_KEY = get_secret("SECRET_KEY")
+
+if not FCM_API_KEY or not SECRET_KEY:
+    logger.error("❌ Missing FCM_API_KEY or SECRET_KEY in Hugging Face Secrets")
+    raise ValueError("FCM_API_KEY and SECRET_KEY must be set in Hugging Face Secrets")
 
 # 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):
@@ -50,7 +58,7 @@ async def send_push_notification(recipient_email, message):
         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}")
+            logger.warning(f"No device token found for {recipient_email} at {datetime.utcnow().isoformat()}")
             return
 
         # Send push notification
@@ -58,14 +66,14 @@ async def send_push_notification(recipient_email, message):
             registration_id=device_token,
             message_title="Risk Alert",
             message_body=message,
-            sound="default"  # Optional: Plays a sound on the device
+            sound="default",  # Plays a sound on the device
+            priority="high"
         )
-        logger.info(f"Push notification sent to {recipient_email}: {result}")
+        logger.info(f"Push notification sent to {recipient_email} at {datetime.utcnow().isoformat()}: {result}")
     except Exception as e:
-        logger.error(f"Failed to send push notification to {recipient_email}: {str(e)}")
+        logger.error(f"Failed to send push notification to {recipient_email} at {datetime.utcnow().isoformat()}: {str(e)}")
 
 # JWT settings
-SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key")
 ALGORITHM = "HS256"
 
 def setup_app(app: FastAPI):