Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -51,7 +51,7 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
     message_url = f"https://www.googleapis.com/gmail/v1/users/me/messages/{message_id}"
     message_response = requests.get(message_url, headers={"Authorization": f"Bearer {access_token}"})
     message_data = message_response.json()
-
+    print(message_data)
     subject = extract_subject_from_mail(message_data)
     company_from_mail = extract_domain_name(message_data['payload']['headers'], subject)
 
@@ -141,6 +141,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(len(messages))
     chunk_size = 100000
     for message in messages:
         message_json = message.to_json()