Initial Commit

Signed-off-by: Daniel Henry <iamdanhenry@gmail.com>

app/routers/classify_email.py

@@ -0,0 +1,30 @@
+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
+