#!/usr/bin/env python3
"""
Copy generated assets from brain folder to proper directories
"""

import shutil
from pathlib import Path

BRAIN = Path("/Users/davidkotnik/.gemini/antigravity/brain/0170fb0d-72f7-4963-a613-14216b9eae14")
BASE = Path("/Users/davidkotnik/repos/novafarma/assets/images")

# Create directories
(BASE / "npcs").mkdir(parents=True, exist_ok=True)
(BASE / "biomes" / "01_lush_forest").mkdir(parents=True, exist_ok=True)
(BASE / "vehicles").mkdir(parents=True, exist_ok=True)
(BASE / "bosses").mkdir(parents=True, exist_ok=True)
(BASE / "tools").mkdir(parents=True, exist_ok=True)

stats = {}

# Copy NPCs
npc_files = list(BRAIN.glob("npc_*.png"))
for f in npc_files:
    shutil.copy2(f, BASE / "npcs" / f.name)
stats['npcs'] = len(npc_files)

# Copy Biome #01
biome_files = list(BRAIN.glob("biome01_*.png"))
for f in biome_files:
    shutil.copy2(f, BASE / "biomes" / "01_lush_forest" / f.name)
stats['biome01'] = len(biome_files)

# Copy Vehicles
vehicle_files = list(BRAIN.glob("vehicle_*.png"))
for f in vehicle_files:
    shutil.copy2(f, BASE / "vehicles" / f.name)
stats['vehicles'] = len(vehicle_files)

# Copy Bosses
boss_files = list(BRAIN.glob("boss_*.png"))
for f in boss_files:
    shutil.copy2(f, BASE / "bosses" / f.name)
stats['bosses'] = len(boss_files)

# Copy Tools
tool_patterns = ["hoe_*.png", "watering_*.png", "axe_*.png"]
tool_count = 0
for pattern in tool_patterns:
    for f in BRAIN.glob(pattern):
        shutil.copy2(f, BASE / "tools" / f.name)
        tool_count += 1
stats['tools'] = tool_count

# Print summary
print("✅ ASSETS COPIED!")
print(f"  NPCs: {stats['npcs']} files")
print(f"  Biome #01: {stats['biome01']} files")
print(f"  Vehicles: {stats['vehicles']} files")
print(f"  Bosses: {stats['bosses']} files")
print(f"  Tools: {stats['tools']} files")
print(f"\n📊 TOTAL: {sum(stats.values())} files copied")