src/api.py

from __future__ import annotations

from pathlib import Path

from fastapi import FastAPI

from src.data import load_training_data
from src.modeling import DEFAULT_MODEL_PATH, HousePriceModel
from src.modeling import train_model
from src.schemas import HouseFeatures, PredictionResponse


def create_app(model_path: str | Path = DEFAULT_MODEL_PATH) -> FastAPI:
    app = FastAPI(title="House Price Prediction API", version="1.0.0")
    model_path = Path(model_path)
    if not model_path.exists():
        train_model(load_training_data(), artifact_path=model_path)
    model = HousePriceModel.load(model_path)

    @app.get("/health")
    def health() -> dict[str, str]:
        return {"status": "ok"}

    @app.get("/metrics")
    def metrics() -> dict[str, float]:
        return model.metrics

    @app.post("/predict", response_model=PredictionResponse)
    def predict(features: HouseFeatures) -> PredictionResponse:
        return PredictionResponse(predicted_price=model.predict(features.model_dump()))

    return app


app = create_app()