Update call_models.py

call_models.py

@@ -3,14 +3,75 @@ from datetime import datetime
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from dotenv import load_dotenv
+import numpy as np
 
 from predict_from_supabase import predict_replacements
+from model import DegradationModel, COMPONENT_NAMES
 
 load_dotenv()
 
 app = Flask(__name__)
 CORS(app)
 
+# Load the default degradation model once at startup
+_DEFAULT_MODEL_PATH = "model.npz"
+_degradation_model: DegradationModel | None = None
+
+def _get_model(path: str = _DEFAULT_MODEL_PATH) -> DegradationModel:
+    global _degradation_model
+    if _degradation_model is None:
+        _degradation_model = DegradationModel.load(path)
+    return _degradation_model
+
+
+@app.route("/api/predict", methods=["POST"])
+def predict():
+    data = request.json or {}
+
+    h0_raw = data.get("h0")
+    X_raw  = data.get("X")
+    weeks  = float(data.get("weeks", 6))
+    model_path = data.get("model_path", _DEFAULT_MODEL_PATH)
+
+    if h0_raw is None:
+        return jsonify({"error": "h0 (current health state) is required"}), 400
+    if X_raw is None:
+        return jsonify({"error": "X (input conditions) is required"}), 400
+
+    try:
+        model = _get_model(model_path)
+    except FileNotFoundError:
+        return jsonify({"error": f"model file not found: {model_path}"}), 404
+
+    h0 = np.array(h0_raw, dtype=float)
+    X  = np.array(X_raw,  dtype=float)
+
+    if h0.shape != (model.N,):
+        return jsonify({"error": f"h0 must have {model.N} elements, got {h0.shape}"}), 400
+    if X.shape != (model.C,):
+        return jsonify({"error": f"X must have {model.C} elements, got {X.shape}"}), 400
+
+    # Euler integration from h0, deterministic, 100 steps per week
+    tau_hours = weeks * 7 * 24.0
+    n_steps   = int(weeks) * 20
+    dt        = tau_hours / n_steps
+    steps_per_week = n_steps // int(weeks)
+
+    h = h0.copy()
+    weekly_snapshots = [h0.tolist()]
+
+    for step in range(1, n_steps + 1):
+        h = np.clip(h + dt * model.f(h, X), 0.0, 1.0)
+        if step % steps_per_week == 0:
+            weekly_snapshots.append(h.tolist())
+
+    return jsonify({
+        "component_names": COMPONENT_NAMES,
+        "weeks": weeks,
+        "weekly_health": weekly_snapshots,   # index 0 = now, index k = after k weeks
+        "final_health": h.tolist(),
+    })
+
 
 @app.route("/api/schedule", methods=["POST"])
 def schedule():