Loginauth / main.py
Gregniuki's picture
Update main.py
44372ef
raw
history blame
1.3 kB
# main.py
from fastapi import FastAPI, Depends, HTTPException
from fastapi.requests import Request # Add this import
from fastapi.responses import HTMLResponse
import auth, tts
from auth import verify_token # Import a function for verifying tokens
app = FastAPI()
router = APIRouter()
# Include the authentication router with the prefix '/auth'
app.include_router(auth.router, prefix="/auth")
# Include the TTS router with the prefix '/tts'
app.include_router(tts.router, prefix="/tts")
# Dependency for verifying the user's token
def get_current_user(token: str = Depends(verify_token)):
if not token:
raise HTTPException(status_code=401, detail="Token not valid")
return token
@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})