initial commit

app/schemas.py

@@ -0,0 +1,23 @@
+from typing import List, Optional, Dict, Any
+from pydantic import BaseModel
+
+
+class ObjectResult(BaseModel):
+    bbox_xyxy: List[float]
+    det_confidence: float
+    det_class_id: int
+    det_class_name: str
+    cls_class_id: int
+    cls_class_name: str
+    cls_confidence: float
+
+
+class PredictResponse(BaseModel):
+    success: bool
+    image_width: int
+    image_height: int
+    num_objects: int
+    results: List[ObjectResult]
+    annotated_image_base64: Optional[str] = None
+    saved_outputs: Optional[Dict[str, Any]] = None
+    message: str = "ok"