daniel/email-classifier
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

Add really basic logging

Browse Source 
Signed-off-by: Daniel Henry <iamdanhenry@gmail.com>
This commit is contained in:
Daniel Henry
2026-01-28 12:58:56 -06:00
parent acd7aca78e
commit 395bc29bae
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/routers/classify_email.py
View File

@@ -6,6 +6,7 @@ from app.helpers.remove_disclaimer import remove_disclaimer
from app.helpers.send_classify_request import send_classify_request from app.helpers.send_classify_request import send_classify_request
from app.models import EmailData from app.models import EmailData
logging = True
class ClassifyRequest(BaseModel): class ClassifyRequest(BaseModel):
email_data: EmailData email_data: EmailData
@@ -14,16 +15,29 @@ router = APIRouter()
@router.post("/classify") @router.post("/classify")
async def classify_route(request: ClassifyRequest): async def classify_route(request: ClassifyRequest):
if logging:
print("Payload Recieved:")
print(request.email_data)
print("\n\n-----")
email = request.email_data email = request.email_data
clean_email = email.copy() clean_email = email.copy()
clean_email.subject = email.subject clean_email.subject = email.subject
clean_email.body = extract_latest_message(clean_email.body) clean_email.body = extract_latest_message(clean_email.body)
clean_email.body = clean_email_html(clean_email.body) clean_email.body = clean_email_html(clean_email.body)
clean_email.body = remove_disclaimer(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) response = await send_classify_request(clean_email)
return response return response