import os
import shutil
import argparse

os.makedirs('manual_test/unknown/real', exist_ok=True)
os.makedirs('manual_test/unknown/screen', exist_ok=True)

parser = argparse.ArgumentParser(description="Copy local manual test images into manual_test/unknown.")
parser.add_argument(
    "source_dir",
    nargs="?",
    default=os.environ.get("SALES_CODE_SOURCE_IMAGE_DIR", ""),
    help="Directory containing the source images, or set SALES_CODE_SOURCE_IMAGE_DIR.",
)
args = parser.parse_args()

source_dir = args.source_dir
if not source_dir:
    raise SystemExit("Provide source_dir or set SALES_CODE_SOURCE_IMAGE_DIR.")

files = {
    'selfie_person_1782829299972.png': 'real/selfie.png',
    'room_1782829339573.png': 'real/room.png',
    'books_1782829349811.png': 'real/books.png',
    'flower_1782829361666.png': 'real/flower.png',
    'outdoor_1782829373383.png': 'real/outdoor.png',
    'laptop_screen_1782829390369.png': 'screen/laptop.png'
}

for src, dst in files.items():
    src_path = os.path.join(source_dir, src)
    if os.path.exists(src_path):
        shutil.copy(src_path, os.path.join('manual_test/unknown', dst))
        print(f"Copied {src} to {dst}")