from flask import Flask, request, jsonify
from flask_cors import CORS
import pandas as pd, joblib, os

app = Flask(__name__)
CORS(app)

model = joblib.load("final_random_forest_model.pkl")

FEATURE_COLUMNS = [
    "Product_Weight", "Product_Allocated_Area", "Product_MRP",
    "Store_Establishment_Year", "Store_Size", "Store_Location_City_Type",
    "Store_Type", "Product_Prefix", "Product_Num", "Store_Age"
]

@app.route("/", methods=["GET"])
def home():
    return "✅ SuperKart Forecast API is running"

@app.route("/predict", methods=["POST"])
def predict():
    data = request.get_json(force=True)["data"]
    df = pd.DataFrame(data)[FEATURE_COLUMNS]
    preds = model.predict(df)
    return jsonify({"predictions": preds.tolist()})

if __name__ == "__main__":
    port = int(os.environ.get("PORT", 7860))
    app.run(host="0.0.0.0", port=port)