yujunjun/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2025-11-10 14:54:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix pdf parse index out of range (#440)

Browse Source 
### What problem does this PR solve?

fix a bug comes when parse some pdf file #436 

### Type of change

- [☑️ ] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
加帆 2024-04-19 08:44:51 +08:00 committed by GitHub
parent 1dada69daa
commit 39f1feaccb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
deepdoc/parser/pdf_parser.py
View File

@ -830,10 +830,13 @@ class HuParser:
pn = [bx["page_number"]] pn = [bx["page_number"]]
top = bx["top"] - self.page_cum_height[pn[0] - 1] top = bx["top"] - self.page_cum_height[pn[0] - 1]
bott = bx["bottom"] - self.page_cum_height[pn[0] - 1] bott = bx["bottom"] - self.page_cum_height[pn[0] - 1]
if pn[-1] - 1 >= len(self.page_images): return "" page_images_cnt = len(self.page_images)
if pn[-1] - 1 >= page_images_cnt: return ""
while bott * ZM > self.page_images[pn[-1] - 1].size[1]: while bott * ZM > self.page_images[pn[-1] - 1].size[1]:
bott -= self.page_images[pn[-1] - 1].size[1] / ZM bott -= self.page_images[pn[-1] - 1].size[1] / ZM
pn.append(pn[-1] + 1) pn.append(pn[-1] + 1)
if pn[-1] - 1 >= page_images_cnt:
return ""
return "@@{}\t{:.1f}\t{:.1f}\t{:.1f}\t{:.1f}##" \ return "@@{}\t{:.1f}\t{:.1f}\t{:.1f}\t{:.1f}##" \
.format("-".join([str(p) for p in pn]), .format("-".join([str(p) for p in pn]),