| """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") |
|
|
