如何自定义XML转换成图片的样式？

import xml.etree.ElementTree as ET
from PIL import Image, ImageDraw, ImageFont

def xml_to_image(xml_file, output_file):
    tree = ET.parse(xml_file)
    root = tree.getroot()

    # 获取图片尺寸
    width = int(root.get('width', 500))  # 默认宽度500
    height = int(root.get('height', 300)) # 默认高度300

    img = Image.new('RGB', (width, height), color = 'white')
    draw = ImageDraw.Draw(img)

    # 遍历XML节点，绘制元素
    for element in root.findall('.//element'):
        type = element.get('type')
        x = int(element.get('x'))
        y = int(element.get('y'))
        if type == 'rect':
            w = int(element.get('width'))
            h = int(element.get('height'))
            color = element.get('color', 'black')
            draw.rectangle([(x, y), (x + w, y + h)], fill=color)
        elif type == 'text':
            text = element.text
            font = ImageFont.load_default() # 可以替换成你喜欢的字体
            draw.text((x, y), text, font=font, fill='black')
        # ... 添加更多元素类型 ...

    img.save(output_file)


# 一个简单的XML例子
xml_data = """
<image width="600" height="400">
  <element type="rect" x="10" y="10" width="100" height="50" color="red"/>
  <element type="text" x="120" y="30">Hello, World!</element>
</image>
"""

with open("temp.xml", "w") as f:
    f.write(xml_data)

xml_to_image("temp.xml", "output.png")