Update app.py

app.py

@@ -23,6 +23,16 @@ if os.path.isdir(cache_dir):
 else:
     gr.Warning("Cache directory creation error")
 
+policy_help_string = """Policies are defined as follows:
+
+1. **Annotate** - replace the PII instance by a \<TYPE:VALUE\> string, i.e. include both the PII type and its value
+2. **Redact** - all PII instances are replaced by a \<PII\> generic string
+3. **Placeholder** - replace with a prototypical value
+4. **Synthetic** - substitute with synthetic data
+
+For more information on the transformation policies, please refer to the guide [here](https://github.com/piisa/pii-transform/blob/main/doc/policies.md#pii-transformation-policies)"""
+
+
 def change_language(language_selection):
     global language_code
     language_code = language_choices[language_selection]
@@ -32,13 +42,13 @@ def change_language(language_selection):
 def process(text, policy):
     # Create the object, defining the language to use and the policy
     # Further customization is possible by providing a config
+    policy = policy.lower()
     if text == "":
         print("Empty text field")
         gr.Warning("No text present")
         return ""
 
     # Custom config to prevent loading of the Presidio plugin
-    # config = {FMT_CONFIG_PLUGIN: {"piisa-detectors-presidio": {"load": False}}}
     proc = PiiTextProcessor(
         lang=language_code, default_policy=policy, config="config.json"
     )
@@ -54,11 +64,22 @@ def get_full_example(idx):
 
 with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column(scale=0, min_width=75):
-            logo = gr.Image("image.jpeg", show_label=False, show_download_button=False)
+        with gr.Column(scale=0, min_width=100):
+            logo = gr.Image(
+                "image.jpeg",
+                height=100,
+                width=100,
+                show_label=False,
+                show_download_button=False,
+                show_share_button=False,
+                mask_opacity=1.0,
+            )
         with gr.Column():
             pass
-        with gr.Column(scale=0, min_width=200):
+        with gr.Column(
+            scale=0,
+            min_width=200,
+        ):
             lang_picker = gr.Dropdown(
                 choices=list(language_choices.keys()),
                 label="Select Language",
@@ -75,35 +96,37 @@ with gr.Blocks() as demo:
             )
         with gr.Column(scale=0, min_width=25):
             pass
-        with gr.Column(scale=0, min_width=100):
-            for i in range(3):
-                with gr.Row():
-                    pass
+        with gr.Column(scale=0, min_width=150):
+            gr.Markdown(value=""" <p style="text-align: center;">Select Policy</p>""")
+            annotate_btn = gr.Button(value="Annotate", variant="primary", size="sm")
             redact_btn = gr.Button(value="Redact", variant="primary", size="sm")
-            anonymize_btn = gr.Button(value="Anonymize", variant="primary", size="sm")
+            anonymize_btn = gr.Button(value="Synthetic", variant="primary", size="sm")
             placeholder_btn = gr.Button(
                 value="Placeholder", variant="primary", size="sm"
             )
+
         with gr.Column(scale=0, min_width=25):
             pass
         with gr.Column(
             scale=2,
             min_width=400,
         ):
-            text_redacted = gr.TextArea(
+            text_modified = gr.TextArea(
                 label="Transformed Text",
                 lines=10,
                 show_copy_button=True,
                 interactive=False,
             )
-
+        annotate_btn.click(
+            fn=process, inputs=[text_original, annotate_btn], outputs=text_modified
+        )
         redact_btn.click(
             fn=process,
             inputs=[
                 text_original,
                 gr.Text(value="redact", visible=False),
             ],
-            outputs=text_redacted,
+            outputs=text_modified,
         )
         anonymize_btn.click(
             fn=process,
@@ -111,22 +134,24 @@ with gr.Blocks() as demo:
                 text_original,
                 gr.Text(value="synthetic", visible=False),
             ],
-            outputs=text_redacted,
+            outputs=text_modified,
         )
         placeholder_btn.click(
             fn=process,
             inputs=[
                 text_original,
-                gr.Text(value="label", visible=False),
+                gr.Text(value="placeholder", visible=False),
             ],
-            outputs=text_redacted,
+            outputs=text_modified,
         )
     with gr.Row():
         example_selector = gr.Dropdown(
             examples_truncated, type="index", label="Examples"
         )
         example_selector.select(
-            get_full_example, inputs=example_selector, outputs=text_original
+            get_full_example, inputs=example_selector, outputs=[text_original]
         )
-
+    with gr.Accordion(label="Help Panel", open=False):
+        gr.Markdown(value=policy_help_string)
+        
 demo.queue().launch()