Update app.py

app.py

@@ -504,8 +504,9 @@ with demo:
         
     gr.Markdown(
         """
-        <h1 style="text-align: center;">Encrypted de-identification Audio Files Using Fully Homomorphic Encryption</h1>
-        <!-- <p align="center">
+        <h1 style="text-align: center;">Secure De-Identification of Audio Files</h1>
+        <!-- 
+        <p align="center">
             <a href="https://github.com/zama-ai/concrete-ml"> <img style="vertical-align: middle; display:inline-block; margin-right: 3px;" width=15 src="file/images/logos/github.png">Concrete-ML</a>
             —
             <a href="https://docs.zama.ai/concrete-ml"> <img style="vertical-align: middle; display:inline-block; margin-right: 3px;" width=15 src="file/images/logos/documentation.png">Documentation</a>
@@ -513,21 +514,27 @@ with demo:
             <a href=" https://community.zama.ai/c/concrete-ml/8"> <img style="vertical-align: middle; display:inline-block; margin-right: 3px;" width=15 src="file/images/logos/community.png">Community</a>
             —
             <a href="https://twitter.com/zama_fhe"> <img style="vertical-align: middle; display:inline-block; margin-right: 3px;" width=15 src="file/images/logos/x.png">@zama_fhe</a>
-        </p> -->
+        </p> 
+        -->
         """
     )
-
+    
     gr.Markdown(
-    """
-    <p align="center" style="font-size: 16px;">
-        Anonymization makes it impossible to identify a person from a data set and thus allows their privacy to be respected </p> Concealing some of PII information is a helpful technique for de-identification. However, this can't be considered as anonymization. 
-    <p align="center" style="font-size: 16px;">
-        De-identification uses Fully Homomorphic Encryption (FHE) to conceal personally 
-        identifiable information (PII) within encrypted documents, enabling computations to be 
-        performed on the encrypted data.</p>
+        """
+        <p align="center" style="font-size: 18px;">
+            Protecting personal data is more important than ever in today’s digital world. <b>Our project ensures privacy-preserving de-identification of audio data</b> using state-of-the-art <b>Fully Homomorphic Encryption (FHE)</b>, offering a secure and transparent solution for data anonymization.
+        </p>
         
-    """
-    )    
+        <p align="center" style="font-size: 18px;">
+            Traditional methods of de-identification often fall short of true anonymization, merely concealing identifiable information. With FHE, we go beyond obfuscation to provide <b>complete security,</b> allowing computations to be performed directly on encrypted data without ever exposing sensitive details.
+        </p>
+        
+        <p align="center" style="font-size: 18px;">
+            This technology is crucial in enabling organizations to use and share sensitive data responsibly, while fully respecting individual privacy.
+        </p>
+        """
+    )
+
 
     # Step 1: Add an audio file
     gr.Markdown("## Step 1: Add an Audio File")
@@ -613,6 +620,34 @@ with demo:
         outputs=[gen_key_btn],
     )
 
+    ########################## Main document Part ##########################
+
+    gr.Markdown("<hr />")
+    gr.Markdown("## Step 2.1: Select the document you want to encrypt\n\n"
+        """To make it simple, we pre-compiled the following document, but you are free to choose 
+        on which part you want to run this example.
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=5):
+            original_sentences_box = gr.CheckboxGroup(
+                ORIGINAL_DOCUMENT,
+                value=ORIGINAL_DOCUMENT,
+                label="Contract:",
+                show_label=True,
+            )
+
+        with gr.Column(scale=1, min_width=6):
+            gr.HTML("<div style='height: 77px;'></div>")
+            encrypt_doc_btn = gr.Button("Encrypt the document")
+
+        with gr.Column(scale=5):
+            encrypted_doc_box = gr.Textbox(
+                label="Encrypted document:", show_label=True, interactive=False, lines=10
+            )
+
+
     ########################## User Query Part ##########################
 
     gr.Markdown("<hr />")
@@ -656,7 +691,7 @@ with demo:
     ########################## FHE processing Part ##########################
 
     gr.Markdown("<hr />")
-    gr.Markdown("## Step 3: De-identify the prompt using FHE")
+    gr.Markdown("## Step 3: De-identify the document and the prompt using FHE")
     gr.Markdown(
         """Once the client encrypts the document and 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 
@@ -667,11 +702,11 @@ with demo:
     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_doc_output = gr.Textbox(
+                label="Decrypted and de-idenntified document", lines=10, interactive=True
+            )
 
         with gr.Column(scale=5):
 
@@ -682,11 +717,11 @@ with demo:
 
     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_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,
@@ -701,8 +736,8 @@ with demo:
 
     run_fhe_btn.click(
         anonymization_with_fn,
-        inputs=[query_box],
-        outputs=[anonymized_query_output, identified_words_output_df],
+        inputs=[original_sentences_box, query_box],
+        outputs=[anonymized_doc_output, anonymized_query_output, identified_words_output_df],
     )