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))