| | """API controllers for request handling and validation.""" |
| |
|
| | import asyncio |
| | from fastapi import HTTPException |
| |
|
| | from app.core.logging import logger |
| | from app.services.base import InferenceService |
| | from app.api.models import ImageRequest, PredictionResponse |
| |
|
| |
|
| | class PredictionController: |
| | """Controller for prediction endpoints.""" |
| |
|
| | @staticmethod |
| | async def predict( |
| | request: ImageRequest, |
| | service: InferenceService |
| | ) -> PredictionResponse: |
| | """Run inference using the configured service.""" |
| | try: |
| | if not service or not service.is_loaded: |
| | raise HTTPException(503, "Service not available") |
| |
|
| | if not request.image.mediaType.startswith('image/'): |
| | raise HTTPException(400, f"Invalid media type: {request.image.mediaType}") |
| |
|
| | return await asyncio.to_thread(service.predict, request) |
| |
|
| | except HTTPException: |
| | raise |
| | except ValueError as e: |
| | logger.error(f"Invalid input: {e}") |
| | raise HTTPException(400, str(e)) |
| | except Exception as e: |
| | logger.error(f"Prediction failed: {e}") |
| | raise HTTPException(500, "Internal server error") |
| |
|
