Create app.py

app.py

@@ -0,0 +1,36 @@
+import base64
+import io
+import os
+import requests
+from PIL import Image
+import gradio as gr
+
+BACKEND_URL = os.getenv("BACKEND_URL", "http://backend:8000")  # docker-compose network name
+
+def call_redact(text):
+    payload = {"text": text, "options": {"temperature": 0.0, "max_tokens": 256}}
+    r = requests.post(f"{BACKEND_URL}/redact", json=payload, timeout=120)
+    r.raise_for_status()
+    data = r.json()
+
+    img = None
+    if data.get("plot_png_b64"):
+        img_bytes = base64.b64decode(data["plot_png_b64"])
+        img = Image.open(io.BytesIO(img_bytes))
+
+    return data["normalized"], img, f"{data['latency_ms']} ms"
+
+with gr.Blocks(title="PII Redaction UI") as demo:
+    gr.Markdown("### PII Redaction (FastAPI backend + Gradio frontend)")
+
+    inp = gr.Textbox(lines=4, label="Enter text")
+    out_text = gr.Textbox(label="Normalized redaction")
+    out_plot = gr.Image(label="Top-K probabilities", type="pil")
+    out_lat = gr.Textbox(label="Latency")
+
+    btn = gr.Button("Redact")
+    btn.click(fn=call_redact, inputs=inp, outputs=[out_text, out_plot, out_lat])
+
+if __name__ == "__main__":
+    # run on 0.0.0.0:7860 for docker
+    demo.launch(server_name="0.0.0.0", server_port=7860)