# Entry point for FastAPI
from fastapi import FastAPI
from pydantic import BaseModel
from models import load_model, classify_email
from utils import mask_pii_multilingual

app = FastAPI()
tokenizer, model, device = load_model()

class EmailInput(BaseModel):
    input_email_body: str

@app.post("/classify")
async def classify_route(request: EmailInput):
    text = request.input_email_body
    masked_text, entities = mask_pii_multilingual(text)
    category = classify_email(masked_text, tokenizer, model, device)
    return {
        "input_email_body": text,
        "list_of_masked_entities": entities,
        "masked_email": masked_text,
        "category_of_the_email": category
    }