from fastapi import Request
from fastapi.responses import JSONResponse

from app.exceptions.invalid_text_exception import InvalidTextException
from app.exceptions.invalid_image_exception import InvalidImageException
from app.models.response.error_response import ErrorResponse


async def invalid_text_exception_handler(
    request: Request,
    exc: InvalidTextException
):
  response = ErrorResponse(
      code="INVALID_TEXT",
      message=exc.message
  )
  return JSONResponse(status_code=400, content=response.model_dump())


async def invalid_image_exception_handler(
    request: Request,
    exc: InvalidImageException
):
  response = ErrorResponse(
      code="INVALID_IMAGE",
      message=exc.message
  )
  return JSONResponse(status_code=400, content=response.model_dump())