"""
Trata as fotos usadas nos criativos 02-10 v2.
Mesma abordagem do v1: brilho/contraste/saturacao suaves, salva _treated.jpg
"""
from pathlib import Path
from PIL import Image, ImageEnhance, ImageFilter

BASE = Path('/opt/mia/workspace/clientes/chacara-sonho-verde')
SRC = BASE / 'fotos_site'
DST = BASE / 'fotos_originais'
DST.mkdir(exist_ok=True)

FOTOS = [
    'foto_01.jpg',   # buffet farto
    'foto_02.jpg',   # cerimonia altar branco
    'foto_06.jpg',   # salao decorado
    'foto_20.jpg',   # fachada noite
    'foto_39.jpg',   # forno pizza
    'foto_hero.jpg', # aerea
    'foto_piscina.jpg', # piscina
]

def tratar(nome):
    src = SRC / nome
    dst = DST / nome.replace('.jpg', '_treated.jpg')
    if dst.exists():
        print(f"skip: {nome} (ja existe)")
        return
    if not src.exists():
        print(f"MISSING: {nome}")
        return
    img = Image.open(src).convert('RGB')
    # Contraste sutil
    img = ImageEnhance.Contrast(img).enhance(1.08)
    # Saturacao um pouco pra cima (natureza fica mais viva)
    img = ImageEnhance.Color(img).enhance(1.1)
    # Nitidez suave
    img = ImageEnhance.Sharpness(img).enhance(1.15)
    # Salva
    img.save(dst, 'JPEG', quality=92)
    print(f"OK: {dst.name} ({img.size})")

for f in FOTOS:
    tratar(f)