Update api.py

api.py

@@ -233,10 +233,11 @@ def get_feature_schema(disaster_type: str):
 
 @app.post("/predict/flood")
 def predict_flood(req: FloodPredictionRequest):
-    errors = flood_predictor.validate_input(req.features)
+    features = req.features.model_dump()  # or .dict() for Pydantic v1
+    errors = flood_predictor.validate_input(features)
     if errors:
         raise HTTPException(422, {"validation_errors": errors})
-    result = flood_predictor.predict(req.features, req.n_mc_samples)
+    result = flood_predictor.predict(features, req.n_mc_samples)
     return {"disaster_type": "flood", **prediction_result_to_dict(result)}
 
 
@@ -319,31 +320,22 @@ import numpy as np  # needed for explain endpoint
 
 @app.get("/predict/cyclone/live")
 def predict_cyclone_live():
-    """
-    Fetch live IMD cyclone bulletin,
-    extract parameters,
-    engineer features,
-    run cyclone predictor.
-    """
     bulletin = imd_fetcher.fetch_hourly_bulletin()
-
     if bulletin["status"] != "success":
         raise HTTPException(502, f"IMD fetch failed: {bulletin.get('message')}")
 
     raw_params = imd_fetcher.parse_cyclone_parameters(bulletin["content"])
-
     if not raw_params:
         raise HTTPException(404, "No cyclone parameters detected in IMD bulletin")
 
-    features = cyclone_engineer.engineer_features(raw_params)
+    engineered = cyclone_engineer.engineer_features(raw_params)
+    features = cyclone_engineer.to_model_features(engineered)  # ← new step
 
-    # Validate against model schema
     errors = cyclone_predictor.validate_input(features)
     if errors:
         raise HTTPException(422, {"validation_errors": errors})
 
     result = cyclone_predictor.predict(features, 50)
-
     return {
         "source": "IMD RSMC live bulletin",
         "raw_parameters": raw_params,