from flask import Flask, request, jsonify
import pickle
import numpy as np

app = Flask(__name__)

# Load model
with open("model/xgb_superkart_model.pkl", "rb") as f:
    model = pickle.load(f)

@app.route("/", methods=["GET"])
def root():
    return jsonify({"message": "SuperKart Backend is live!"})

@app.route("/predict", methods=["POST"])
def predict():
    try:
        content = request.json
        data = content.get("data")
        if not data or not isinstance(data, list):
            return jsonify({"error": "Invalid input format. Expecting { 'data': [[...]] }"}), 400

        arr = np.array(data)
        preds = model.predict(arr)
        return jsonify({"prediction": preds.tolist()})
    except Exception as e:
        return jsonify({"error": str(e)}), 500

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