Update src/price_predictor.py

src/price_predictor.py

@@ -24,7 +24,8 @@ def _heuristic_estimate(input_data: dict[str, Any]) -> dict[str, Any]:
     base -= (km / 1000) * 35
     base += (hp_kw - 100) * 45
 
-    predicted_class = str(input_data.get("body_type", "")).lower()
+    # Extract brand from make_model for adjustments
+    make_model = str(input_data.get("make_model", "")).lower()
     luxury_adjustments = {
         "rolls": 70000,
         "bentley": 45000,
@@ -50,19 +51,19 @@ def _heuristic_estimate(input_data: dict[str, Any]) -> dict[str, Any]:
 
     applied = False
     for key, delta in luxury_adjustments.items():
-        if key in predicted_class:
+        if key in make_model:
             base += delta
             applied = True
             break
 
     if not applied:
         for key, delta in budget_adjustments.items():
-            if key in predicted_class:
+            if key in make_model:
                 base += delta
                 applied = True
                 break
 
-    if not applied and ("suv" in predicted_class or "benz" in predicted_class):
+    if not applied and ("suv" in make_model or "benz" in make_model):
         base += 3500
 
     estimated = max(5000, min(180000, base))