Hi Prithiviraj,
Unfortunately I haven’t quite made this to a Python package. You can download this project and place it as a submodule in your code if you’d like to use it for your work.