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

app = Flask(__name__)
model = joblib.load("superkart_best_model.pkl")

FEATURE_COLUMNS = ["Product_Weight","Product_Sugar_Content","Product_Allocated_Area","Product_Type","Product_MRP","Store_Establishment_Year","Store_Size","Store_Location_City_Type","Store_Type","Store_Current_Age","Product_Type_Group"]

@app.route("/predict", methods=["POST"])
def predict():
    data = request.get_json()
    df_input = pd.DataFrame([data])
    df_input = df_input.reindex(columns=FEATURE_COLUMNS, fill_value=0)
    preds = model.predict(df_input)
    return jsonify({"predictions": preds.tolist()})

@app.route("/", methods=["GET"])
def home():
    return "SuperKart API Ready!"

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