Update config.py

config.py

@@ -7,6 +7,8 @@ from pyfcm import FCMNotification
 from datetime import datetime
 import httpx
 import os
+import json
+import tempfile
 
 # Logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(name)s - %(message)s")
@@ -43,15 +45,20 @@ notifications_collection = db["notifications"]
 logger.info("📡 Connected to MongoDB synchronously at %s", datetime.utcnow().isoformat())
 
 # Retrieve secrets from environment variables (set in Hugging Face Space Secrets)
-FCM_API_KEY = os.getenv("FCM_API_KEY")
+FCM_SERVICE_ACCOUNT_JSON = os.getenv("FCM_SERVICE_ACCOUNT")
 SECRET_KEY = os.getenv("SECRET_KEY")
 
-if not FCM_API_KEY or not SECRET_KEY:
-    logger.error("❌ Missing FCM_API_KEY or SECRET_KEY in environment variables")
-    raise ValueError("FCM_API_KEY and SECRET_KEY must be set in Hugging Face Space Secrets")
+if not FCM_SERVICE_ACCOUNT_JSON or not SECRET_KEY:
+    logger.error("❌ Missing FCM_SERVICE_ACCOUNT or SECRET_KEY in environment variables")
+    raise ValueError("FCM_SERVICE_ACCOUNT and SECRET_KEY must be set in Hugging Face Space Secrets")
+
+# Write the FCM service account JSON to a temporary file
+with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.json') as temp_file:
+    json.dump(json.loads(FCM_SERVICE_ACCOUNT_JSON), temp_file)
+    temp_file_path = temp_file.name
 
 # FCM settings
-push_service = FCMNotification(FCM_API_KEY)  # Corrected to use positional argument
+push_service = FCMNotification(service_account_file=temp_file_path)
 
 # Auth Space URL
 AUTH_SPACE_URL = "https://rocketfarmstudios-cps-api.hf.space/auth"
@@ -83,6 +90,10 @@ async def send_push_notification(recipient_email, message):
         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} at {datetime.utcnow().isoformat()}: {str(e)}")
+    finally:
+        # Clean up the temporary file
+        if os.path.exists(temp_file_path):
+            os.remove(temp_file_path)
 
 # JWT settings (minimal implementation for token creation)
 from datetime import timedelta