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

ai-moderation/app/exceptions/invalid_image_exception.py

@@ -1,2 +1,5 @@
-class InvalidImageException(Exception):
+from app.exceptions.moderation_exception import ModerationException
+
+
+class InvalidImageException(ModerationException):
   pass

ai-moderation/app/exceptions/moderation_exception.py

@@ -1,4 +1,4 @@
 class ModerationException(Exception):
-
   def __init__(self, message: str):
-    self.message = message
+    self.message = message
+    super().__init__(message)