CoolFace
Apppublic

prithvirajput16/email-classifier-api

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes
App README

๐Ÿ“ฌ Email Classification API with PII Masking

This project implements a secure email classification system designed for a support team.

๐Ÿš€ Features

  • โ€”โœ… Detects and masks PII/PCI info (e.g., names, emails, phone, Aadhar, card numbers)
  • โ€”โœ… Classifies emails into categories: Billing Issues, Technical Support, Account Management
  • โ€”โœ… Returns structured output with positions of masked entities
  • โ€”โœ… Deployable API using Flask on Hugging Face Spaces (via Docker)

๐Ÿ”ง API Endpoint

POST /classify

๐Ÿ”ธ Input JSON:
json
{
  "email_body": "Hello, my name is John Doe. My Aadhar is 123456789012. Iโ€™m facing billing issues."
}
{
  "input_email_body": "Hello, my name is John Doe. My Aadhar is 123456789012. Iโ€™m facing billing issues.",
  "list_of_masked_entities": [
    {
      "position": [18, 26],
      "classification": "full_name",
      "entity": "John Doe"
    },
    {
      "position": [41, 53],
      "classification": "aadhar_num",
      "entity": "123456789012"
    }
  ],
  "masked_email": "Hello, my name is [full_name]. My Aadhar is [aadhar_num]. Iโ€™m facing billing issues.",
  "category_of_the_email": "Billing Issues"
}