Detect image orientation if the OCR supports it

Fixes issue #47.
jonaswinkler · Feb 18, 2016 · c34d57a · c34d57a
1 parent 1e7ece8
commit c34d57a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/documents/consumer.py b/src/documents/consumer.py
@@ -28,6 +28,9 @@
 def image_to_string(args):
     self, png, lang = args
     with Image.open(os.path.join(self.SCRATCH, png)) as f:
+        if self.OCR.can_detect_orientation():
+            orientation = self.OCR.detect_orientation(f, lang=lang)
+            f = f.rotate(orientation["angle"], expand=1)
         return self.OCR.image_to_string(f, lang=lang)