Update app.py

app.py

@@ -223,7 +223,37 @@ async def registration_successful(request: Request, db: Session = Depends(get_db
 
 @app.get("/register", response_class=HTMLResponse)
 async def register_get(request: Request):
-    return templates.TemplateResponse("register.html", {"request": request})
+    return templates.TemplateResponse("register.html", {"request": request, "google_oauth_url": request.url_for("login_oauth")})
+
+@app.post("/register")
+async def register_post(
+    request: Request,
+    username: str = Form(...),
+    email: str = Form(...),
+    password: str = Form(...),
+    confirm_password: str = Form(...),
+    recaptcha_token: str = Form(...),
+    db: Session = Depends(get_db)
+):
+    if not await verify_recaptcha(recaptcha_token):
+        return templates.TemplateResponse("register.html", {"request": request, "error_message": "reCAPTCHA validation failed."})
+
+    if password != confirm_password:
+        return templates.TemplateResponse("register.html", {"request": request, "error_message": "Passwords do not match."})
+
+    user_data = UserCreate(username=username, email=email, password=password)
+    try:
+        registered_user = register_user(user_data, db)
+        # Store user info in the session after successful registration
+        request.session["user_info"] = {"username": registered_user.username, "email": registered_user.email}
+        # Create an access token
+        access_token = create_access_token(data={"sub": registered_user.email})
+        # Redirect to the protected route
+        response = RedirectResponse(url="/protected")
+        response.set_cookie(key="access_token", value=f"Bearer {access_token}", httponly=True)
+        return response
+    except HTTPException as e:
+        return templates.TemplateResponse("register.html", {"request": request, "error_message": e.detail})
     
 @app.get("/", response_class=HTMLResponse)
 async def landing(request: Request):