import joblib
import pandas as pd
from flask import Flask, request, jsonify

app = Flask(__name__)
model = joblib.load("superkart_sales_model.joblib")

@app.route("/health", methods=["GET"])
def health():
    return jsonify({"status": "ok"})

@app.route("/predict", methods=["POST"])
def predict():
    data = request.get_json()
    if data is None:
        return jsonify({"error": "No JSON provided"}), 400

    if isinstance(data, dict):
        df = pd.DataFrame([data])
    else:
        df = pd.DataFrame(data)

    # 🔴 REPLICATE FEATURE ENGINEERING
    df["Store_Age"] = 2025 - df["Store_Establishment_Year"]
    df["Product_Category_Code"] = df["Product_Id"].str[:2]

    preds = model.predict(df)
    return jsonify({"predictions": preds.tolist()})

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)