split_planning.py 668 B

123456789101112131415161718
  1. # -*- coding: utf-8 -*-
  2. """拆分 planning.json 为每页一个 JSON(visual_qa 需要 --planning-dir 或单页 --planning)"""
  3. import json, os, sys
  4. sys.stdout.reconfigure(encoding='utf-8')
  5. d = json.load(open(r'D:\workspace\cfc\ppt-output\planning.json', encoding='utf-8'))
  6. pages = d['ppt_planning']['pages']
  7. out_dir = r'D:\workspace\cfc\ppt-output\planning'
  8. os.makedirs(out_dir, exist_ok=True)
  9. for i, p in enumerate(pages, 1):
  10. fname = os.path.join(out_dir, f'planning{i}.json')
  11. with open(fname, 'w', encoding='utf-8') as f:
  12. json.dump(p, f, ensure_ascii=False, indent=2)
  13. print(f'wrote {fname}')
  14. print(f'OK: {len(pages)} page planning files')