import os

def pad_prefix(filename):
  parts = filename.split('.', 1)
  print(parts)
  if len(parts) == 2 and parts[0].isdigit():
    prefix = parts[0].zfill(5)
    return f"{prefix}.{parts[1]}"
  return filename

def rename_images(root_dir):
  for dirpath, _, filenames in os.walk(root_dir):
    print(filenames)
    for filename in filenames:
      if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp')):
        new_name = pad_prefix(filename)
        if new_name != filename:
          src = os.path.join(dirpath, filename)
          dst = os.path.join(dirpath, new_name)
          os.rename(src, dst)
          print(f"Renamed: {src} -> {dst}")

if __name__ == "__main__":
  root = os.path.dirname(os.path.abspath(__file__))
  print(root)
  rename_images(root)