如何将创建单个镜像时使用的随机选择镜像的文件名附加到csv文件中？

import os
import random
import csv

def random_image_filename(directory):
    filenames = os.listdir(directory)
    return random.choice(filenames)

def append_filename_to_csv(filename, csv_file):
    with open(csv_file, 'a', newline='') as file:
        writer = csv.writer(file)
        writer.writerow([filename])

def main():
    image_directory = '/path/to/image/directory'
    csv_file = '/path/to/csv/file.csv'
    
    random_filename = random_image_filename(image_directory)
    append_filename_to_csv(random_filename, csv_file)

if __name__ == '__main__':
    main()