from fastapi import APIRouter
import cv2

from app.services.classification_service import classify_region

router = APIRouter(tags=["ML"])


@router.post("/classify")
async def classify(image_path: str):

    image = cv2.imread(image_path)

    if image is None:
        return {
            "error": "Image not found"
        }

    prediction = classify_region(image)

    return {
        "prediction": prediction
    }