import os

if __name__ == "__main__":
    path = "/data/pufanyi/project/lmms-eval/tools/temp/2024-09"
    new_path = "/data/pufanyi/project/lmms-eval/tools/temp/processed_images_2"
    if not os.path.exists(new_path):
        os.makedirs(new_path)
    subjects = os.listdir(path)
    for subject in subjects:
        subject_folder = os.path.join(path, subject)
        if not os.path.isdir(subject_folder):
            continue
        print(f"Processing {subject_folder}")
        images = os.listdir(subject_folder)
        for id, image in enumerate(images):
            image_ext = image.split(".")[-1]
            new_image_name = f"{subject}_{id}.{image_ext}"
            os.rename(os.path.join(subject_folder, image), os.path.join(new_path, new_image_name))