Spaces:
Sleeping
Sleeping
File size: 1,208 Bytes
f639e70 5d31d6a f639e70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from starlette.exceptions import HTTPException as StarletteHTTPException
from fastapi.responses import RedirectResponse
from routes import auth
from routes import user
from routes import ml_models
from routes import chat
from routes import assisstant
import logging
logging.basicConfig(level=logging.INFO)
app = FastAPI()
app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")
app.include_router(auth.router)
app.include_router(user.router)
app.include_router(ml_models.router)
app.include_router(chat.router)
app.include_router(assisstant.router)
@app.get("/")
async def root():
return RedirectResponse(url="/login")
@app.exception_handler(StarletteHTTPException)
async def custom_404_handler(request: Request, exc: StarletteHTTPException):
if exc.status_code == 404:
return templates.TemplateResponse("404.html", {"request": request}, status_code=404)
return HTMLResponse(f"Error {exc.status_code}: {exc.detail}", status_code=exc.status_code)
|