update tmp_dir

app.py

@@ -36,8 +36,7 @@ def predict(pdf, confidence, generate_file):
     document = fitz.open(pdf.name)
     results = []
     images = []
-    tmp_dir = tempfile.TemporaryDirectory().name
-    # tmp_dir = tempfile.gettempdir()
+    tmp_dir = tempfile.gettempdir()
     for page_num, page in enumerate(document, start=1):
         image_pixmap = page.get_pixmap()
         image = image_pixmap.tobytes()
@@ -47,11 +46,13 @@ def predict(pdf, confidence, generate_file):
         )
         if probs[0] > (confidence / 100):
             redaction_count = len(images)
-            image_pixmap.save(os.path.join(tmp_dir, f"page-{page_num}.png"))
+            image_pixmap.save(
+                os.path.join(tmp_dir, pdf.name, f"page-{page_num}.png")
+            )
             images.append(
                 [
                     f"Redacted page #{redaction_count + 1} on page {page_num}",
-                    os.path.join(tmp_dir, f"page-{page_num}.png"),
+                    os.path.join(tmp_dir, pdf.name, f"page-{page_num}.png"),
                 ]
             )
 
@@ -60,12 +61,12 @@ def predict(pdf, confidence, generate_file):
         for page in range(len(results))
         if results[page]["redacted"] > (confidence / 100)
     ]
-    report = os.path.join(tmp_dir, "redacted_pages.pdf")
+    report = os.path.join(tmp_dir, pdf.name, "redacted_pages.pdf")
     if generate_file:
         pdf = FPDF()
         pdf.set_auto_page_break(0)
         imagelist = sorted(
-            [i for i in os.listdir(tmp_dir) if i.endswith("png")]
+            [i for i in os.listdir(tmp_dir, pdf.name) if i.endswith("png")]
         )
         for image in imagelist:
             # with PILImage.open(os.path.join(tmp_dir, image)) as img:
@@ -75,7 +76,7 @@ def predict(pdf, confidence, generate_file):
             #     else:
             #         pdf.add_page("P")
             # pdf.image(os.path.join(tmp_dir, image))
-            with PILImage.open(os.path.join(tmp_dir, image)) as img:
+            with PILImage.open(os.path.join(tmp_dir, pdf.name, image)) as img:
                 size = img.size
                 if size[0] > size[1]:
                     pdf.add_page("L")
@@ -93,8 +94,10 @@ def predict(pdf, confidence, generate_file):
                     font_size=16,
                     label_color="#FF59D6",
                 )
-                pred_dict["img"].save(os.path.join(tmp_dir, f"pred-{image}"))
-            pdf.image(os.path.join(tmp_dir, f"pred-{image}"))
+                pred_dict["img"].save(
+                    os.path.join(tmp_dir, pdf.name, f"pred-{image}")
+                )
+            pdf.image(os.path.join(tmp_dir, pdf.name, f"pred-{image}"))
         pdf.output(report, "F")
 
     text_output = f"A total of {len(redacted_pages)} pages were redacted. \n\n The redacted page numbers were: {', '.join(redacted_pages)}."