Rewrite [https://github.com/labstructbioinf/pLM-BLAST/blob/main/scripts/plot.py](url) and integrate with the current version.