Update api/routes.py

api/routes.py

@@ -161,18 +161,25 @@ async def delete_patient(patient_id: str, current_user: dict = Depends(get_curre
 @router.get("/patients/count")
 async def count_patients(current_user: dict = Depends(get_current_user)):
     try:
-        email = current_user["email"]
-        print("🧪 Counting patients for email:", email)
+        email = current_user.get("email")
+        print("📌 Authenticated user email:", email)
 
-        filter_query = { "created_by": email }
-        print("🔍 Mongo filter:", filter_query)
+        if not email or not isinstance(email, str):
+            raise HTTPException(status_code=400, detail="Invalid email in token")
+
+        if patients_collection is None:
+            print("❌ patients_collection is not initialized")
+            raise HTTPException(status_code=500, detail="Database not connected")
+
+        filter_query = {"created_by": email}
+        print("🔍 Running count_documents with:", filter_query)
 
         count = await patients_collection.count_documents(filter_query)
-        print("✅ Count result:", count)
 
+        print("✅ Patient count result:", count)
         return {"count": count}
-    except Exception as e:
-        print("❌ Exception in /patients/count:", repr(e))
-        raise HTTPException(status_code=500, detail="Failed to count patients")
 
+    except Exception as e:
+        print("❌ ERROR in /patients/count:", repr(e))
+        raise HTTPException(status_code=500, detail="Internal server error")