app/api/routes/prediction.py

"""Prediction API routes."""

from fastapi import APIRouter, Depends

from app.api.controllers import PredictionController
from app.api.models import ImageRequest, PredictionResponse
from app.core.dependencies import get_inference_service
from app.services.base import InferenceService

router = APIRouter()


@router.post("/predict", response_model=PredictionResponse)
async def predict(
    request: ImageRequest,
    service: InferenceService = Depends(get_inference_service)
):
    """
    Run inference on base64-encoded image.

    Returns prediction, confidence, predicted label, model name, and media type.
    """
    return await PredictionController.predict(request, service)