#!/usr/bin/env python3
import torch
import sys
sys.path.insert(0, 'c:/MONIKA')
from salience_os_seed.proto_lm._torch_impl import ProtoLanguageModel, TrainingConfig

# Create TRULY fresh model at NEW location
m = ProtoLanguageModel(TrainingConfig(checkpoint_path=None))
for text in ["I am learning.", "The sky is blue.", "Music is pleasant."]:
    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
}

torch.save(checkpoint, 'c:/MONIKA/storage/monika_fresh.pt')
print(f"Fresh checkpoint at NEW location: step={m.step}, vocab={m.vocab.size()}")