aopstudio commited on
Commit
b4b278b
·
1 Parent(s): 0cb3254

Fix occasional errors in pdf table recognition (#1277)

Browse files

### What problem does this PR solve?

Fix occasional errors in pdf table recognition

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. deepdoc/parser/pdf_parser.py +1 -1
deepdoc/parser/pdf_parser.py CHANGED
@@ -952,7 +952,7 @@ class RAGFlowPdfParser:
952
  fnm, str) else pdfplumber.open(BytesIO(fnm))
953
  self.page_images = [p.to_image(resolution=72 * zoomin).annotated for i, p in
954
  enumerate(self.pdf.pages[page_from:page_to])]
955
- self.page_chars = [[c for c in page.chars if self._has_color(c)] for page in
956
  self.pdf.pages[page_from:page_to]]
957
  self.total_page = len(self.pdf.pages)
958
  except Exception as e:
 
952
  fnm, str) else pdfplumber.open(BytesIO(fnm))
953
  self.page_images = [p.to_image(resolution=72 * zoomin).annotated for i, p in
954
  enumerate(self.pdf.pages[page_from:page_to])]
955
+ self.page_chars = [[{**c, 'top': max(0, c['top'] - 10), 'bottom': max(0, c['bottom'] - 10)} for c in page.chars if self._has_color(c)] for page in
956
  self.pdf.pages[page_from:page_to]]
957
  self.total_page = len(self.pdf.pages)
958
  except Exception as e: