diff --git a/app/routers/classify_email.py b/app/routers/classify_email.py index 81dac45..53e7f5a 100644 --- a/app/routers/classify_email.py +++ b/app/routers/classify_email.py @@ -6,6 +6,7 @@ from app.helpers.remove_disclaimer import remove_disclaimer from app.helpers.send_classify_request import send_classify_request from app.models import EmailData +logging = True class ClassifyRequest(BaseModel): email_data: EmailData @@ -14,16 +15,29 @@ router = APIRouter() @router.post("/classify") async def classify_route(request: ClassifyRequest): + if logging: + print("Payload Recieved:") + print(request.email_data) + print("\n\n-----") + + 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) + if logging: + print("Cleaned Payload:") + print(clean_email) + print("\n\n-----") + response = await send_classify_request(clean_email) return response