Update main.py

main.py

@@ -94,16 +94,25 @@ async def registration_successful(request: Request):
 
 @app.get("/verify/{verification_token}", response_class=HTMLResponse)
 async def verify_email(verification_token: str, request: Request, db: Session = Depends(get_db)):
-    # Perform email verification
-    verification_result = await verify_email(verification_token, db)
-    
-    # Handle a successful verification
-    if "message" in verification_result:
-        # Redirect the user to the protected area
-        return RedirectResponse("/protected")
+    # Verify the email using the token
+    user_email = verify_token(verification_token)
+    if not user_email:
+        return HTTPException(status_code=400, detail="Invalid verification token")
+
+    # Get the user by email
+    user = get_user_by_email(db, user_email)
+    if not user:
+        return HTTPException(status_code=400, detail="User not found")
 
-    # Handle any other cases, such as errors
-    return HTTPException(status_code=400, detail="Verification failed")
+    if user.is_verified:
+        return HTTPException(status_code=400, detail="Email already verified")
+
+    # Mark the email as verified in the database
+    user.is_verified = True
+    db.commit()
+
+    # Handle a successful verification
+    return RedirectResponse("/protected")
 
 # User authentication (protected route)
 @app.post("/protected", response_model=str)