Update routes/auth.py

routes/auth.py

@@ -11,7 +11,7 @@ from datetime import datetime, timedelta
 
 router = APIRouter()
 app = FastAPI()
-
+templates = Jinja2Templates(directory="templates")
 class AuthViews:
     def __init__(self):
         self.pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
@@ -87,9 +87,26 @@ from app.database import get_user_by_email
 
 # ...
 
+
+
+auth_views = AuthViews()
+
+@app.get("/login", response_class=HTMLResponse)
+async def login(request: Request):
+    return templates.TemplateResponse("login.html", {"request": request})
+
+@app.get("/register", response_class=HTMLResponse)
+async def register(request: Request):
+    return templates.TemplateResponse("register.html", {"request": request})
+
+@app.get("/verify/{verification_token}", response_class=HTMLResponse)
+async def verify_email(verification_token: str, request: Request):
+    # Perform verification and return an appropriate template
+    return templates.TemplateResponse("verify.html", {"request": request})
+    
 # User authentication (protected route)
-@router.get("/protected", response_model=str)
-def protected_route(token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)):
+@app.get("/protected", response_model=str)
+async def protected_route(request: Request, token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)):
     # Verify the access token
     user = verify_token(token, SECRET_KEY, ALGORITHM)
     if user is None:
@@ -101,28 +118,5 @@ def protected_route(token: str = Depends(oauth2_scheme), db: Session = Depends(g
     if db_user is None:
         raise HTTPException(status_code=401, detail="User not found in the database")
 
-    # The user exists in the database, and you can perform the protected route logic
-    return f"Protected route for user: {user}"
-
-
-auth_views = AuthViews()
-
-# User registration route
-@router.post("/register", response_model=User)
-def register(user: UserCreate, db: Session = Depends(get_db)):
-    return auth_views.register(user, db)
-
-# Email verification route
-@router.get("/verify/{verification_token}")
-def verify_email(verification_token: str, db: Session = Depends(get_db)):
-    return auth_views.verify_email(verification_token, db)
-
-# User login route
-@router.post("/login", response_model=dict)
-def login(form_data: OAuth2PasswordRequestForm = Depends()):
-    return auth_views.login(form_data)
-
-# User authentication (protected route)
-@router.get("/protected", response_model=str)
-def protected_route(token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)):
-    return auth_views.protected_route(token)
+    # The user exists in the database, and you can render the protected route template
+    return templates.TemplateResponse("protected.html", {"request": request, "user": db_user.username})