Update app.py

app.py

@@ -197,6 +197,30 @@ async def auth_callback(request: Request, db: Session = Depends(get_db)):
     # Redirect to a success or dashboard page
     return RedirectResponse(url="/registration_successful")
 
+@app.get("/registration_successful", response_class=HTMLResponse)
+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 = 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("/", response_class=HTMLResponse)
 async def landing(request: Request):
     return templates.TemplateResponse("landing.html", {"request": request})