Update main.py

main.py

@@ -123,6 +123,40 @@ def create_assessment(
         print(f"Assessment name: {assessment_name}")
     return response
 """
+
+@app.post("/verify-google-token")
+async def verify_google_token(token_data: TokenData, db: Session = Depends(get_db)):
+    # Verify the token with Google
+    response = requests.get(f'https://www.googleapis.com/oauth2/v3/tokeninfo?id_token={token_data.token}')
+    if response.status_code != 200:
+        raise HTTPException(status_code=400, detail="Invalid Google token")
+
+    google_user_info = response.json()
+    email = google_user_info.get('email')
+
+    # Check if user exists in database and verify them
+    db_user = db.query(User).filter(User.email == email).first()
+    if not db_user:
+        # Create a new user if doesn't exist
+        db_user = User(email=email, is_verified=True, name=google_user_info.get('name'))
+        db.add(db_user)
+        db.commit()
+        db.refresh(db_user)
+    elif not db_user.is_verified:
+        # Verify the user if not already verified
+        db_user.is_verified = True
+        db.commit()
+
+    # 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
 # Dependency for verifying the user's token
 def get_current_user(token: str = Depends(verify_token)):
     if not token: