diff --git a/pageindex/config.yaml b/pageindex/config.yaml
index 6b57144ea..be9420cb6 100644
--- a/pageindex/config.yaml
+++ b/pageindex/config.yaml
@@ -4,4 +4,5 @@ max_page_num_each_node: 10
 max_token_num_each_node: 20000
 if_add_node_id: "yes"
 if_add_node_summary: "no"
-if_add_doc_description: "yes"
\ No newline at end of file
+if_add_doc_description: "yes"
+if_add_node_text: "no"
diff --git a/pageindex/page_index.py b/pageindex/page_index.py
index e3e633058..2bbd4095b 100644
--- a/pageindex/page_index.py
+++ b/pageindex/page_index.py
@@ -7,11 +7,12 @@
 from .utils import *
 import os
 from concurrent.futures import ThreadPoolExecutor, as_completed
-import argparse
+
+
 
 
 ################### check title in page #########################################################
-def check_title_appearance(item, page_list, start_index=1, model=None):    
+async def check_title_appearance(item, page_list, start_index=1, model=None):    
     title=item['title']
     if 'physical_index' not in item or item['physical_index'] is None:
         return {'list_index': item.get('list_index'), 'answer': 'no', 'title':title, 'page_number': None}
@@ -37,7 +38,7 @@ def check_title_appearance(item, page_list, start_index=1, model=None):
     }}
     Directly return the final JSON structure. Do not output anything else."""
 
-    response = ChatGPT_API(model=model, prompt=prompt)
+    response = await ChatGPT_API_async(model=model, prompt=prompt)
     response = extract_json(response)
     if 'answer' in response:
         answer = response['answer']
@@ -46,9 +47,9 @@ def check_title_appearance(item, page_list, start_index=1, model=None):
     return {'list_index': item['list_index'], 'answer': answer, 'title': title, 'page_number': page_number}
 
 
-def check_title_appearance_in_start(title, page_text, model=None, logger=None):    
+async def check_title_appearance_in_start(title, page_text, model=None, logger=None):    
     prompt = f"""
-    You will be given given the current section title and the current page_text.
+    You will be given the current section title and the current page_text.
     Your job is to check if the current section starts in the beginning of the given page_text.
     If there are other contents before the current section title, then the current section does not start in the beginning of the given page_text.
     If the current section title is the first content in the given page_text, then the current section starts in the beginning of the given page_text.
@@ -65,36 +66,40 @@ def check_title_appearance_in_start(title, page_text, model=None, logger=None):
     }}
     Directly return the final JSON structure. Do not output anything else."""
 
-    response = ChatGPT_API(model=model, prompt=prompt)
+    response = await ChatGPT_API_async(model=model, prompt=prompt)
     response = extract_json(response)
     if logger:
         logger.info(f"Response: {response}")
-    if 'start_begin' in response:
-       return response['start_begin']
-    else:
-       return  'no'
+    return response.get("start_begin", "no")
 
 
-def check_title_appearance_in_start_parallel(structure, page_list, model=None, logger=None):
+async def check_title_appearance_in_start_concurrent(structure, page_list, model=None, logger=None):
     if logger:
-        logger.info(f"Checking title appearance in start parallel")
-    with ThreadPoolExecutor(max_workers=10) as executor:
-        future_to_item = {
-            executor.submit(check_title_appearance_in_start, item['title'], page_list[item['physical_index']-1][0], model=model, logger=logger): item 
-            for item in structure
-        }
-        
-        # Process completed futures and attach results to items
-        for future in as_completed(future_to_item):
-            item = future_to_item[future]
-            try:
-                result = future.result()
-                item['appear_start'] = result
-            except Exception as e:
-                if logger:
-                    logger.error(f"Error processing item {item['title']}: {str(e)}")
-                item['appear_start'] = 'no'
+        logger.info("Checking title appearance in start concurrently")
     
+    # skip items without physical_index
+    for item in structure:
+        if item.get('physical_index') is None:
+            item['appear_start'] = 'no'
+
+    # only for items with valid physical_index
+    tasks = []
+    valid_items = []
+    for item in structure:
+        if item.get('physical_index') is not None:
+            page_text = page_list[item['physical_index'] - 1][0]
+            tasks.append(check_title_appearance_in_start(item['title'], page_text, model=model, logger=logger))
+            valid_items.append(item)
+
+    results = await asyncio.gather(*tasks, return_exceptions=True)
+    for item, result in zip(valid_items, results):
+        if isinstance(result, Exception):
+            if logger:
+                logger.error(f"Error checking start for {item['title']}: {result}")
+            item['appear_start'] = 'no'
+        else:
+            item['appear_start'] = result
+
     return structure
 
 
@@ -505,14 +510,15 @@ def generate_toc_continue(toc_content, part, model="gpt-4o-2024-11-20"):
     For the title, you need to extract the original title from the text, only fix the space inconsistency.
 
     The provided text contains tags like <physical_index_X> and <physical_index_X> to indicate the start and end of page X. \
-
+    
+    For the physical_index, you need to extract the physical index of the start of the section from the text. Keep the <physical_index_X> format.
 
     The response should be in the following format. 
         [
             {
-                "structure": <structure index, "x.x.x" or None> (string),
+                "structure": <structure index, "x.x.x"> (string),
                 "title": <title of the section, keep the original title>,
-                "physical_index": "<physical_index_X> (keep the format)" or None
+                "physical_index": "<physical_index_X> (keep the format)"
             },
             ...
         ]    
@@ -538,13 +544,15 @@ def generate_toc_init(part, model=None):
 
     The provided text contains tags like <physical_index_X> and <physical_index_X> to indicate the start and end of page X. 
 
+    For the physical_index, you need to extract the physical index of the start of the section from the text. Keep the <physical_index_X> format.
+
     The response should be in the following format. 
         [
-            {
-                "structure": <structure index, "x.x.x" or None> (string),
+            {{
+                "structure": <structure index, "x.x.x"> (string),
                 "title": <title of the section, keep the original title>,
-                "physical_index": "<physical_index_X> (keep the format)" or None
-            },
+                "physical_index": "<physical_index_X> (keep the format)"
+            }},
             
         ],
 
@@ -738,7 +746,7 @@ def single_toc_item_index_fixer(section_title, content, model="gpt-4o-2024-11-20
 
 
 
-def fix_incorrect_toc(toc_with_page_number, page_list, incorrect_results, start_index=1, model=None, logger=None):
+async def fix_incorrect_toc(toc_with_page_number, page_list, incorrect_results, start_index=1, model=None, logger=None):
     print(f'start fix_incorrect_toc with {len(incorrect_results)} incorrect results')
     incorrect_indices = {result['list_index'] for result in incorrect_results}
     
@@ -746,7 +754,7 @@ def fix_incorrect_toc(toc_with_page_number, page_list, incorrect_results, start_
     
     incorrect_results_and_range_logs = []
     # Helper function to process and check a single incorrect item
-    def process_and_check_item(incorrect_item):
+    async def process_and_check_item(incorrect_item):
         list_index = incorrect_item['list_index']
         # Find the previous correct item
         prev_correct = None
@@ -786,7 +794,7 @@ def process_and_check_item(incorrect_item):
         # Check if the result is correct
         check_item = incorrect_item.copy()
         check_item['physical_index'] = physical_index_int
-        check_result = check_title_appearance(check_item, page_list, start_index, model)
+        check_result = await check_title_appearance(check_item, page_list, start_index, model)
 
         return {
             'list_index': list_index,
@@ -794,20 +802,19 @@ def process_and_check_item(incorrect_item):
             'physical_index': physical_index_int,
             'is_valid': check_result['answer'] == 'yes'
         }
-        
-    
-    results = []
-    with ThreadPoolExecutor() as executor:
-        future_to_item = {executor.submit(process_and_check_item, item): item for item in incorrect_results}
-        for future in as_completed(future_to_item):
-            item = future_to_item[future]
-            
-            try:
-                result = future.result()
-                results.append(result)
-            except Exception as exc:
-                print(f"Processing item {item} generated an exception: {exc}")
-    
+
+    # Process incorrect items concurrently
+    tasks = [
+        process_and_check_item(item)
+        for item in incorrect_results
+    ]
+    results = await asyncio.gather(*tasks, return_exceptions=True)
+    for item, result in zip(incorrect_results, results):
+        if isinstance(result, Exception):
+            print(f"Processing item {item} generated an exception: {result}")
+            continue
+    results = [result for result in results if not isinstance(result, Exception)]
+
     # Update the toc_with_page_number with the fixed indices and check for any invalid results
     invalid_results = []
     for result in results:
@@ -827,7 +834,7 @@ def process_and_check_item(incorrect_item):
 
 
 
-def fix_incorrect_toc_with_retries(toc_with_page_number, page_list, incorrect_results, start_index=1, max_attempts=3, model=None, logger=None):
+async def fix_incorrect_toc_with_retries(toc_with_page_number, page_list, incorrect_results, start_index=1, max_attempts=3, model=None, logger=None):
     print('start fix_incorrect_toc')
     fix_attempt = 0
     current_toc = toc_with_page_number
@@ -836,7 +843,7 @@ def fix_incorrect_toc_with_retries(toc_with_page_number, page_list, incorrect_re
     while current_incorrect:
         print(f"Fixing {len(current_incorrect)} incorrect results")
         
-        current_toc, current_incorrect = fix_incorrect_toc(current_toc, page_list, current_incorrect, start_index, model, logger)
+        current_toc, current_incorrect = await fix_incorrect_toc(current_toc, page_list, current_incorrect, start_index, model, logger)
                 
         fix_attempt += 1
         if fix_attempt >= max_attempts:
@@ -849,7 +856,7 @@ def fix_incorrect_toc_with_retries(toc_with_page_number, page_list, incorrect_re
 
 
 ################### verify toc #########################################################
-def verify_toc(page_list, list_result, start_index=1, N=None, model=None):
+async def verify_toc(page_list, list_result, start_index=1, N=None, model=None):
     print('start verify_toc')
     # Find the last non-None physical_index
     last_physical_index = None
@@ -878,17 +885,13 @@ def verify_toc(page_list, list_result, start_index=1, N=None, model=None):
         item_with_index = item.copy()
         item_with_index['list_index'] = idx  # Add the original index in list_result
         indexed_sample_list.append(item_with_index)
-
-    # Run checks in parallel
-    results = []
-    with ThreadPoolExecutor(max_workers=10) as executor:
-        future_to_item = {
-            executor.submit(check_title_appearance, item, page_list, start_index, model): item 
-            for item in indexed_sample_list
-        }
-        
-        for future in as_completed(future_to_item):
-            results.append(future.result())
+    
+    # Run checks concurrently
+    tasks = [
+        check_title_appearance(item, page_list, start_index, model)
+        for item in indexed_sample_list
+    ]
+    results = await asyncio.gather(*tasks)
     
     # Process results
     correct_count = 0
@@ -910,7 +913,7 @@ def verify_toc(page_list, list_result, start_index=1, N=None, model=None):
 
 
 ################### main process #########################################################
-def meta_processor(page_list, mode=None, toc_content=None, toc_page_list=None, start_index=1, opt=None, logger=None):
+async def meta_processor(page_list, mode=None, toc_content=None, toc_page_list=None, start_index=1, opt=None, logger=None):
     print(mode)
     print(f'start_index: {start_index}')
     
@@ -922,7 +925,7 @@ def meta_processor(page_list, mode=None, toc_content=None, toc_page_list=None, s
         toc_with_page_number = process_no_toc(page_list, start_index=start_index, model=opt.model, logger=logger)
             
     toc_with_page_number = [item for item in toc_with_page_number if item.get('physical_index') is not None] 
-    accuracy, incorrect_results = verify_toc(page_list, toc_with_page_number, start_index=start_index, model=opt.model)
+    accuracy, incorrect_results = await verify_toc(page_list, toc_with_page_number, start_index=start_index, model=opt.model)
         
     logger.info({
         'mode': 'process_toc_with_page_numbers',
@@ -932,26 +935,26 @@ def meta_processor(page_list, mode=None, toc_content=None, toc_page_list=None, s
     if accuracy == 1.0 and len(incorrect_results) == 0:
         return toc_with_page_number
     if accuracy > 0.6 and len(incorrect_results) > 0:
-        toc_with_page_number, incorrect_results = fix_incorrect_toc_with_retries(toc_with_page_number, page_list, incorrect_results,start_index=start_index, max_attempts=3, model=opt.model, logger=logger)
+        toc_with_page_number, incorrect_results = await fix_incorrect_toc_with_retries(toc_with_page_number, page_list, incorrect_results,start_index=start_index, max_attempts=3, model=opt.model, logger=logger)
         return toc_with_page_number
     else:
         if mode == 'process_toc_with_page_numbers':
-            return meta_processor(page_list, mode='process_toc_no_page_numbers', toc_content=toc_content, toc_page_list=toc_page_list, start_index=start_index, opt=opt, logger=logger)
+            return await meta_processor(page_list, mode='process_toc_no_page_numbers', toc_content=toc_content, toc_page_list=toc_page_list, start_index=start_index, opt=opt, logger=logger)
         elif mode == 'process_toc_no_page_numbers':
-            return meta_processor(page_list, mode='process_no_toc', start_index=start_index, opt=opt, logger=logger)
+            return await meta_processor(page_list, mode='process_no_toc', start_index=start_index, opt=opt, logger=logger)
         else:
             raise Exception('Processing failed')
         
  
-def process_large_node_recursively(node, page_list, opt=None, logger=None):
-    node_page_list = page_list[node['start_index']-1:node['end_index']-1]
+async def process_large_node_recursively(node, page_list, opt=None, logger=None):
+    node_page_list = page_list[node['start_index']-1:node['end_index']]
     token_num = sum([page[1] for page in node_page_list])
     
     if node['end_index'] - node['start_index'] > opt.max_page_num_each_node and token_num >= opt.max_token_num_each_node:
         print('large node:', node['title'], 'start_index:', node['start_index'], 'end_index:', node['end_index'], 'token_num:', token_num)
 
-        node_toc_tree = meta_processor(node_page_list, mode='process_no_toc', start_index=node['start_index'], opt=opt, logger=logger)
-        node_toc_tree = check_title_appearance_in_start_parallel(node_toc_tree, page_list, model=opt.model, logger=logger)
+        node_toc_tree = await meta_processor(node_page_list, mode='process_no_toc', start_index=node['start_index'], opt=opt, logger=logger)
+        node_toc_tree = await check_title_appearance_in_start_concurrent(node_toc_tree, page_list, model=opt.model, logger=logger)
         
         if node['title'].strip() == node_toc_tree[0]['title'].strip():
             node['nodes'] = post_processing(node_toc_tree[1:], node['end_index'])
@@ -961,37 +964,55 @@ def process_large_node_recursively(node, page_list, opt=None, logger=None):
             node['end_index'] = node_toc_tree[0]['start_index']
         
     if 'nodes' in node and node['nodes']:
-        for child_node in node['nodes']:
+        # for child_node in node['nodes']:
+        #     await process_large_node_recursively(child_node, page_list, opt, logger=logger)
+        tasks = [
             process_large_node_recursively(child_node, page_list, opt, logger=logger)
+            for child_node in node['nodes']
+        ]
+        await asyncio.gather(*tasks)
     
     return node
 
-def tree_parser(page_list, opt, logger=None):    
-    check_toc_result = check_toc(page_list, opt)    
-    logger.info(check_toc_result)
-
-    if check_toc_result['toc_content'] is not None and check_toc_result['page_index_given_in_toc'] == 'yes':
-        toc_with_page_number = meta_processor(
-            page_list, 
-            mode='process_toc_with_page_numbers', 
-            start_index=1, 
-            toc_content=check_toc_result['toc_content'], 
-            toc_page_list=check_toc_result['toc_page_list'], 
-            opt=opt,
-            logger=logger)
+async def tree_parser(page_list, opt, doc=None, logger=None):
+    # Try embedded PDF ToC first
+    embedded_toc = extract_embedded_pdf_toc(doc)
+    if embedded_toc:
+        logger.info(f"Using embedded PDF ToC ({len(embedded_toc)} entries)")
+        toc_with_page_number = add_page_offset_to_toc_json(embedded_toc, offset=0)
+        toc_with_page_number = process_none_page_numbers(toc_with_page_number, page_list, model=opt.model)
     else:
-        toc_with_page_number = meta_processor(
-            page_list, 
-            mode='process_no_toc', 
-            start_index=1, 
-            opt=opt,
-            logger=logger)
+        check_toc_result = check_toc(page_list, opt)
+        logger.info(check_toc_result)
+
+        if check_toc_result.get("toc_content") and check_toc_result["toc_content"].strip() and check_toc_result["page_index_given_in_toc"] == "yes":
+            toc_with_page_number = await meta_processor(
+                page_list, 
+                mode='process_toc_with_page_numbers', 
+                start_index=1, 
+                toc_content=check_toc_result['toc_content'], 
+                toc_page_list=check_toc_result['toc_page_list'], 
+                opt=opt,
+                logger=logger)
+        else:
+            toc_with_page_number = await meta_processor(
+                page_list, 
+                mode='process_no_toc', 
+                start_index=1, 
+                opt=opt,
+                logger=logger)
 
     toc_with_page_number = add_preface_if_needed(toc_with_page_number)
-    toc_with_page_number = check_title_appearance_in_start_parallel(toc_with_page_number, page_list, model=opt.model, logger=logger)
+    toc_with_page_number = await check_title_appearance_in_start_concurrent(toc_with_page_number, page_list, model=opt.model, logger=logger)
     toc_tree = post_processing(toc_with_page_number, len(page_list))
-    for node in toc_tree:
+
+    # for node in toc_tree:
+    #     await process_large_node_recursively(node, page_list, opt, logger=logger)
+    tasks = [
         process_large_node_recursively(node, page_list, opt, logger=logger)
+        for node in toc_tree
+    ]
+    await asyncio.gather(*tasks)
     
     return toc_tree
 
@@ -1012,13 +1033,15 @@ def page_index_main(doc, opt=None):
     logger.info({'total_page_number': len(page_list)})
     logger.info({'total_token': sum([page[1] for page in page_list])})
     
-    structure = tree_parser(page_list, opt, logger=logger)
+    structure = asyncio.run(tree_parser(page_list, opt, doc=doc, logger=logger))
     if opt.if_add_node_id == 'yes':
         write_node_id(structure)    
     if opt.if_add_node_summary == 'yes':
         add_node_text(structure, page_list)
         asyncio.run(generate_summaries_for_structure(structure, model=opt.model))
-        remove_structure_text(structure)  
+        remove_structure_text(structure)
+        if opt.if_add_node_text == 'yes':
+            add_node_text_with_labels(structure, page_list)
         if opt.if_add_doc_description == 'yes':
             doc_description = generate_doc_description(structure, model=opt.model)
             return {
@@ -1033,7 +1056,7 @@ def page_index_main(doc, opt=None):
 
 
 def page_index(doc, model=None, toc_check_page_num=None, max_page_num_each_node=None, max_token_num_each_node=None,
-               if_add_node_id=None, if_add_node_summary=None, if_add_doc_description=None):
+               if_add_node_id=None, if_add_node_summary=None, if_add_doc_description=None, if_add_node_text=None):
     
     user_opt = {
         arg: value for arg, value in locals().items()
diff --git a/pageindex/utils.py b/pageindex/utils.py
index fde5f56fc..2840cf622 100644
--- a/pageindex/utils.py
+++ b/pageindex/utils.py
@@ -14,6 +14,8 @@
 load_dotenv()
 import logging
 import yaml
+import cuid
+from mistralai import Mistral
 from pathlib import Path
 from types import SimpleNamespace as config
 
@@ -304,6 +306,43 @@ def get_pdf_name(pdf_path):
     return pdf_name
 
 
+def extract_embedded_pdf_toc(pdf_path):
+    if not pdf_path:
+        return None
+    try:
+        doc = pymupdf.open(pdf_path)
+        raw_toc = doc.get_toc()  # each entry: [level, title, page]
+    except Exception as e:
+        print(f"Error extracting embedded TOC: {e}")
+        return None
+    
+    if not raw_toc:
+        return None
+
+    fixed_toc = []
+    counters = []  # section numbers
+
+    for entry in raw_toc:
+        level, title, page = entry
+        if level <= 0 or page <= 0 or not title.strip():
+            continue
+
+        while len(counters) < level:
+            counters.append(0)
+
+        counters = counters[:level]
+        counters[-1] += 1
+
+        structure_index = ".".join(str(num) for num in counters)
+        fixed_toc.append({
+            "structure": structure_index,
+            "title": title.strip(),
+            "page": page
+        })
+
+    return fixed_toc
+
+
 class JsonLogger:
     def __init__(self, file_path):
         # Extract PDF name for logger name
@@ -408,26 +447,74 @@ def add_preface_if_needed(data):
 
 
 
-def get_page_tokens(pdf_path, model="gpt-4o-2024-11-20", pdf_parser="PyPDF2"):
+def get_page_tokens(pdf_path, model="gpt-4o-2024-11-20", pdf_parser="Mistral"):
+    enc = tiktoken.encoding_for_model(model)
     if pdf_parser == "PyPDF2":
         pdf_reader = PyPDF2.PdfReader(pdf_path)
-    elif pdf_parser == "PyMuPDF":
-        pdf_reader = pymupdf.open(pdf_path)
-    else:
-        raise ValueError(f"Unsupported PDF parser: {pdf_parser}")
-
-    enc = tiktoken.encoding_for_model(model)
+        page_list = []
+        for page_num in range(len(pdf_reader.pages)):
+            page = pdf_reader.pages[page_num]
+            page_text = page.extract_text()
+            token_length = len(enc.encode(page_text))
+            page_list.append((page_text, token_length))
+        return page_list
     
-    page_list = []
-    for page_num in range(len(pdf_reader.pages)):
-        page = pdf_reader.pages[page_num]
-        page_text = page.extract_text()
-        token_length = len(enc.encode(page_text))
-        page_list.append((page_text, token_length))
+    elif pdf_parser == "PyMuPDF":
+        if isinstance(pdf_path, BytesIO):
+            pdf_stream = pdf_path
+            doc = pymupdf.open(stream=pdf_stream, filetype="pdf")
+        elif isinstance(pdf_path, str) and os.path.isfile(pdf_path) and pdf_path.lower().endswith(".pdf"):
+            doc = pymupdf.open(pdf_path)
+        page_list = []
+        for page in doc:
+            page_text = page.get_text()
+            token_length = len(enc.encode(page_text))
+            page_list.append((page_text, token_length))
+        return page_list
     
-    return page_list
-
+    elif pdf_parser == "Mistral":
+        print("Using Mistral for OCR...")
+        mistral_api_key = os.getenv("MISTRAL_API_KEY")
+        client = Mistral(api_key=mistral_api_key)
+
+        if isinstance(pdf_path, BytesIO):
+            pdf_stream = pdf_path
+            os.makedirs("./pageindex_uploads", exist_ok=True)
+            file_name = get_pdf_name(pdf_stream)
+            file_name = f"{file_name}_{cuid.cuid()}.pdf"
+            save_path = f"./pageindex_uploads/{file_name}"
+            with open(save_path, "wb") as f:
+                f.write(pdf_stream.getvalue())
+            pdf_path = save_path
+        elif isinstance(pdf_path, str) and os.path.isfile(pdf_path) and pdf_path.lower().endswith(".pdf"):
+            file_name = os.path.basename(pdf_path)
+            file_name = f"{file_name}_{cuid.cuid()}.pdf"
+
+        uploaded_pdf = client.files.upload(
+            file={
+                "file_name": file_name,
+                "content": open(pdf_path, "rb"),
+            },
+            purpose="ocr"
+        )
+        signed_url = client.files.get_signed_url(file_id=uploaded_pdf.id)
+        ocr_response = client.ocr.process(
+            model="mistral-ocr-latest",
+            document={
+                "type": "document_url",
+                "document_url": signed_url.url,
+            }
+        )
+        page_list = []
+        for page in ocr_response.pages:
+            page_text = page.markdown
+            token_length = len(enc.encode(page_text))
+            page_list.append((page_text, token_length))
+        print("Finished OCR with Mistral...")
+        return page_list
 
+    else:
+        raise ValueError(f"Unsupported PDF parser: {pdf_parser}")
 
         
 
@@ -437,6 +524,12 @@ def get_text_of_pdf_pages(pdf_pages, start_page, end_page):
         text += pdf_pages[page_num][0]
     return text
 
+def get_text_of_pdf_pages_with_labels(pdf_pages, start_page, end_page):
+    text = ""
+    for page_num in range(start_page-1, end_page):
+        text += f"<physical_index_{page_num+1}>\n{pdf_pages[page_num][0]}\n<physical_index_{page_num+1}>\n"
+    return text
+
 def get_number_of_pages(pdf_path):
     pdf_reader = PyPDF2.PdfReader(pdf_path)
     num = len(pdf_reader.pages)
@@ -534,18 +627,6 @@ def convert_page_to_int(data):
                 pass
     return data
 
-def write_node_id(data, node_id=0):
-    if isinstance(data, dict):
-        data['node_id'] = str(node_id).zfill(4)
-        node_id += 1
-        for key in list(data.keys()):
-            if 'nodes' in key:
-                node_id = write_node_id(data[key], node_id)
-    elif isinstance(data, list):
-        for index in range(len(data)):
-            node_id = write_node_id(data[index], node_id)
-    return node_id
-
 
 def add_node_text(node, pdf_pages):
     if isinstance(node, dict):
@@ -559,6 +640,20 @@ def add_node_text(node, pdf_pages):
             add_node_text(node[index], pdf_pages)
     return
 
+
+def add_node_text_with_labels(node, pdf_pages):
+    if isinstance(node, dict):
+        start_page = node.get('start_index')
+        end_page = node.get('end_index')
+        node['text'] = get_text_of_pdf_pages_with_labels(pdf_pages, start_page, end_page)
+        if 'nodes' in node:
+            add_node_text_with_labels(node['nodes'], pdf_pages)
+    elif isinstance(node, list):
+        for index in range(len(node)):
+            add_node_text_with_labels(node[index], pdf_pages)
+    return
+
+
 async def generate_node_summary(node, model=None):
     prompt = f"""You are given a part of a document, your task is to generate a description of the partial document about what are main points covered in the partial document.
 
diff --git a/requirements.txt b/requirements.txt
index ad43fe1b7..89b401ac8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,5 @@ PyPDF2==3.0.1
 python-dotenv==1.1.0
 tiktoken==0.7.0
 pyyaml==6.0.2
+mistralai==1.6.0
+cuid==0.4
diff --git a/results/2023-annual-report_structure.json b/results/2023-annual-report_structure.json
index 5ff2f36a2..a4b466635 100644
--- a/results/2023-annual-report_structure.json
+++ b/results/2023-annual-report_structure.json
@@ -164,7 +164,7 @@
     {
       "title": "Appendixes",
       "start_index": 107,
-      "end_index": 108,
+      "end_index": 109,
       "node_id": "0025"
     },
     {
@@ -207,7 +207,7 @@
         {
           "title": "Meeting Minutes",
           "start_index": 147,
-          "end_index": 148,
+          "end_index": 149,
           "node_id": "0032"
         }
       ],
@@ -325,13 +325,31 @@
         {
           "title": "Federal Reserve open market transactions, 2023",
           "start_index": 187,
-          "end_index": 188,
+          "end_index": 187,
           "nodes": [
             {
-              "title": "Federal Reserve open market transactions, 2023\u2014continued",
-              "start_index": 188,
+              "title": "Type of security and transaction",
+              "start_index": 187,
               "end_index": 188,
               "node_id": "0051"
+            },
+            {
+              "title": "Federal agency obligations",
+              "start_index": 188,
+              "end_index": 188,
+              "node_id": "0052"
+            },
+            {
+              "title": "Mortgage-backed securities",
+              "start_index": 188,
+              "end_index": 188,
+              "node_id": "0053"
+            },
+            {
+              "title": "Temporary transactions",
+              "start_index": 188,
+              "end_index": 188,
+              "node_id": "0054"
             }
           ],
           "node_id": "0050"
@@ -339,162 +357,90 @@
         {
           "title": "Federal Reserve Bank holdings of U.S. Treasury and federal agency securities, December 31, 2021\u201323",
           "start_index": 189,
-          "end_index": 190,
+          "end_index": 189,
           "nodes": [
             {
-              "title": "Federal Reserve Bank holdings of U.S. Treasury and federal agency securities, December 31, 2021\u201323\u2014continued",
+              "title": "By remaining maturity",
+              "start_index": 189,
+              "end_index": 189,
+              "node_id": "0056"
+            },
+            {
+              "title": "By type",
+              "start_index": 189,
+              "end_index": 190,
+              "node_id": "0057"
+            },
+            {
+              "title": "By issuer",
               "start_index": 190,
               "end_index": 190,
-              "node_id": "0053"
+              "node_id": "0058"
             }
           ],
-          "node_id": "0052"
+          "node_id": "0055"
         },
         {
           "title": "Reserve requirements of depository institutions, December 31, 2023",
           "start_index": 191,
           "end_index": 191,
-          "node_id": "0054"
+          "node_id": "0059"
         },
         {
           "title": "Banking offices and banks affiliated with bank holding companies in the United States, December 31, 2022 and 2023",
           "start_index": 192,
           "end_index": 192,
-          "node_id": "0055"
+          "node_id": "0060"
         },
         {
           "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1984\u20132023 and month-end 2023",
           "start_index": 193,
-          "end_index": 194,
-          "nodes": [
-            {
-              "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1984\u20132023 and month-end 2023\u2014continued",
-              "start_index": 194,
-              "end_index": 194,
-              "node_id": "0057"
-            },
-            {
-              "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1984\u20132023 and month-end 2023\u2014continued",
-              "start_index": 195,
-              "end_index": 196,
-              "node_id": "0058"
-            },
-            {
-              "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1984\u20132023 and month-end 2023\u2014continued",
-              "start_index": 196,
-              "end_index": 196,
-              "node_id": "0059"
-            }
-          ],
-          "node_id": "0056"
+          "end_index": 196,
+          "node_id": "0061"
         },
         {
           "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1918\u20131983",
           "start_index": 197,
-          "end_index": 198,
-          "nodes": [
-            {
-              "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1918\u20131983\u2014continued",
-              "start_index": 198,
-              "end_index": 198,
-              "node_id": "0061"
-            },
-            {
-              "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1918\u20131983\u2014continued",
-              "start_index": 199,
-              "end_index": 200,
-              "node_id": "0062"
-            },
-            {
-              "title": "Reserves of depository institutions, Federal Reserve Bank credit, and related items, year-end 1918\u20131983\u2014continued",
-              "start_index": 200,
-              "end_index": 200,
-              "node_id": "0063"
-            }
-          ],
-          "node_id": "0060"
+          "end_index": 200,
+          "node_id": "0062"
         },
         {
           "title": "Principal assets and liabilities of insured commercial banks, by class of bank, June 30, 2023 and 2022",
           "start_index": 201,
           "end_index": 201,
-          "node_id": "0064"
+          "node_id": "0063"
         },
         {
           "title": "Initial margin requirements under Regulations T, U, and X",
           "start_index": 202,
           "end_index": 203,
-          "node_id": "0065"
+          "node_id": "0064"
         },
         {
           "title": "Statement of condition of the Federal Reserve Banks, by Bank, December 31, 2023 and 2022",
           "start_index": 203,
-          "end_index": 204,
-          "nodes": [
-            {
-              "title": "Statement of condition of the Federal Reserve Banks, by Bank, December 31, 2023 and 2022\u2014continued",
-              "start_index": 204,
-              "end_index": 206,
-              "node_id": "0067"
-            },
-            {
-              "title": "Statement of condition of the Federal Reserve Banks, by Bank, December 31, 2023 and 2022\u2014continued",
-              "start_index": 206,
-              "end_index": 206,
-              "node_id": "0068"
-            },
-            {
-              "title": "Statement of condition of the Federal Reserve Banks, by Bank, December 31, 2023 and 2022\u2014continued",
-              "start_index": 206,
-              "end_index": 207,
-              "node_id": "0069"
-            },
-            {
-              "title": "Statement of condition of the Federal Reserve Banks, by Bank, December 31, 2023 and 2022\u2014continued",
-              "start_index": 207,
-              "end_index": 208,
-              "node_id": "0070"
-            },
-            {
-              "title": "Statement of condition of the Federal Reserve Banks, by Bank, December 31, 2023 and 2022\u2014continued",
-              "start_index": 208,
-              "end_index": 209,
-              "node_id": "0071"
-            }
-          ],
-          "node_id": "0066"
+          "end_index": 209,
+          "node_id": "0065"
         },
         {
           "title": "Statement of condition of the Federal Reserve Banks, December 31, 2023 and 2022",
           "start_index": 209,
           "end_index": 210,
-          "node_id": "0072"
+          "node_id": "0066"
         },
         {
           "title": "Income and expenses of the Federal Reserve Banks, by Bank, 2023",
           "start_index": 210,
-          "end_index": 211,
+          "end_index": 212,
           "nodes": [
-            {
-              "title": "Income and expenses of the Federal Reserve Banks, by Bank, 2023\u2014continued",
-              "start_index": 211,
-              "end_index": 212,
-              "node_id": "0074"
-            },
             {
               "title": "Income and expenses of the Federal Reserve Banks, by Bank, 2023\u2014continued",
               "start_index": 212,
-              "end_index": 213,
-              "node_id": "0075"
-            },
-            {
-              "title": "Income and expenses of the Federal Reserve Banks, by Bank, 2023\u2014continued",
-              "start_index": 213,
               "end_index": 214,
-              "node_id": "0076"
+              "node_id": "0068"
             }
           ],
-          "node_id": "0073"
+          "node_id": "0067"
         },
         {
           "title": "Income and expenses of the Federal Reserve Banks, 1914\u20132023",
@@ -505,40 +451,40 @@
               "title": "Income and expenses of the Federal Reserve Banks, 1914\u20132023\u2014continued",
               "start_index": 215,
               "end_index": 216,
-              "node_id": "0078"
+              "node_id": "0070"
             },
             {
               "title": "Income and expenses of the Federal Reserve Banks, 1914\u20132023\u2014continued",
               "start_index": 216,
               "end_index": 217,
-              "node_id": "0079"
+              "node_id": "0071"
             },
             {
               "title": "Income and expenses of the Federal Reserve Banks, 1914\u20132023\u2014continued",
               "start_index": 217,
               "end_index": 217,
-              "node_id": "0080"
+              "node_id": "0072"
             }
           ],
-          "node_id": "0077"
+          "node_id": "0069"
         },
         {
           "title": "Operations in principal departments of the Federal Reserve Banks, 2020\u201323",
           "start_index": 218,
           "end_index": 218,
-          "node_id": "0081"
+          "node_id": "0073"
         },
         {
           "title": "Number and annual salaries of officers and employees of the Federal Reserve Banks, December 31, 2023",
           "start_index": 219,
-          "end_index": 219,
-          "node_id": "0082"
+          "end_index": 220,
+          "node_id": "0074"
         },
         {
           "title": "Acquisition costs and net book value of the premises of the Federal Reserve Banks and Branches, December 31, 2023",
           "start_index": 220,
           "end_index": 222,
-          "node_id": "0083"
+          "node_id": "0075"
         }
       ],
       "node_id": "0049"