XML转换成图片的背景如何设置？

<image>
  <background>red</background>
  <width>500</width>
  <height>300</height>
  <text>Hello, World!</text>
</image>

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

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

    background_color = root.find('background').text
    width = int(root.find('width').text)
    height = int(root.find('height').text)
    text = root.find('text').text

    # 颜色转换，这里简单处理，实际应用中需要更健壮的处理
    color_map = {'red': (255, 0, 0), 'green': (0, 255, 0), 'blue': (0, 0, 255)}
    try:
        bg_color = color_map[background_color]
    except KeyError:
        print(f"Unknown background color: {background_color}")
        return None

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

    # 这里假设你已经安装了合适的字体，否则需要调整字体路径
    try:
        font = ImageFont.truetype("arial.ttf", 30)  # 替换成你的字体文件
        draw.text((10, 10), text, font=font, fill=(0, 0, 0))
    except IOError:
        print("Font file not found.")
        return None

    img.save('output.png')
    return 'output.png'

xml_file = 'image.xml'
output_file = xml_to_image(xml_file)
if output_file:
    print(f"Image saved to {output_file}")