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

app = Flask("Superkart Sales Predictor")
model = joblib.load("superkart.joblib")

@app.get('/')
def home():
  return "Welcome to the Superkart Sales Predictor API"

@app.post('/v1/sales')
def predict_sales():
  product_data = request.get_json()

  sample = {
        'Product_Type': product_data['Product_Type'],
        'Product_Sugar_Content': product_data['Product_Sugar_Content'],
        'Product_Weight': product_data['Product_Weight'],
        'Product_Allocated_Area': product_data['Product_Allocated_Area'],
        'Product_MRP': product_data['Product_MRP'],
        'Store_Id': product_data['Store_Id'],
        'Store_Size': product_data['Store_Size'],
        'Store_Type': product_data['Store_Type'],
        'Store_Location_City_Type': product_data['Store_Location_City_Type'],
        'Store_Establishment_Year': product_data['Store_Establishment_Year']  
  }
  input_data = pd.DataFrame([sample])
  input_data

  prediction = model.predict(input_data).tolist()[0]
  return jsonify({'prediction': prediction})

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