sudacode/owocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle no text detected for Google Vision and Azure

Browse Source
This commit is contained in:
AuroraWright
2024-02-02 15:44:27 +01:00
parent c2feb52233
commit 42303faa3e
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
owocr/ocr.py
View File

@@ -167,7 +167,8 @@ class GoogleVision:
except: except:
return (False, 'Unknown error!') return (False, 'Unknown error!')
texts = response.text_annotations texts = response.text_annotations
x = (True, texts[0].description) res = texts[0].description if len(texts) > 0 else ''
x = (True, res)
return x return x
def _preprocess(self, img): def _preprocess(self, img):
@@ -379,7 +380,8 @@ class AzureImageAnalysis:
res = '' res = ''
if read_result.read: if read_result.read:
for line in read_result.read.blocks[0].lines: for block in read_result.read.blocks:
for line in block.lines:
res += line.text + ' ' res += line.text + ' '
else: else:
return (False, 'Unknown error!') return (False, 'Unknown error!')