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

app = Flask(__name__)

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

@app.route("/")
def home():
    return "SuperKart Sales Forecast API is Live"

@app.route("/predict", methods=["POST"])
def predict():
    data = request.get_json(force=True)
    input_features = np.array(data["features"]).reshape(1, -1)
    prediction = model.predict(input_features)
    return jsonify({"predicted_sales": prediction[0]})

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