Plot functions for ast in generation package

[dr2001dlr]

Motivation / Current Behaviour

No response

Enhancement description

Improve graph visualization for ast in generation package. Output the graph in text format or as a png file.

Additional context

No response

Checklist

• Attached labels, especially loc:: or model:: labels.
• Linked to project