Update app.py

app.py

@@ -6,6 +6,27 @@ def upload_file(filepath):
     name = Path(filepath).name
     return [gr.UploadButton(visible=False), gr.DownloadButton(label=f"Download {name}", value=filepath, visible=True)]
 
+def scrub_file(filepath):
+    # Temporary list to hold stripped lines
+    stripped_lines = []
+
+    # 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())
+
+    # Overwrite the original file with stripped 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
+
+    # Extract the filename for the download button label
+    name = Path(filepath).name
+
+    # Return the Gradio interface elements
+    return [gr.UploadButton(visible=False), gr.DownloadButton(label=f"Download {name}", value=filepath, visible=True)]
+
+
 def download_file():
     return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]
 
@@ -13,10 +34,11 @@ 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"))
     with gr.Row():
-        u = gr.UploadButton("Upload a file", file_count="single")
+        u = gr.UploadButton("Upload a file", file_count="single", filetypes=["txt", "csv", "tsv"])
         d = gr.DownloadButton("Download the file", visible=False)
 
-    u.upload(upload_file, u, [u, d])
+    #u.upload(upload_file, u, [u, d])
+    u.upload(scrub_file, u, [u, d])
     d.click(download_file, None, [u, d])