Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -58,10 +58,15 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
     body = extract_body_from_mail(message_data)
 
     attachments = extract_attachments_from_mail(access_token, message_data)
+    print("subject: ")
     print(subject)
+    print("company name: ")
     print(company_from_mail)
+    print("Printing the body of the mail: ")
     print(body)
+    print("Printing attachment Data: ")
     print(attachments)
+    print("Completed this mail.")
     return Message(message_id=message_id, body=body, attachments=attachments, company=company_from_mail)
 
 
@@ -141,6 +146,7 @@ async def websocket_main(code: str, websocket: WebSocket):
     access_token = code
     messages = get_messages(access_token)
     await websocket.send_json({"total_messages": len(messages)})
+    print("Total Length of messages")
     print(len(messages))
     chunk_size = 100000
     for message in messages:
@@ -148,6 +154,7 @@ async def websocket_main(code: str, websocket: WebSocket):
         await send_message_in_chunks(websocket, message_json, chunk_size)
         await websocket.send_text("NEXT_MESSAGE")
 
+    
     await websocket.close()