Update app.py

app.py

@@ -695,6 +695,87 @@ with demo:
     )
 
     ########################## Presidio ##########################
+    gr.Markdown("<hr />")
+    gr.Markdown("## Step 3: De-identify the document and the prompt")
+    gr.Markdown(
+        """This step will demonstrate de-identification using both FHE and Presidio methods. 
+        The same prompt will be used for both to allow for direct comparison."""
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1, min_width=6):
+            run_fhe_btn = gr.Button("De-identify using FHE")
+        with gr.Column(scale=1, min_width=6):
+            run_presidio_btn = gr.Button("De-identify using Presidio")
+        with gr.Column(scale=1, min_width=6):
+            run_both_btn = gr.Button("Run Both De-identification Methods")
+
+    with gr.Row():
+        with gr.Column(scale=5):
+            anonymized_doc_output = gr.Textbox(
+                label="FHE: Decrypted and de-identified document", lines=10, interactive=True
+            )
+
+        with gr.Column(scale=5):
+            anonymized_query_output = gr.Textbox(
+                label="FHE: Decrypted and de-identified prompt", lines=10, interactive=True
+            )
+
+    with gr.Row():
+        presidio_output = gr.Textbox(
+            label="Presidio: De-identified prompt", lines=10, interactive=True
+        )
+
+    identified_words_output_df = gr.Dataframe(label="FHE: Identified words:", visible=False)
+
+    encrypt_doc_btn.click(
+        fn=encrypt_doc_fn,
+        inputs=[original_sentences_box],
+        outputs=[encrypted_doc_box, anonymized_doc_output],
+    )
+
+    encrypt_query_btn.click(
+        fn=encrypt_query_fn,
+        inputs=[query_box],
+        outputs=[
+            query_box,
+            output_encrypted_box,
+            anonymized_query_output,
+            identified_words_output_df,
+        ],
+    )
+
+    run_fhe_btn.click(
+        anonymization_with_fn,
+        inputs=[original_sentences_box, query_box],
+        outputs=[anonymized_doc_output, anonymized_query_output, identified_words_output_df],
+    )
+
+    run_presidio_btn.click(
+        anonymization_with_presidio,
+        inputs=[query_box],
+        outputs=[presidio_output],
+    )
+
+    def run_both_deidentification(selected_sentences, query):
+        # Run FHE de-identification
+        fhe_results = anonymization_with_fn(selected_sentences, query)
+        
+        # Run Presidio de-identification
+        presidio_result = anonymization_with_presidio(query)
+        
+        # Combine results
+        return {
+            **fhe_results,
+            'presidio_output': gr.update(value=presidio_result)
+        }
+
+    run_both_btn.click(
+        run_both_deidentification,
+        inputs=[original_sentences_box, query_box],
+        outputs=[anonymized_doc_output, anonymized_query_output, identified_words_output_df, presidio_output],
+    )
+
 
 # Launch the app
 demo.launch(share=False)