如何在python中循环所有图像并为所有图像添加边框？

from PIL import Image, ImageDraw
import os

def add_border(image_path, output_path, border_color, border_width):
    # 打开图像
    image = Image.open(image_path)
    
    # 计算新图像的尺寸
    new_width = image.width + 2 * border_width
    new_height = image.height + 2 * border_width
    new_size = (new_width, new_height)
    
    # 创建一个新的带有边框的图像
    new_image = Image.new(image.mode, new_size, border_color)
    new_image.paste(image, (border_width, border_width))
    
    # 保存图像
    new_image.save(output_path)

def add_border_to_all_images(input_dir, output_dir, border_color, border_width):
    # 确保输出目录存在
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    
    # 遍历输入目录中的所有图像文件
    for filename in os.listdir(input_dir):
        if filename.endswith('.jpg') or filename.endswith('.png'):
            # 构建输入和输出路径
            input_path = os.path.join(input_dir, filename)
            output_path = os.path.join(output_dir, filename)
            
            # 调用添加边框函数
            add_border(input_path, output_path, border_color, border_width)

input_directory = '输入目录路径'
output_directory = '输出目录路径'
border_color = '边框颜色（例如：'black'）'
border_width = '边框宽度（整数）'

add_border_to_all_images(input_directory, output_directory, border_color, border_width)