#!/usr/bin/env python3
import sys
import base64

# Read arguments: payload and filename
if len(sys.argv) != 3:
    print("Usage: base64_to_image.py <base64_payload> <filename>")
    sys.exit(1)

payload = sys.argv[1]
filename = sys.argv[2]

try:
    # Decode base64 and save
    image_data = base64.b64decode(payload)
    with open(filename, 'wb') as f:
        f.write(image_data)
    print(f"Saved {len(image_data)} bytes to {filename}")
except Exception as e:
    print(f"Error: {e}")
    sys.exit(1)