Update main.py

main.py

@@ -2,7 +2,7 @@
 
 from fastapi import FastAPI, Form, Depends, HTTPException
 from fastapi.requests import Request
-from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.responses import HTMLResponse, RedirectResponse, JSONResponse
 from fastapi.templating import Jinja2Templates
 from sqlalchemy.orm import Session
 from auth import verify_token, oauth2_scheme, auth_views, register, UserCreate, authenticate_user, get_user_by_verification_token
@@ -123,26 +123,23 @@ 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)):
-    # Verify the email using the token
+async def verify_email(verification_token: str, db: Session = Depends(get_db)):
     user = get_user_by_verification_token(db, verification_token)
-    
     if not user:
         raise HTTPException(status_code=400, detail="Invalid verification token")
 
     if user.is_verified:
         raise HTTPException(status_code=400, detail="Email already verified")
 
-    # Mark the email as verified in the database
     user.is_verified = True
-    #user.email_verification_token = None  # Optionally clear the verification token
+    user.email_verification_token = None  # Clear the verification token
     db.commit()
 
-    # Handle a successful verification
-  #  return templates.TemplateResponse("verification_successful.html", {"request": request})
-    return RedirectResponse("/protected")
+    # Create access token for the user after successful verification
+    access_token = auth_views.create_access_token(data={"sub": user.email}, expires_delta=timedelta(minutes=auth_views.ACCESS_TOKEN_EXPIRE_MINUTES))
+    # Redirect to the protected route with the token as a query parameter (or as required by your front-end/client)
+    return RedirectResponse(url=f"/protected?token={access_token}")
 
 # User authentication (protected route)
 @app.post("/protected", response_class=HTMLResponse)  # Specify response_class as HTMLResponse