Commit - 2025-12-10 20:22:17

app.py

@@ -6,24 +6,17 @@ from flask import Flask, request, jsonify  # For creating the Flask API
 from flask_cors import CORS
 
 superkart_api = Flask(__name__)
-# Enable CORS for all domains with explicit configuration
-CORS(superkart_api,
-     resources={r"/*": {
-         "origins": "*",
-         "methods": ["GET", "POST", "OPTIONS"],
-         "allow_headers": ["Content-Type", "Authorization"]
-     }})
+CORS(superkart_api)
+# Load the trained model at the start of the application
 model = joblib.load('superkart.joblib')
 
-@superkart_api.route('/', methods=['GET'])
+# Define an endpoint for the home route.  This method tells flask that when a user goes to the root URL, this function should be called.
+@superkart_api.get('/')
 def home():
     return "Welcome to the SuperKart Sales Prediction API!"
 
-@superkart_api.route('/health', methods=['GET'])
-def health():
-    return jsonify({'status': 'healthy', 'model_loaded': model is not None})
-
-@superkart_api.route('/predict', methods=['POST', 'OPTIONS'])
+# A method to take a single prediction request in JSON format and return the results in JSON
+@superkart_api.post('/predict')
 def predict():
     """
     API endpoint to predict sales using the trained model.
@@ -45,7 +38,8 @@ def predict():
     except Exception as e:
         return jsonify({'error': str(e)}), 400
 
-@superkart_api.route('/predict/batch', methods=['POST', 'OPTIONS'])
+# A method to take a batch prediction request in CSV file and return the results in JSON
+@superkart_api.post('/predict/batch')
 def predict_batch():
     """
     API endpoint to predict sales for a batch of inputs using the trained model.
@@ -66,7 +60,6 @@ def predict_batch():
     except Exception as e:
         return jsonify({'error': str(e)}), 400
 
+# We can run this app locally with just the python command.  This block enables that behavior.
 if __name__ == '__main__':
-    import os
-    port = int(os.environ.get('PORT', 7860))  # HF Spaces uses port 7860 by default
-    superkart_api.run(host='0.0.0.0', port=port, debug=False)
+    superkart_api.run(host='0.0.0.0', port=5001, debug=True)