Update api/routes/auth.py

api/routes/auth.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, HTTPException, Depends, status
+from fastapi import APIRouter, HTTPException, Depends, status, Request
 from fastapi.security import OAuth2PasswordRequestForm
 from db.mongo import users_collection
 from core.security import hash_password, verify_password, create_access_token, get_current_user
@@ -122,28 +122,35 @@ async def login(form_data: OAuth2PasswordRequestForm = Depends()):
     }
 
 @router.get("/me")
-async def get_me(current_user: dict = Depends(get_current_user)):
-    print(f"Headers: {request.headers}")
+async def get_me(request: Request, current_user: dict = Depends(get_current_user)):
     logger.info(f"Fetching user profile for {current_user['email']}")
-    user = await users_collection.find_one({"email": current_user["email"]})
-    if not user:
-        logger.warning(f"User not found: {current_user['email']}")
+    print(f"Headers: {request.headers}")
+    try:
+        user = await users_collection.find_one({"email": current_user["email"]})
+        if not user:
+            logger.warning(f"User not found: {current_user['email']}")
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="User not found"
+            )
+        
+        response = {
+            "id": str(user["_id"]),
+            "email": user["email"],
+            "full_name": user.get("full_name", ""),
+            "role": user.get("role", "patient"),
+            "specialty": user.get("specialty"),
+            "created_at": user.get("created_at"),
+            "updated_at": user.get("updated_at")
+        }
+        logger.info(f"User profile retrieved for {current_user['email']}")
+        return response
+    except Exception as e:
+        logger.error(f"Database error for user {current_user['email']}: {str(e)}")
         raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="User not found"
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"Database error: {str(e)}"
         )
-    
-    response = {
-        "id": str(user["_id"]),
-        "email": user["email"],
-        "full_name": user.get("full_name", ""),
-        "role": user.get("role", "patient"),
-        "specialty": user.get("specialty"),
-        "created_at": user.get("created_at"),
-        "updated_at": user.get("updated_at")
-    }
-    logger.info(f"User profile retrieved for {current_user['email']}")
-    return response
 
 # Export the router as 'auth' for api.__init__.py
 auth = router