rule summarize_root:
    params:
        max_iter = config.get('max_iterations', 3)
    output:
        [
            "{BASE_DIR}/logs/file_list.txt",
            "{BASE_DIR}/logs/root_summary.txt"
        ]
    shell:
        "python supervisor_coder.py --prompt summarize_root --out_dir {BASE_DIR} --config {CONFIG} --output-files {output} --max_iterations {params.max_iter}"
        
rule insert_root_summary:
    input:
        [
            "{BASE_DIR}/logs/root_summary.txt"
        ]
    output:
        [
            "{BASE_DIR}/logs/create_numpy_modified.txt"
        ]
    shell:
        "python write_prompt.py --prompt create_numpy --insertion root_summary --description 'ROOT file summary' --out_dir {BASE_DIR}"