That sounds like something that could be done using a Jupyter Notebook and SageMath.