fixed code

ai-moderation/app/exceptions/handlers.py

@@ -2,8 +2,10 @@ 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
@@ -12,8 +14,15 @@ async def invalid_text_exception_handler(
       code="INVALID_TEXT",
       message=exc.message
   )
+  return JSONResponse(status_code=400, content=response.model_dump())
 
-  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())