main.py

from fastapi import FastAPI
from contextlib import asynccontextmanager
from models.routers import router as model_routes
from inferences import router as inferences_router
from metrics import router as metrics_router
from prometheus_fastapi_instrumentator import Instrumentator

tags_metadata = [
    {
        "name": "AI Backend",
        "description": "AI Backend Developer, Technical Test",
    }
]

@asynccontextmanager
async def lifespan(app: FastAPI):
    print("Application startup")
    yield
    print("Application shutdown")
    
    
    
app = FastAPI(
    title="AI Backend",
    description="Technical Test",
    contact={
        "name": "Mehrdad Motaghian",
        "url": "http://imehrdad.ir/",
        "email": "motaghian@outlook.com",
    },
    lifespan=lifespan,
    openapi_tags=tags_metadata
)
Instrumentator().instrument(app).expose(app)

app.include_router(model_routes)
app.include_router(inferences_router.router)
app.include_router(metrics_router.router)
app.include_router(metrics_router.router)

@app.get("/")
def read_root():
    return {"message": "AI Inference API is running!"}