Update app.py

app.py

@@ -686,40 +686,24 @@ with demo:
             )
 
     ########################## FHE processing Part ##########################
-
     gr.Markdown("<hr />")
-    gr.Markdown("## Step 3: De-identify the document and the prompt using FHE")
+    gr.Markdown("## Step 3: De-identify the prompt using FHE")
     gr.Markdown(
-        """Once the client encrypts the document and the prompt locally, it will be sent to a remote 
+        """Once the client encrypts the prompt locally, it will be sent to a remote 
         server to perform the de-identification on encrypted data. When the computation is done, the 
         server will return the result to the client for decryption."""
-        
     )
 
     run_fhe_btn = gr.Button("De-identify using FHE")
 
     with gr.Row():
         with gr.Column(scale=5):
-
-            anonymized_doc_output = gr.Textbox(
-                label="Decrypted and de-idenntified document", lines=10, interactive=True
-            )
-
-        with gr.Column(scale=5):
-
             anonymized_query_output = gr.Textbox(
                 label="Decrypted and de-identified prompt", lines=10, interactive=True
             )
-
-
+    
     identified_words_output_df = gr.Dataframe(label="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],
@@ -733,10 +717,9 @@ with demo:
 
     run_fhe_btn.click(
         anonymization_with_fn,
-        inputs=[original_sentences_box, query_box],
-        outputs=[anonymized_doc_output, anonymized_query_output, identified_words_output_df],
+        inputs=[query_box],
+        outputs=[anonymized_query_output, identified_words_output_df],
     )
-    
 
     ########################## Presidio ##########################
     gr.Markdown("<hr />")