import os
import shutil

src_base = "assets/DEMO_FAZA1/Vegetation"
dst_base = "assets/vegetation"

mapping = {
    "trava_sop.png": "trava_zelena.png",
    "visoka_trava.png": "trava_suha.png",
    "grass_cluster_dense.png": "trava_divja.png"
}

for src_name, dst_name in mapping.items():
    src = os.path.join(src_base, src_name)
    dst = os.path.join(dst_base, dst_name)
    
    if os.path.exists(src):
        shutil.copy(src, dst)
        print(f"Copied {src_name} -> {dst_name}")
    else:
        # Fallback: Generate if missing
        print(f"Missing {src_name}, generating placeholder for {dst_name}")
        from PIL import Image, ImageDraw
        img = Image.new('RGBA', (64, 64), (0,0,0,0))
        draw = ImageDraw.Draw(img)
        color = (0, 255, 0)
        if "suha" in dst_name: color = (200, 200, 100)
        if "divja" in dst_name: color = (0, 100, 0)
        draw.ellipse([10, 10, 54, 54], fill=color + (200,))
        img.save(dst)

print("Vegetation variation ready.")