app.py

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

# Load model
model = joblib.load("random_forest_final_model.pkl")

# Create Flask app
app = Flask(__name__)

# Route: Home
@app.route('/')
def home():
    return "Welcome to SuperKart Sales Forecasting API"

# Route: Predict
@app.route('/predict', methods=['POST'])
def predict():
    try:
        data = request.get_json()
        df = pd.DataFrame([data])  # convert dict to DataFrame
        prediction = model.predict(df)[0]
        return jsonify({'Predicted_Sales': round(prediction, 2)})
    except Exception as e:
        return jsonify({'error': str(e)})

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