Update api/routes.py

api/routes.py

@@ -67,13 +67,19 @@ async def login(form_data: OAuth2PasswordRequestForm = Depends()):
 # --- GET CURRENT USER ---
 @router.get("/me")
 async def get_me(current_user: dict = Depends(get_current_user)):
+    user = await users_collection.find_one({"email": current_user["email"]})
+    if not user:
+        raise HTTPException(status_code=404, detail="User not found")
+
     return {
-        "email": current_user["email"],
-        "full_name": current_user.get("full_name", ""),
-        "role": current_user.get("role", "unknown"),
-        "created_at": current_user.get("created_at", "")
+        "email": user["email"],
+        "full_name": user.get("full_name", ""),
+        "role": user.get("role", "unknown"),
+        "specialty": user.get("specialty", None) if user.get("role") == "doctor" else None,
+        "created_at": user.get("created_at", "")
     }
 
+
 # --- ADD NEW PATIENT ---
 @router.post("/patients")
 async def add_patient(data: PatientCreate, current_user: dict = Depends(get_current_user)):