Update app.py

app.py

@@ -1,7 +1,7 @@
 from fastapi.security import OAuth2PasswordBearer
 from pydantic import BaseModel
 from sqlalchemy.orm import Session
-from database import get_db, get_user_by_email, get_user_by_verification_token
+from database import get_db, get_user_by_email
 from models import User
 from passlib.context import CryptContext
 from datetime import datetime, timedelta
@@ -125,6 +125,8 @@ def register_user(user_data: UserCreate, db: Session):
     db.commit()
     db.refresh(new_user)
     return new_user
+def get_user_by_verification_token(db: Session, verification_token: str):
+    return db.query(User).filter(User.email_verification_token == verification_token).first()
 
 def reset_password(user: User, db: Session):
     verification_token = generate_verification_token(user.email)