#!/usr/bin/env python3
"""Force create and verify fresh checkpoint in ALL locations."""
import torch
import shutil
from pathlib import Path

# Create fresh checkpoint
print("Creating fresh checkpoint...")
import sys
sys.path.insert(0, 'c:/MONIKA')
from salience_os_seed.proto_lm._torch_impl import ProtoLanguageModel, TrainingConfig

m = ProtoLanguageModel(TrainingConfig(checkpoint_path=None))
for text in ["Hello, I am learning.", "The sky is blue.", "I can think."]:
    m.training_step(text)

checkpoint = {
    'model': m.state_dict(),
    'optimizer': m.optimizer.state_dict(),
    'step': m.step,
    'vocab': {'tokens': m.vocab.tokens, 'merges': m.vocab.merges},
    'scheduler': None
}

# Save to multiple locations
paths = [
    Path('c:/MONIKA/storage/proto_lm/checkpoint.pt'),
    Path('c:/Users/shank/AppData/Local/Programs/Windsurf/storage/proto_lm/checkpoint.pt'),
]

for p in paths:
    p.parent.mkdir(parents=True, exist_ok=True)
    torch.save(checkpoint, p)
    print(f"Saved to: {p}")
    
    # Verify
    loaded = torch.load(p, map_location='cpu')
    print(f"  Verified: step={loaded['step']}, vocab={len(loaded['vocab']['tokens'])}")