Update app.py

app.py

@@ -13,9 +13,10 @@ def scrub_file(filepath):
     # Open the original file and strip each line
     with open(filepath, 'r', encoding='utf-8') as file:
         for line in file:
-            stripped_lines.append(line.strip())
+            #stripped_lines.append(line.strip())
+            stripped_lines.append(pii_anonymizer.anonymize_text_including_proper_nouns_and_addresses(line))
 
-    # Overwrite the original file with stripped lines
+    # Overwrite the original file with pii replaced lines
     with open(filepath, 'w', encoding='utf-8') as file:
         for line in stripped_lines:
             file.write(line + '\n')  # Add '\n' to keep the file structure
@@ -31,8 +32,8 @@ def download_file():
     return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]
 
 with gr.Blocks() as demo:
-    gr.Markdown("First upload a file and and then you'll be able download it (but only once!)")
-    gr.Markdown("PII Anonymized:"+pii_anonymizer.anonymize_text_including_proper_nouns_and_addresses("Hello Kripa"))
+    gr.Markdown("First upload a file and and then you'll be able download a PII stripped version of it (but only once!)")
+    #gr.Markdown("PII Anonymized:"+pii_anonymizer.anonymize_text_including_proper_nouns_and_addresses("Hello Kripa"))
     with gr.Row():
         u = gr.UploadButton("Upload a file", file_count="single", file_types=["text"])
         d = gr.DownloadButton("Download the file", visible=False)