Update app.py

app.py

@@ -99,7 +99,49 @@ async def login(form_data: OAuth2PasswordRequestForm = Depends(), db: Session =
         )
     access_token = create_access_token(data={"sub": user.username})
     return {"access_token": access_token, "token_type": "bearer"}
-    
+@app.get("/login", response_class=HTMLResponse)
+async def login(request: Request, db: Session = Depends(get_db)):
+    access_token = request.cookies.get("access_token")
+
+    if access_token:
+        try:
+            user_email = verify_token(access_token.split("Bearer ")[1])
+            if user_email:
+                # Retrieve the user from the database
+                db_user = db.query(User).filter(User.email == user_email).first()
+                if not db_user:
+                    raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail="User not found")
+
+                # Check if user is verified
+                if not db_user.is_verified:
+                    raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail="User is not verified")
+
+                # Create a new access token for the user
+                new_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
+                url = app.url_path_for("get_protected")
+                response = RedirectResponse(url)
+                response.set_cookie(key="access_token", value=f"Bearer {new_access_token}", httponly=True)
+                return response
+        except ExpiredSignatureError:
+            # Token has expired. You could redirect to the login page or inform the user
+            raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail="Token expired")
+        except InvalidTokenError:
+            # Token is invalid, inform the user or redirect
+            raise HTTPException(status_code=HTTP_400_BAD_REQUEST, detail="Invalid token")
+        except Exception as e:
+            # General exception, log this exception for debugging
+            # Respond with a generic error message
+            raise HTTPException(status_code=HTTP_400_BAD_REQUEST, detail="An error occurred")
+
+    # If not authenticated, show the login page with Google OAuth option
+    google_oauth_url = request.url_for("login_oauth")  # URL to initiate Google OAuth
+    return templates.TemplateResponse("login.html", {"request": request, "google_oauth_url": google_oauth_url})
+  
 @app.get("/", response_class=HTMLResponse)
 async def landing(request: Request):
     return templates.TemplateResponse("landing.html", {"request": request})