Update app.py

app.py

@@ -9,6 +9,7 @@ from typing import Any
 import gradio as gr
 
 from src.price_predictor import predict_price
+from src import price_predictor
 from src.recommendation_engine import generate_recommendation
 from src.utils import format_currency_chf, to_float
 from src.vision_analyzer import analyze_car_image
@@ -183,14 +184,20 @@ def _run_advisor(
     age = _normalize_age(car_age_years, "Age")
     make_model_input = make_model.strip() if make_model and make_model.strip() else predicted_class
 
-    # For price input, only use make_model; body_type from vision is not reliable vehicle type info
+    # For manual input without image, use heuristic (ML model needs too many features we don't have)
+    # For image-based input, use ML model if available
     price_input = {
         "make_model": make_model_input,
-        "body_type": None,  # Let preprocessor handle imputation instead of using predicted class
+        "body_type": predicted_class if vision_results.get("method") != "manual_input" else None,
         "km": to_float(mileage_km, 80000),
         "age": age,
     }
-    price_prediction = predict_price(price_input)
+    
+    # If manual input (no vision results), use heuristic directly to get brand-aware pricing
+    if vision_results.get("method") == "manual_input":
+        price_prediction = price_predictor._heuristic_estimate(price_input)
+    else:
+        price_prediction = predict_price(price_input)
 
     user_inputs = {
         "budget_chf": to_float(budget_chf, 0),