Example #1
0
def generate_ast(code):
    ast = parse_program(code)
    dot = Digraph(comment="AST", format='png')
    dot.lid = 0
    dot.aid = 0
    viz_rec(dot, ast)
    image_raw = dot.pipe()
    return Image.open(io.BytesIO(image_raw))