Function post_process_levels

mainprogress/determine_toc_levels.py:409–440

(output_path: Path)

                page_data = json.load(f)
            if not page_data: continue
            
            current_max_level = max(item.get("level", 0) for item in page_data)
            if current_max_level < first_page_max_level:
                level_diff = first_page_max_level - current_max_level
                write_log(f"修正文件 {file.name}: 层级提升 {level_diff}")
                for item in page_data:
                    item["level"] += level_diff
                with open(file, 'w', encoding='utf-8') as f:
                    json.dump(page_data, f, ensure_ascii=False, indent=2)
    except Exception as e:
        write_log(f"后处理异常：{str(e)}")

async def run_batch_processing(image_files: list, output_path: Path):
    if not image_files:
        return

    write_log(f"正在预处理并缓存 {len(image_files)} 张图片...")
    for img in image_files:
        get_encoded_image(img)
        
    semaphore = asyncio.Semaphore(CONCURRENT_LIMIT)
    
    # 1. 串行处理第一张图片
    first_img = image_files[0]
    first_csv = output_path / f"{first_img.stem}.csv"
    
    write_log("阶段 1: 处理首图以生成 Few-shot 示例 (CSV 格式)")
    success = await process_first_page(first_img, first_csv, output_path)
    
    if not success:
        write_log("严重错误：首图处理失败，无法生成参考示例，终止后续并发处理。")
        print("首图处理失败，脚本停止。请检查日志。")
        return

    # 2. 并发处理剩余图片