update orientation handling

app.py

@@ -7,6 +7,7 @@ import fitz
 import tempfile
 import os
 from fpdf import FPDF
+from PIL import Image as PILImage
 
 learn = load_learner(
     hf_hub_download("strickvl/redaction-classifier-fastai", "model.pkl")
@@ -50,7 +51,12 @@ def predict(pdf, confidence, generate_file):
             [i for i in os.listdir(tmp_dir) if i.endswith("png")]
         )
         for image in imagelist:
-            pdf.add_page()
+            with PILImage.open(os.path.join(tmp_dir, image)) as img:
+                size = img.size
+                if size[0] > size[1]:
+                    pdf.add_page("L")
+                else:
+                    pdf.add_page("P")
             # pdf.image(os.path.join(tmp_dir, image), w=190, h=280)
             pdf.image(os.path.join(tmp_dir, image))
         pdf.output(report, "F")