python gen_diagram.py