Update config.py

config.py

@@ -6,6 +6,7 @@ import asyncio
 from pyfcm import FCMNotification
 from huggingface_hub import get_secret
 from datetime import datetime
+import httpx
 
 # Logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(name)s - %(message)s")
@@ -52,11 +53,21 @@ if not FCM_API_KEY or not SECRET_KEY:
 # FCM settings
 push_service = FCMNotification(api_key=FCM_API_KEY)
 
+# Auth Space URL
+AUTH_SPACE_URL = "https://your-auth-space-name.hf.space/auth"
+
 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
+        # Fetch the user's device token from the auth Space
+        async with httpx.AsyncClient() as client:
+            headers = {"Authorization": f"Bearer {create_access_token(recipient_email)}"}
+            response = await client.get(f"{AUTH_SPACE_URL}/me", headers=headers)
+            if response.status_code != 200:
+                logger.warning(f"Failed to fetch user {recipient_email} from auth Space: {response.text}")
+                return
+            user_data = response.json()
+            device_token = user_data.get("device_token")
+
         if not device_token:
             logger.warning(f"No device token found for {recipient_email} at {datetime.utcnow().isoformat()}")
             return
@@ -73,8 +84,24 @@ async def send_push_notification(recipient_email, message):
     except Exception as e:
         logger.error(f"Failed to send push notification to {recipient_email} at {datetime.utcnow().isoformat()}: {str(e)}")
 
-# JWT settings
+# JWT settings (minimal implementation for token creation)
+from datetime import timedelta
+import jwt
+
 ALGORITHM = "HS256"
+ACCESS_TOKEN_EXPIRE_MINUTES = 30
+
+def create_access_token(email: str):
+    to_encode = {"sub": email, "exp": datetime.utcnow() + timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)}
+    return jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM)
+
+def decode_access_token(token: str):
+    try:
+        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
+        return payload
+    except jwt.PyJWTError as e:
+        logger.error(f"Token decode error: {str(e)}")
+        return None
 
 def setup_app(app: FastAPI):
     @app.on_event("startup")