Update main.py

main.py

@@ -371,9 +371,28 @@ async def register_post(
     
     
 @app.get("/registration_successful", response_class=HTMLResponse)
-async def registration_successful(request: Request):
-    # Render the successful registration page
-    return templates.TemplateResponse("registration_successful.html", {"request": request})
+async def registration_successful(request: Request, db: Session = Depends(get_db)):
+    # Assuming the OAuth process has been completed and user info is stored in the session or a similar mechanism
+    user_info = request.session.get("user_info")  # Replace with your method of retrieving user info
+
+    if not user_info:
+        raise HTTPException(status_code=401, detail="User not authenticated")
+
+    email = user_info["email"]
+    db_user = db.query(User).filter(User.email == email).first()
+    if not db_user:
+        raise HTTPException(status_code=404, detail="User not found")
+
+    # Create an access token for the user
+    access_token = auth_views.create_access_token(
+        data={"sub": db_user.email},
+        expires_delta=timedelta(minutes=auth_views.ACCESS_TOKEN_EXPIRE_MINUTES)
+    )
+
+    # Redirect the user to the protected route
+    response = RedirectResponse(url="/protected")
+    response.set_cookie(key="access_token", value=f"Bearer {access_token}", httponly=True)
+    return response
 
 
 @app.get("/verify", response_class=HTMLResponse)