"""Restore the publicly backed, immutable preference initialization and base."""
import json
import os
from pathlib import Path
ROOT=Path('/marimo/storage/road')

def main():
 from huggingface_hub import snapshot_download
 from run_manifest import digest
 freeze=json.loads((ROOT/'visual_preference_oct2.freeze.json').read_text())
 initial=freeze['initialization'];destination=ROOT/'preference_initial_oct2'
 downloaded=Path(snapshot_download(initial['repository'],revision=initial['revision'],
    allow_patterns=[initial['prefix']+'/*'],token=(ROOT/'.hf_token').read_text().strip()))
 folder=downloaded/initial['prefix']
 assert digest(folder/'adapter_model.safetensors')==initial['weight_sha256']
 if not destination.exists():destination.symlink_to(folder)
 assert destination.resolve()==folder.resolve()
 config=json.loads((folder/'adapter_config.json').read_text())
 assert config['base_model_name_or_path']==freeze['model']
 assert config['modules_to_save']==['visual']
 snapshot_download(freeze['model'],revision=freeze['revision'],ignore_patterns=['*.bin','*.pt','*.msgpack','*.h5'])
 (ROOT/'visual_preference_assets_oct2.json').write_text(json.dumps({
   'initialization':initial,'base_revision':freeze['revision'],'adapter_config':config,
   'verified':True,'competition_data_published':False},indent=2)+'\n')
 print('PREFERENCE_ASSETS_VERIFIED',flush=True)

if __name__=='__main__':main()