import os, sys, random, argparse
from pathlib import Path

VALID_EXTS = {".jpg", ".jpeg", ".png", ".gif", ".webp"}

def shuffle_photos(folder):
    images = sorted([f for f in folder.iterdir() if f.suffix.lower() in VALID_EXTS])
    if not images:
        print(f"No images found in {folder}"); return
    random.shuffle(images)
    padding = len(str(len(images)))
    print(f"Found {len(images)} images. Shuffling...")
    temp_paths = []
    for i, img in enumerate(images):
        temp = img.with_name(f"__tmp_{i}{img.suffix.lower()}")
        img.rename(temp)
        temp_paths.append((temp, img.suffix.lower()))
    for i, (temp, ext) in enumerate(temp_paths, start=1):
        final = folder / f"{str(i).zfill(padding)}{ext}"
        print(f"  {temp.name} → {final.name}")
        temp.rename(final)
    print(f"\nDone! {len(images)} images shuffled.")

parser = argparse.ArgumentParser()
parser.add_argument("folder")
args = parser.parse_args()
folder = Path(args.folder).resolve()
if not folder.exists() or not folder.is_dir():
    print(f"Error: '{folder}' is not a valid directory."); sys.exit(1)
shuffle_photos(folder)