Update app.py

app.py

@@ -5,7 +5,6 @@ from presidio_anonymizer import AnonymizerEngine
 from presidio_image_redactor import ImageRedactorEngine
 import numpy as np
 import re
-import os
 from docx import Document
 from PIL import Image
 import pytesseract
@@ -262,7 +261,7 @@ def agentic_compliance(doc, regime):
     exec_summary = executive_summary_template(relevant, score, regime)
 
     findings_md = "\n".join([
-        f"- **{f['entity']}** ({f['text']}), score: {f.get('score', 0):.2f}"
+        f"- **{f['entity']}** (`{f['text']}`), score: {f.get('score', 0):.2f}"
         for f in relevant
     ]) if relevant else "No relevant PII found for this regime."
 
@@ -278,10 +277,10 @@ def agentic_compliance(doc, regime):
         fname = doc.name.lower()
         if fname.endswith((".png", ".jpg", ".jpeg")):
             redacted_file_path = redact_image_with_presidio(doc.name)
-            redacted_image = redacted_file_path  # Image path for inline preview
+            redacted_image = redacted_file_path
         elif fname.endswith(".pdf"):
             redacted_file_path = redact_pdf_with_presidio(doc.name)
-            redacted_image = None  # No inline preview for PDFs
+            redacted_image = None
 
     md = f"""### Compliance Regime: **{regime}**
 **Executive Summary:**  
@@ -295,40 +294,25 @@ def agentic_compliance(doc, regime):
 {summary}
 ---
 {legend_md}
----
-**Redacted Document Preview:**  
-<details>
-<summary>Show/Hide Redacted Text</summary>
-</details>
 """
     return md.strip(), redacted_path, redacted_file_path, redacted_image
 
-def preview_file(file):
-    if not file:
-        return ""
-    if file.name.lower().endswith((".png", ".jpg", ".jpeg")):
-        return f"<img src='file={file.name}' width='500'/>"
-    elif file.name.lower().endswith(".pdf"):
-        return f"""<embed src='file={file.name}' width='100%' height='600px' type='application/pdf' />"""
-    else:
-        return "<i>No preview available for this file type.</i>"
+# ---- Gradio App UI: No previews ----
 
 with gr.Blocks(title="Agentic Compliance MCP Server") as demo:
     gr.Markdown("# Agentic Compliance MCP\nUpload a document to check it for PII then select a compliance regime.")
     with gr.Tab("Compliance Agent"):
         doc = gr.File(label="Upload Document", file_types=SUPPORTED_FILE_TYPES)
-        preview = gr.HTML(label="Original Preview")
         regime = gr.Dropdown(choices=list(COMPLIANCE_ENTITIES.keys()), label="Compliance Regime")
         out = gr.Markdown(label="Compliance Output")
         redacted_out = gr.File(label="Download Redacted Text")
         file_redacted_out = gr.File(label="Download Redacted PDF/Image")
         redacted_img = gr.Image(label="Redacted Image Preview")
 
-        doc.change(preview_file, inputs=doc, outputs=preview) 
-
         gr.Button("Run Compliance Agent").click(
             agentic_compliance,
             inputs=[doc, regime],
             outputs=[out, redacted_out, file_redacted_out, redacted_img]
         )
-demo.launch(mcp_server=True)
+
+demo.launch(mcp_server=True)