initial commit

backend/app/models/prediction.py

@@ -0,0 +1,21 @@
+from pydantic import BaseModel, Field
+from typing import List
+
+class HousePredictionInput(BaseModel):
+    square_feet: float = Field(..., gt=0, description="Square footage of the house")
+    bedrooms: int = Field(..., ge=1, description="Number of bedrooms")
+    bathrooms: float = Field(..., gt=0, description="Number of bathrooms")
+    
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "square_feet": 2000,
+                "bedrooms": 3,
+                "bathrooms": 2.5
+            }
+        }
+
+class HousePredictionOutput(BaseModel):
+    predicted_price: float
+    confidence_score: float
+    similar_listings: List[float]