import json,time,tomllib,hashlib,os
from pathlib import Path
root=Path('/data/realstyle')
while True:
 try:r=json.loads((root/'run-result.json').read_text())
 except (OSError,ValueError):time.sleep(5);continue
 if r['status']=='failed':break
 if r['status']=='complete':
  time.sleep(5)
  r=json.loads((root/'run-result.json').read_text())
  r['batch_size']=tomllib.loads((root/'dataset.toml').read_text())['datasets'][0]['batch_size']
  r['gradient_checkpointing']=False
  r['final_model_sha256']=hashlib.sha256(Path(r['final_model']).read_bytes()).hexdigest()
  (root/'run-result.json').write_text(json.dumps(r,indent=2))
  p=root/'README.txt';s=p.read_text();s=s.replace('"batch_size": 1','"batch_size": '+str(r['batch_size']));p.write_text(s)
  p=root/'status.json';s=json.loads(p.read_text());s.update(phase='complete',step=r['steps'],epoch=r['epochs'],remaining_seconds_estimate=0);p.write_text(json.dumps(s,indent=2))
  (root/'report-finalized').write_text('Batch size verified from dataset configuration; final checksum recorded.\n');os.sync();break
 time.sleep(5)