app.py

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

# Load the trained pipeline
model = joblib.load("best_pipeline.joblib")

app = Flask(__name__)

@app.route("/")
def home():
    return {"message": "SuperKart Forecasting Service is running!"}

@app.route("/predict", methods=["POST"])
def predict():
    try:
        data = request.get_json(force=True)

        # Convert JSON to DataFrame
        df = pd.DataFrame(data)

        # Predict
        preds = model.predict(df)

        return jsonify({"predictions": preds.tolist()})
    except Exception as e:
        return jsonify({"error": str(e)}), 400

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