Update app.py

app.py

@@ -1,17 +1,47 @@
 from flask import Flask, request, jsonify
 import joblib
 import numpy as np
+import traceback
 
 app = Flask(__name__)
 
+# Load model
 model = joblib.load("predictive_model.pkl")
 
-@app.route("/")
+@app.route("/", methods=["GET"])
 def home():
-    return "Predictive Maintenance API Running"
+    return "Predictive Maintenance API Running 🚀"
+
 
 @app.route("/predict", methods=["POST"])
 def predict():
-    data = request.json["features"]
-    prediction = model.predict([data])
-    return jsonify({"prediction": int(prediction[0])})
+    try:
+        data = request.get_json()
+
+        # Expecting named features
+        features = [
+            data["Air_temperature"],
+            data["Process_temperature"],
+            data["Rotational_speed"],
+            data["Torque"],
+            data["Tool_wear"],
+            data["Type_L"],
+            data["Type_M"]
+        ]
+
+        prediction = model.predict([features])
+
+        return jsonify({
+            "prediction": int(prediction[0]),
+            "status": "No Failure" if prediction[0] == 0 else "Machine Failure"
+        })
+
+    except Exception as e:
+        return jsonify({
+            "error": str(e),
+            "trace": traceback.format_exc()
+        })
+
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)