from fastapi import APIRouter

from services.gpu_monitor import get_gpu_info

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


@router.get("/gpu")
async def gpu_status():
    info = await get_gpu_info()
    if not info:
        return {"error": {"message": "GPU info unavailable", "type": "api_error", "code": "gpu_unavailable"}}
    return info