fix issue for orientation

app.py

@@ -4,7 +4,7 @@ import subprocess
 import streamlit as st
 import io 
 import pypdfium2 
-from PIL import Image
+from PIL import Image, ExifTags
 import logging
 
 # 设置日志记录器
@@ -63,6 +63,41 @@ def resize_image_if_needed(pil_image, max_size_mb=1, max_edge_length=1024):
     return pil_image
 
 
+def correct_image_orientation(pil_image):
+    """
+    自动检测PIL Image对象是否包含EXIF信息，如果包含则根据EXIF信息重新修改图片的朝向。
+    
+    :param pil_image: 输入的PIL Image对象
+    :return: 返回修正后的PIL Image对象
+    """
+    try:
+        # 获取EXIF信息
+        exif = pil_image._getexif()
+        if exif is not None:
+            # 查找Orientation的EXIF标签编号
+            for orientation in ExifTags.TAGS.keys():
+                if ExifTags.TAGS[orientation] == 'Orientation':
+                    break
+
+            # 获取图片的朝向信息
+            orientation_value = exif.get(orientation)
+            print(f"Orientation value: {orientation_value}")
+
+            # 根据朝向信息调整图片方向
+            if orientation_value == 3:
+                pil_image = pil_image.rotate(180, expand=True)
+            elif orientation_value == 6:
+                pil_image = pil_image.rotate(270, expand=True)
+            elif orientation_value == 8:
+                pil_image = pil_image.rotate(90, expand=True)
+
+    except (AttributeError, KeyError, IndexError):
+        # 如果没有EXIF信息或者没有朝向信息，跳过处理
+        pass
+
+    return pil_image
+
+
 def clone_repo():
     # 从环境变量中获取 GitHub Token
     github_token = os.getenv('GH_TOKEN')
@@ -153,6 +188,7 @@ if clone_repo():
         pil_image = get_page_image(in_file, page_number)
     else:
         pil_image = Image.open(in_file).convert("RGB")
+        pil_image = correct_image_orientation(pil_image)
     pil_image = resize_image_if_needed(pil_image)
     
     text_rec = st.sidebar.button("認識開始")