Add really basic logging
Signed-off-by: Daniel Henry <iamdanhenry@gmail.com>
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user