update app

streamlit_app.py

@@ -1,8 +1,11 @@
-from io import BytesIO
 import os
 import pathlib
 import tempfile
 import time
+from io import BytesIO
+
+import pandas as pd
+import altair as alt
 
 import fitz
 import gradio as gr
@@ -16,7 +19,6 @@ from icevision.all import *
 from icevision.models.checkpoint import *
 from PIL import Image as PILImage
 
-
 CHECKPOINT_PATH = "./allsynthetic-imgsize768.pth"
 
 
@@ -115,7 +117,7 @@ def get_pdf_document(input):
         f.write(uploaded_pdf.getbuffer())
     return fitz.open(
         str(pathlib.Path(filename_without_extension) / "output.pdf")
-        )
+    )
 
 
 def get_image_predictions(img):
@@ -241,5 +243,26 @@ else:
 
         redaction_analysis = f"- {total_redaction_proportion}% of the total area of the redacted pages was redacted. \n- {content_redaction_proportion}% of the actual content of those redacted pages was redacted."
 
+        source = pd.DataFrame(
+            {
+                "category": ["Unredacted", "Redacted"],
+                "value": [
+                    total_redaction_proportion,
+                    100 - total_redaction_proportion,
+                ],
+            }
+        )
+
+        c = (
+            alt.Chart(source)
+            .mark_arc()
+            .encode(
+                theta=alt.Theta(field="value", type="quantitative"),
+                color=alt.Color(field="category", type="nominal"),
+            )
+        )
+
+        st.altair_chart(c, use_container_width=True)
+
         st.text(text_output + redaction_analysis)
         # DISPLAY IMAGES