yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-09-09 00:28:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for anchor generation on pdfs with no text elements

Browse Source
This commit is contained in:
Jake Poznanski 2024-10-11 15:01:01 +00:00
parent af03358c47
commit aea3f7f1fe
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pdelfin/prompts/anchor.py
View File

@ -302,11 +302,12 @@ def _linearize_pdf_report(report: PageReport, max_length: int = 4000) -> str:
if report.text_elements:
text_elements = [e for e in report.text_elements if len(e.text.strip()) > 0]
min_x_text = min(text_elements, key=lambda e: e.x)
max_x_text = max(text_elements, key=lambda e: e.x)
min_y_text = min(text_elements, key=lambda e: e.y)
max_y_text = max(text_elements, key=lambda e: e.y)
edge_elements.update([min_x_text, max_x_text, min_y_text, max_y_text])
if text_elements:
min_x_text = min(text_elements, key=lambda e: e.x)
max_x_text = max(text_elements, key=lambda e: e.x)
min_y_text = min(text_elements, key=lambda e: e.y)
max_y_text = max(text_elements, key=lambda e: e.y)
edge_elements.update([min_x_text, max_x_text, min_y_text, max_y_text])
# Keep track of element IDs to prevent duplication
selected_element_ids = set()