Update app/api/routes.py

app/api/routes.py

@@ -2,21 +2,22 @@ import time
 from typing import Any
 
 from fastapi import APIRouter, Depends, HTTPException
+from fastapi.responses import JSONResponse
 
 from app.dependencies import get_api_logger, get_category_service
-from app.schemas.categories import CategorizeRequest, CategoryPrediction
+from app.schemas.categories import CategorizeRequest, CategorizeResponse
 from app.services.autocategorizer import AutoCategoryService
 from app.services.api_logger import ApiLogger
 
 router = APIRouter()
 
 
-@router.post("/categorize", response_model=CategoryPrediction, summary="Categorize a transaction note")
+@router.post("/categorize", response_model=CategorizeResponse, summary="Categorize a transaction note")
 async def categorize_transaction(
     payload: CategorizeRequest,
     service: AutoCategoryService = Depends(get_category_service),
     api_logger: ApiLogger = Depends(get_api_logger),
-) -> CategoryPrediction:
+) -> CategorizeResponse | JSONResponse:
     started_at = time.monotonic()
     try:
         result = await service.categorize(payload.notes)
@@ -26,7 +27,7 @@ async def categorize_transaction(
             response_time=time.monotonic() - started_at,
             user_id=payload.user_id,
         )
-        return result
+        return CategorizeResponse(status="success", data=result)
     except HTTPException as exc:
         await api_logger.log_categorization(
             name="Auto Expense Categorization",
@@ -35,7 +36,10 @@ async def categorize_transaction(
             user_id=payload.user_id,
             error_message=_normalize_error_detail(exc.detail),
         )
-        raise
+        return JSONResponse(
+            status_code=exc.status_code,
+            content={"status": "fail", "message": _normalize_error_detail(exc.detail)},
+        )
     except Exception as exc:
         await api_logger.log_categorization(
             name="Auto Expense Categorization",
@@ -44,7 +48,10 @@ async def categorize_transaction(
             user_id=payload.user_id,
             error_message=str(exc),
         )
-        raise
+        return JSONResponse(
+            status_code=500,
+            content={"status": "fail", "message": "Internal server error"},
+        )
 
 
 def _normalize_error_detail(detail: Any) -> str: