Irannas/Masked_Email_Classification

1"""Streamlit app to classify emails and mask PII."""
2import streamlit as st
3from core import classify_email_pipeline
4
5st.title("📧 Email Classifier with PII Masking")
6
7email_input = st.text_area("Enter your email content:")
8
9if st.button("Classify"):
10
11    result = classify_email_pipeline(email_input)
12
13    st.subheader("Masked Email")
14    st.code(result["masked_email"])
15
16    st.subheader("Detected Entities")
17    st.write(result["list_of_masked_entities"])
18
19    st.subheader("Predicted Category")
20    st.success(result["category_of_the_email"])
21