117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
import os.path
|
|
|
|
import pytesseract
|
|
import requests
|
|
import docx
|
|
import xml.etree.ElementTree as ET
|
|
from PyPDF2 import PdfReader
|
|
from swagger_server.utils import cl_utils
|
|
import cv2
|
|
import numpy as np
|
|
import magic
|
|
import re
|
|
tika_server = "http://tika2:9999/tika"
|
|
|
|
# endpoint below to be used only for development purposes (don't need to run docker)
|
|
# tika_server = "http://rsdo.lhrs.feri.um.si:9998/tika"
|
|
|
|
|
|
def extract_text_prepResp(file, content_type=""):
|
|
content_type = file.content_type
|
|
if content_type is None:
|
|
content_type = magic.from_file(file.stream.name, mime=True)
|
|
|
|
content = ""
|
|
if tika_responding():
|
|
try:
|
|
response = requests.put(tika_server, data=file, headers={"Accept": "text/plain; charset=UTF-8"})
|
|
content = response.text
|
|
#preveri če je pretvorba uspešna
|
|
|
|
# original string
|
|
res = re.findall(r'\w+', content)
|
|
|
|
#preveri, če imamo vsaj 10 besed in če je povprečna dolžina >3 in < 12
|
|
#če to drži, idi v ocr
|
|
reslen=map(lambda n:len(n),res)
|
|
print(f"Število besed je {len(res)}")
|
|
|
|
if len(res)>0 :
|
|
avglen=sum(reslen)/len(res)
|
|
else:
|
|
avglen=0
|
|
|
|
print(f"Povprečna dolžina besede je {avglen}")
|
|
|
|
if(len(res)<10 or avglen<4 or avglen>11):
|
|
print("Besedilo je sumljivo, gremo v OCR in damo file na začetek!")
|
|
file.seek(0)
|
|
response = requests.put(tika_server, data=file, headers={"X-Tika-PDFOcrStrategy": "ocr_only", "X-Tika-OCRLanguage": "slv+eng",
|
|
"Accept": "text/plain; charset=UTF-8"})
|
|
content = response.text
|
|
|
|
#odstranim še vse prelome vrstic, ker imamo s tem probleme
|
|
content=' '.join(content.splitlines())
|
|
except:
|
|
content = "ERROR - something went wrong when reading file with tika"
|
|
|
|
#if content == "":
|
|
# if "openxmlformats-officedocument.wordprocessingml.document" in content_type:
|
|
# content = '\n'.join([p.text for p in docx.Document(file).paragraphs])
|
|
# elif "application/pdf" in content_type:
|
|
# reader = PdfReader(file)
|
|
# content = '\n'.join([p.extract_text() for p in reader.pages])
|
|
# content = content
|
|
# elif "text/xml" in content_type:
|
|
# root = ET.parse(file).getroot()
|
|
# plainText = root.findall('PlainText')
|
|
# if len(plainText) == 0:
|
|
# return "Didn't find anything in PlainText", 400
|
|
# content = '\n'.join([pt.text for pt in plainText])
|
|
# # elif "text/plain" in file.content_type:
|
|
# else:
|
|
# try:
|
|
# content = file.read().decode('utf-8')
|
|
# except:
|
|
# content = "ERROR - something went wrong when reading file with not-tika method!"
|
|
|
|
return content, 200
|
|
|
|
|
|
def ocr_text_prepResp(file):
|
|
content = ""
|
|
if tika_responding():
|
|
try:
|
|
response = requests.put(tika_server, data=file,
|
|
headers={"X-Tika-PDFOcrStrategy": "ocr_only", "X-Tika-OCRLanguage": "slv+eng",
|
|
"Accept": "text/plain; charset=UTF-8"})
|
|
content = response.text
|
|
except:
|
|
content = "ERROR - something went wrong when reading file with tika (OCR)"
|
|
|
|
if content == "":
|
|
try:
|
|
win_p = "C:/Program Files/Tesseract-OCR/tesseract.exe"
|
|
if os.path.exists(win_p):
|
|
pytesseract.pytesseract.tesseract_cmd = win_p
|
|
|
|
# convert string data to numpy array
|
|
file_bytes = np.fromstring(file.read(), np.uint8)
|
|
# convert numpy array to image
|
|
img = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
|
|
|
|
conf = '-l eng+slv'
|
|
content = pytesseract.image_to_string(img, config=conf)
|
|
except:
|
|
content = "ERROR - something went wrong when reading file with not-tika method! (OCR)"
|
|
|
|
return content, 200
|
|
|
|
|
|
def tika_responding():
|
|
try:
|
|
ret = requests.get(tika_server)
|
|
return ret.status_code == 200
|
|
except:
|
|
return False
|