Update app.py

app.py

@@ -1,37 +1,49 @@
 from flask import Flask, request, jsonify
+
 import joblib
-import numpy as np
-import traceback
 
-# Initialize Flask app
+import pandas as pd
+
+
+
+
 app = Flask(__name__)
 
-# Load model
 model = joblib.load("tuned_gradient_boosting_model.pkl")
 
-# Home route
+
+
+
 @app.route('/')
+
 def home():
-    return "Sales Prediction API is running."
 
-# Prediction route
+    return "Welcome to SuperKart Sales Forecast API"
+
+
+
+
 @app.route('/predict', methods=['POST'])
+
 def predict():
+
     try:
-        # Parse JSON input
-        data = request.get_json(force=True)
 
-        # Extract features as a 2D array (expecting list or single sample dict)
-        features = np.array(data['features']).reshape(1, -1)
+        data = request.get_json()
+
+        input_df = pd.DataFrame([data])
 
-        # Make prediction
-        prediction = model.predict(features)
+        prediction = model.predict(input_df)[0]
 
-        # Return prediction as JSON
-        return jsonify({"predicted_sales": prediction[0]})
+        return jsonify({'Predicted_Sales': round(float(prediction), 2)})
 
     except Exception as e:
-        return jsonify({"error": str(e), "trace": traceback.format_exc()})
+
+        return jsonify({'error': str(e)})
+
+
+
 
 if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=5000)
+
+    app.run(host='0.0.0.0', port=7860)