#!/usr/bin/env python3
import requests
from pathlib import Path

PEXELS_KEY = "YDqxXGXvOaPQnitSVyxbhghbH49w95DV2fbAkynBjNVvus1TI60Qffdo"
OUT = Path('/opt/mia/workspace/clientes/px3lab/lancamento_photorf2/criativos/pexels_faces')

# mais IDs pra pegar HOMEM formando
IDS = [25643178, 25643176, 36432729, 25643183]

def fetch(pid):
    headers = {'Authorization': PEXELS_KEY}
    r = requests.get(f'https://api.pexels.com/v1/photos/{pid}', headers=headers, timeout=30)
    r.raise_for_status()
    return r.json()

for pid in IDS:
    try:
        data = fetch(pid)
        src = data['src']
        url = src.get('large2x') or src.get('large')
        dest = OUT / f'{pid}.jpg'
        r = requests.get(url, timeout=60); r.raise_for_status()
        dest.write_bytes(r.content)
        print(f'{pid}: {dest.name} ({dest.stat().st_size//1024}KB)')
    except Exception as e:
        print(f'{pid}: FAIL {e}')

# Busca mais - "graduation friends" e "graduates group"
QUERIES2 = ["graduation friends", "graduates group", "university friends graduation", "black man graduation cap", "african graduate man"]

for q in QUERIES2:
    r = requests.get(
        'https://api.pexels.com/v1/search',
        headers={'Authorization': PEXELS_KEY},
        params={'query': q, 'orientation': 'square', 'per_page': 10, 'size': 'medium'},
        timeout=30,
    )
    data = r.json()
    print(f'\n=== "{q}" ===')
    for ph in data.get('photos', []):
        alt = ph.get('alt', '')[:60]
        print(f'  id={ph["id"]:>10} por={ph["photographer"][:20]:<20} alt="{alt}"')