fixed code

ai-moderation/app/main.py

@@ -1,25 +1,15 @@
 from fastapi import FastAPI
 
 from app.config.settings import settings
-from app.lifespan import lifespan
-
-from app.routers.text_moderation_router import (
-  router as moderation_router
-)
-
-from app.routers.image_moderation_router import (
-  router as image_router
-)
-
-
 from app.exceptions.handlers import (
-  invalid_text_exception_handler
+  invalid_text_exception_handler,
+  invalid_image_exception_handler
 )
-
-from app.exceptions.invalid_text_exception import (
-  InvalidTextException
-)
-
+from app.exceptions.invalid_text_exception import InvalidTextException
+from app.exceptions.invalid_image_exception import InvalidImageException
+from app.lifespan import lifespan
+from app.routers.image_moderation_router import router as image_router
+from app.routers.text_moderation_router import router as moderation_router
 
 app = FastAPI(
     title=settings.APP_NAME,
@@ -27,18 +17,8 @@ app = FastAPI(
     lifespan=lifespan
 )
 
+app.include_router(moderation_router)
+app.include_router(image_router)
 
-app.include_router(
-    moderation_router
-)
-
-
-app.include_router(
-    image_router
-)
-
-
-app.add_exception_handler(
-    InvalidTextException,
-    invalid_text_exception_handler
-)
+app.add_exception_handler(InvalidTextException, invalid_text_exception_handler)
+app.add_exception_handler(InvalidImageException, invalid_image_exception_handler)