diff --git a/pixelocr/gui/ocrengine.py b/pixelocr/gui/ocrengine.py index 505a6fa..7882627 100644 --- a/pixelocr/gui/ocrengine.py +++ b/pixelocr/gui/ocrengine.py @@ -55,8 +55,7 @@ class OCREngine(QThread): return Page(Image.fromfile(filename), filename) def run(self): - for page_text in self.recognize(): - print(page_text) + self.recognize() if self.quit: qApp.quit() @@ -64,7 +63,10 @@ class OCREngine(QThread): for filename in self.filenames: page = self.load_page(filename) self.pageChanged.emit(page) - yield self.recognize_page(page) + page_text = self.recognize_page(page) + print(page_text) + with open(filename + '.txt', 'w') as page_text_file: + page_text_file.write(page_text) def recognize_page(self, page): glyph_data_seq = itertools.chain(*(self.recognize_line(line) for line in page.lines))