Update app.py

app.py

@@ -709,82 +709,21 @@ with demo:
     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
-            )
+        run_presidio_btn = gr.Button("De-identify using Presidio")
 
     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)