document id

app.py

@@ -12,19 +12,30 @@ from spacy.tokens import Span
 os.system('python -m spacy download en_core_web_sm')
 
 # # set up colors for PII types ---
+# options = {
+#     "colors": {
+#         "NAME_STUDENT": "#7FDBFF",  # Soft blue
+#         "EMAIL": "#008080",         # Dark cyan
+#         "USERNAME": "#C3B1E1",      # Pastel violet
+#         "ID_NUM": "#2ECC40",        # Medium green
+#         "PHONE_NUM": "#FF851B",     # Deep orange
+#         "URL_PERSONAL": "#4682B4",  # Steel blue
+#         "STREET_ADDRESS": "#808000",  # Muted olive
+#     }
+# }
+
 options = {
     "colors": {
-        "NAME_STUDENT": "#7FDBFF",  # Soft blue
-        "EMAIL": "#008080",         # Dark cyan
-        "USERNAME": "#C3B1E1",      # Pastel violet
-        "ID_NUM": "#2ECC40",        # Medium green
-        "PHONE_NUM": "#FF851B",     # Deep orange
-        "URL_PERSONAL": "#4682B4",  # Steel blue
-        "STREET_ADDRESS": "#808000",  # Muted olive
+        "NAME_STUDENT": "#6EB5FF",    # Lighter blue
+        "EMAIL": "#42D4B5",           # Light teal
+        "USERNAME": "#D8B4E2",        # Light lavender
+        "ID_NUM": "#7AE88F",          # Light green
+        "PHONE_NUM": "#FFB87D",       # Light peach
+        "URL_PERSONAL": "#81C3FF",    # Light sky blue
+        "STREET_ADDRESS": "#B4B77F"   # Light olive
     }
 }
 
-
 # download datamix ---
 
 
@@ -87,7 +98,7 @@ def render_sample(subset, pii_type):
 
     doc.ents = ents
     output = displacy.render(doc, style="ent", jupyter=False, options=options)
-    return output
+    return {'document': sample['document']}, output
 
 
 # app layout & callback ---
@@ -109,6 +120,7 @@ with gr.Blocks() as demo:
         )
 
     sample_btn = gr.Button("Sample")
+    document_id_display = gr.JSON(label="Document ID")
 
     sample_display = gr.HTML(label="Example")
 
@@ -116,7 +128,7 @@ with gr.Blocks() as demo:
     sample_btn.click(
         fn=render_sample,
         inputs=[subset_dropdown, focus_pii],
-        outputs=sample_display,
+        outputs=[document_id_display, sample_display],
     )
 
 # launch app ---