Renames jpg or png files to sequence 01.jpg and output jpg

rename.py

import os

def rename_files(extension):
    files = [f for f in os.listdir() if f.endswith(extension)]
    for i, file in enumerate(sorted(files), start=1):
        os.rename(file, f"{str(i).zfill(2)}{extension}")

if __name__ == "__main__":
    rename_files(".jpg")
    rename_files(".png")