31 lines
858 B
Python
31 lines
858 B
Python
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
from helpers.extract_latest_message import extract_latest_message
|
|
from helpers.clean_email_html import clean_email_html
|
|
from helpers.remove_disclaimer import remove_disclaimer
|
|
from helpers.send_classify_request import send_classify_request
|
|
from models import EmailData
|
|
|
|
|
|
class ClassifyRequest(BaseModel):
|
|
email_data: EmailData
|
|
|
|
router = APIRouter()
|
|
@router.post("/classify")
|
|
async def classify_route(request: ClassifyRequest):
|
|
|
|
email = request.email_data
|
|
|
|
clean_email = email.copy()
|
|
|
|
clean_email.subject = email.subject
|
|
|
|
clean_email.body = extract_latest_message(clean_email.body)
|
|
clean_email.body = clean_email_html(clean_email.body)
|
|
clean_email.body = remove_disclaimer(clean_email.body)
|
|
|
|
response = await send_classify_request(clean_email)
|
|
|
|
return response
|
|
|