Update app/models.py

app/models.py

@@ -1,6 +1,5 @@
 from pydantic import BaseModel
 from typing import Optional
-from datetime import datetime
 
 class PredictionResponse(BaseModel):
     request_id: str
@@ -9,8 +8,17 @@ class PredictionResponse(BaseModel):
     prediction: str
     confidence: float
     model_version: str
-    status: str = "completed"
+    timestamp: str
+    status: str
+    segmentation_path: Optional[str] = None
 
-class HistoryItem(PredictionResponse):
+class HistoryItem(BaseModel):
+    request_id: str
+    filename: str
+    image_path: str
+    prediction: str
+    confidence: float
+    model_version: str
     timestamp: str
+    status: str
     segmentation_path: Optional[str] = None