Update app/schemas/categories.py

app/schemas/categories.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 from pydantic import BaseModel, Field
@@ -14,6 +16,15 @@ class CategoryPrediction(BaseModel):
     title: str = Field(..., description="High-level category title.")
     subcategory: str = Field(..., description="Specific subcategory chosen by the model.")
 
+
+class CategorizeResponse(BaseModel):
+    status: str = Field(..., description="Outcome of the request (success/fail).")
+    data: Optional[CategoryPrediction] = Field(
+        None, description="Predicted category payload when successful."
+    )
+    message: Optional[str] = Field(None, description="Error message when failed.")
+
+
 # from pydantic import BaseModel, Field