rajendradayma/Email-Classification-and-PII-Masking-System

1# streamlit_app.py2import streamlit as st3from models import load_model4from utils import mask_pii5 6st.set_page_config(page_title="📧 Email Classification System", page_icon="📧")7 8st.title("📧 Email Classification and PII Masking System")9 10# Load model11model = load_model("model/classifier.pkl")12 13# Prediction function14def classify_email_direct(email_body):15    masked_text, entities = mask_pii(email_body)16    category = model.predict([masked_text])[0]17    return {18        "input_email_body": email_body,19        "list_of_masked_entities": entities,20        "masked_email": masked_text,21        "category_of_the_email": category22    }23 24# Streamlit UI25email_input = st.text_area("✉️ Enter your email text:")26 27if st.button("🚀 Classify Email"):28    if email_input.strip() != "":29        result = classify_email_direct(email_input)30 31        st.subheader("🔎 Masked Email")32        st.write(result['masked_email'])33 34        st.subheader("🔐 List of Masked Entities")35        st.json(result['list_of_masked_entities'])36 37        st.subheader("📂 Predicted Category")38        st.success(result['category_of_the_email'])39    else:40        st.warning("⚠️ Please enter some text first.")41 42