Generate Diagram From Python Code