帮我用PIL库写一个给图片添加水印的代码

好的，以下是使用PIL库给图片添加水印的Python代码示例：

from PIL import Image, ImageDraw, ImageFont

def add_watermark(text, input_image_path, output_image_path):
    photo = Image.open(input_image_path)

    # 设置文本字体
    font = ImageFont.truetype("arial.ttf", 36)

    # 创建绘图对象
    drawing = ImageDraw.Draw(photo)

    # 在图片指定位置绘制文本
    textwidth, textheight = drawing.textsize(text, font)
    width, height = photo.size
    pos = (width - textwidth)//2, (height - textheight)//2
    drawing.text(pos, text, fill=(255, 255, 255), font=font)

    # 保存处理后的图片
    photo.save(output_image_path)

# 测试
add_watermark("chit-chat AI", "input.jpg", "output.jpg")

以上代码将输入图片（input.jpg）中心位置添加了“chit-chat AI”水印，并将结果保存到输出图片(output.jpg)。如果需要修改水印文本、字体、字号、颜色、位置等参数，可以根据实际需求进行调整。