import os
from PIL import Image

# Source map from BRAIN (using the best versions available)
source_map = {
    "okvir_zarjavel.png": "/Users/davidkotnik/.gemini/antigravity/brain/998d0b10-1733-4e5b-85ed-249b986ba9b3/okvir_zarjavel_v3_1768956071037.png",
    "srce_postapo.png": "/Users/davidkotnik/.gemini/antigravity/brain/998d0b10-1733-4e5b-85ed-249b986ba9b3/srce_postapo_1768956084773.png",
    "merilec_zdravja.png": "/Users/davidkotnik/.gemini/antigravity/brain/998d0b10-1733-4e5b-85ed-249b986ba9b3/merilec_zdravja_v2_1768954479566.png",
    "gumb_start.png": "/Users/davidkotnik/.gemini/antigravity/brain/998d0b10-1733-4e5b-85ed-249b986ba9b3/gumb_recikliran_v2_1768954494464.png",
    "amnezija_maska.png": "/Users/davidkotnik/.gemini/antigravity/brain/998d0b10-1733-4e5b-85ed-249b986ba9b3/amnezija_maska_v2_1768954510228.png"
}

# Source map fallback (if v3 doesn't exist, use v2)
source_map_fallback = {
    "okvir_zarjavel.png": "/Users/davidkotnik/.gemini/antigravity/brain/998d0b10-1733-4e5b-85ed-249b986ba9b3/okvir_zarjavel_v2_1768954465913.png",
}

# Updated target dimensions from User Request
dimensions = {
    "okvir_zarjavel.png": (800, 250),
    "srce_postapo.png": (128, 128),
    "merilec_zdravja.png": (192, 100),
    "gumb_start.png": (300, 100),
    "amnezija_maska.png": (1920, 1080)
}

output_dir = "/Users/davidkotnik/repos/novafarma/assets/slike/NOVE_SLIKE/UI/UI/"
os.makedirs(output_dir, exist_ok=True)

for filename, source_path in source_map.items():
    if not os.path.exists(source_path):
        if filename in source_map_fallback and os.path.exists(source_map_fallback[filename]):
            print(f"Fallback: using v2 for {filename}")
            source_path = source_map_fallback[filename]
        else:
            print(f"Source missing for {filename}: {source_path}")
            continue

    try:
        with Image.open(source_path) as img:
            img = img.convert("RGBA")
            target_size = dimensions[filename]
            resized_img = img.resize(target_size, Image.Resampling.LANCZOS)
            
            dest_path = os.path.join(output_dir, filename)
            resized_img.save(dest_path, "PNG")
            print(f"Processed: {filename} -> {target_size}")
            
    except Exception as e:
        print(f"Error processing {filename}: {e}")

# Remove old/garbage files if they define conflicts or are broken
files_to_remove = ["gumb_recikliran.png", "srce_health.png"] # Renamed/Replaced
for f in files_to_remove:
    path = os.path.join(output_dir, f)
    if os.path.exists(path):
        os.remove(path)
        print(f"Removed old file: {f}")