Update app.py

app.py

@@ -52,6 +52,51 @@ def predict_need_maintenance ():
         }), 500
 
 
+# Define an endpoint to predict sales for Super Kart
+@pred_mainteanance_api.post ('/v1/EngPredMaintenanceForBatch')
+def predict_need_maintenance_for_batch ():
+    # Get JSON data from the request
+    engine_sensor_inputs = request.get_json ()
+
+    # validate request (json)
+    # if input is valid - return prediction
+    # in case of error - return appropriate error
+    try:
+        input_json = request.get_json()
+
+        # Convert JSON list → DataFrame
+        input_df = pd.DataFrame([input_json])
+
+        # Validate entire batch
+        validated_df = validate_and_prepare_input(input_df, model)
+
+        # predict for given input
+        predictions = model.predict(validated_df)
+
+        # Convert numpy array → Python list
+        prediction_list = predictions.tolist()
+
+        return jsonify({
+                "status": "success",                 # overall batch status
+                "total_records": len(prediction_list),
+                "predictions": prediction_list,     # simple list version
+            })        
+
+    except InputValidationError as e:
+        return jsonify({
+            "status": "error",
+            "error_type": "validation_error",
+            "message": str(e)
+        }), 400
+
+    except Exception as e:
+        return jsonify({
+            "status": "error",
+            "error_type": "internal_error",
+            "message": "Unexpected server error"
+        }), 500
+
+
 # Run the Flask app
 if __name__ == "__main__":
     import os