I've had a go at making this a plug-in it's very early days https://github.com/datajoely/kedro-rich