from fastapi import APIRouter

from app.schemas.models import (
    DecisionRequest,
    DecisionResponse,
    ImagingReportRequest,
    ImagingReportResponse,
)
from app.services.report_generator import generate_decision, generate_imaging_report

router = APIRouter(prefix="/report", tags=["report"])


@router.post("/imaging", response_model=ImagingReportResponse)
def report_imaging(body: ImagingReportRequest) -> ImagingReportResponse:
    return generate_imaging_report(body)


@router.post("/decision", response_model=DecisionResponse)
def report_decision(body: DecisionRequest) -> DecisionResponse:
    return generate_decision(body)