Python Code Diagram Generator