update app

streamlit_app.py

@@ -187,46 +187,48 @@ else:
     )
 
     if extract_images_checkbox:
-        pdf = FPDF(unit="cm", format="A4")
-        pdf.set_auto_page_break(0)
-        imagelist = sorted(
-            [
-                i
-                for i in os.listdir(
-                    os.path.join(tmp_dir, filename_without_extension)
-                )
-                if i.endswith("png")
-            ]
-        )
-        for image in imagelist:
-            with PILImage.open(
-                os.path.join(tmp_dir, filename_without_extension, image)
-            ) as img:
-                size = img.size
-                width, height = size
-                if width > height:
-                    pdf.add_page(orientation="L")
-                else:
-                    pdf.add_page(orientation="P")
-                pred_dict = get_image_predictions(img)
-
-                total_image_areas += pred_dict["width"] * pred_dict["height"]
-                total_content_areas += get_content_area(pred_dict)
-                total_redaction_area += get_redaction_area(pred_dict)
-
-                pred_dict["img"].save(
+        with st.spinner('Calculating redaction proportions...'):
+            pdf = FPDF(unit="cm", format="A4")
+            pdf.set_auto_page_break(0)
+            imagelist = sorted(
+                [
+                    i
+                    for i in os.listdir(
+                        os.path.join(tmp_dir, filename_without_extension)
+                    )
+                    if i.endswith("png")
+                ]
+            )
+            for image in imagelist:
+                with PILImage.open(
+                    os.path.join(tmp_dir, filename_without_extension, image)
+                ) as img:
+                    size = img.size
+                    width, height = size
+                    if width > height:
+                        pdf.add_page(orientation="L")
+                    else:
+                        pdf.add_page(orientation="P")
+                    pred_dict = get_image_predictions(img)
+
+                    total_image_areas += pred_dict["width"] * pred_dict["height"]
+                    total_content_areas += get_content_area(pred_dict)
+                    total_redaction_area += get_redaction_area(pred_dict)
+
+                    pred_dict["img"].save(
+                        os.path.join(
+                            tmp_dir, filename_without_extension, f"pred-{image}"
+                        ),
+                    )
+                pdf.image(
                     os.path.join(
                         tmp_dir, filename_without_extension, f"pred-{image}"
                     ),
+                    w=pdf.w,
+                    h=pdf.h,
                 )
-            pdf.image(
-                os.path.join(
-                    tmp_dir, filename_without_extension, f"pred-{image}"
-                ),
-                w=pdf.w,
-                h=pdf.h,
-            )
-        pdf.output(report, "F")
+            pdf.output(report, "F")
+        st.success('Image predictions complete!')
 
     text_output = f"A total of {len(redacted_pages)} pages were redacted. \n\nThe redacted page numbers were: {', '.join(redacted_pages)}. \n\n"