Updates to logging and filename

app.py

@@ -103,6 +103,7 @@ class PiiRegex(object):
         for match in regexes.keys():
             # If we've got a result, add it to matches.
             if getattr(self, match):
+                print (f"PII located in document: {match}")
                 matches.append(match)
 
         return True if matches else False
@@ -127,14 +128,8 @@ class Redactor:
     def redaction(self):
        
         """ main redactor code """ 
-        # opening the pdf
         doc = fitz.open(self.file)       
-        # iterating through pages
         for page in doc:
-            # _wrapContents is needed for fixing
-            # alignment issues with rect boxes in some
-            # cases where there is alignment issue
-            # page._wrapContents()
             sensitive = self.get_sensitive_data(page.get_text("text")
                                                 .split('\n'))
             for data in sensitive:
@@ -142,14 +137,14 @@ class Redactor:
                 # drawing outline over sensitive datas
                 if data:
                     for area in areas:
-                        # annot = page.add_redact_annot(area)
                         annot = page.add_redact_annot(area.quad, text='REDACTED', fontname=None, fontsize=11, fill=(1, 1, 1), text_color=(0, 0, 0), cross_out=True)
                         annot.update()
             # applying the redaction
             page.apply_redactions()
         # saving it to a new pdf
-        doc.save('redacted.pdf')
-        print("Successfully redacted")
+        redacted_file = os.path.splitext(self.file.name)[0] + '_redacted.pdf'
+        doc.save(redacted_file)
+        print(f"Successfully redacted. The redacted file is saved as {redacted_file}")
  
 
 
@@ -159,11 +154,11 @@ def redact_pdf(file):
     redactor = Redactor(file)
     redactor.redaction()
     # Return the redacted pdf file
-    return 'redacted.pdf'
+    return os.path.splitext(file.name)[0] + '_redacted.pdf'
 
 
 inputs = [
-    gr.File(label="Upload PDF")
+    gr.inputs.File(type="file", label="Upload PDF")
 ]
 
 outputs = [