lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix : fix size image

Browse Source
This commit is contained in:
otxtan@gmail.com 2024-05-28 16:39:03 +07:00
parent 8df4f82c4f
commit 74af6d8853
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/paperless_ocr_custom/parsers.py
View File

@ -12,7 +12,7 @@ from typing import Optional
from django.conf import settings from django.conf import settings
import requests import requests
from PyPDF2 import PdfReader from PyPDF2 import PdfReader, PdfWriter
from reportlab.pdfgen import canvas from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter from reportlab.lib.pagesizes import letter
from PIL import Image from PIL import Image
@ -217,7 +217,7 @@ class RasterisedDocumentParser(DocumentParser):
font_name = 'Arial' font_name = 'Arial'
data = self.ocr_file(input_path) data = self.ocr_file(input_path)
if not data: if not data:
return return
font_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'fonts', 'arial-font/arial.ttf') font_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'fonts', 'arial-font/arial.ttf')
with open(sidecar, "w") as txt_sidecar: with open(sidecar, "w") as txt_sidecar:
txt_sidecar.write(data.get("content","")) txt_sidecar.write(data.get("content",""))
@ -258,9 +258,10 @@ class RasterisedDocumentParser(DocumentParser):
first_page=1, first_page=1,
last_page=input_pdf.getNumPages()+1) last_page=input_pdf.getNumPages()+1)
can = canvas.Canvas(str(output_path), pagesize=letter) can = canvas.Canvas(str(output_path), pagesize=letter)
for page_num, page in enumerate(input_pdf.pages): for page_num, image in enumerate(images):
page_height = input_pdf.pages[page_num].mediabox[3] page_width, page_height = image.size
page_width = input_pdf.pages[page_num].mediabox[2] # page_height = input_pdf.pages[page_num].mediabox[3]
# page_width = input_pdf.pages[page_num].mediabox[2]
# set size new page # set size new page
can.setPageSize((page_width, page_height)) can.setPageSize((page_width, page_height))
byte_image = io.BytesIO() byte_image = io.BytesIO()
@ -274,13 +275,14 @@ class RasterisedDocumentParser(DocumentParser):
pdfmetrics.registerFont(TTFont('Arial', font_path)) pdfmetrics.registerFont(TTFont('Arial', font_path))
width_api_img = data["pages"][page_num]["dimensions"][1] width_api_img = data["pages"][page_num]["dimensions"][1]
height_api_img = data["pages"][page_num]["dimensions"][0] height_api_img = data["pages"][page_num]["dimensions"][0]
# print(f'kich thuoc goc: height{page_height}, width{page_width}, kich thuoc api: height{height_api_img} width{width_api_img}')
rolate_height = height_api_img /page_height rolate_height = height_api_img /page_height
rolate_width = width_api_img /page_width rolate_width = width_api_img /page_width
for block in data["pages"][page_num]["blocks"]: for block in data["pages"][page_num]["blocks"]:
for line in block.get("lines", []): for line in block.get("lines", []):
y1 = (line.get("bbox")[0][1] / float(rolate_height)) y1 = (line.get("bbox")[0][1] / float(rolate_height))
y2 = (line.get("bbox")[1][1] / float(rolate_height)) y2 = (line.get("bbox")[1][1] / float(rolate_height))
font_size = (y2 - y1) * 72 / 96 font_size = math.floor((y2 - y1) * 72 / 96)-2
y_center_coordinates = y2 - (y2 - y1)/2 y_center_coordinates = y2 - (y2 - y1)/2
for word in line.get("words", []): for word in line.get("words", []):
x1 = word["bbox"][0][0] / float(rolate_width) x1 = word["bbox"][0][0] / float(rolate_width)
@ -294,7 +296,7 @@ class RasterisedDocumentParser(DocumentParser):
w = can.stringWidth(value, font_name, font_size) w = can.stringWidth(value, font_name, font_size)
can.setFont('Arial', font_size) can.setFont('Arial', font_size)
can.drawString(x_center_coordinates - w/2, can.drawString(x_center_coordinates - w/2,
int(page_height) - y_center_coordinates - (font_size/3), int(page_height) - y_center_coordinates - (font_size/2),
value) value)
can.drawImage(ImageReader(io.BytesIO(jpg_image)), can.drawImage(ImageReader(io.BytesIO(jpg_image)),
0, 0, 0, 0,