Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -48,7 +48,8 @@ def get_messages(code: str,brand_name: Optional[str] = None) -> List[Message]:
             page_token = gmail_data["nextPageToken"]
         else:
             break
-
+    logging.info("printing messages")
+    logging.info(messages)
     return messages
 
 
@@ -184,7 +185,9 @@ def extract_text(html_content: str) -> str:
 async def websocket_main(code: str,  websocket: WebSocket,brand_name: Optional[str] = None):
     access_token = code
     messages = get_messages(access_token,brand_name)
-    logging.info(f"brand_name:{brand_name}")
+    logging.info("websocket_main")
+    logging.info(messages)
+    # logging.info(f"brand_name:{brand_name}")
     await websocket.send_json({"total_messages": len(messages)})
     print("Total Length of messages")
     print(len(messages))
@@ -200,8 +203,7 @@ async def websocket_main(code: str,  websocket: WebSocket,brand_name: Optional[s
 
 async def send_message_in_chunks(websocket: WebSocket, message_json: dict, chunk_size: int):
     message_str = json.dumps(message_json)
-    logging.info("Printing the message_str : ")
-    print(message_str)
+ 
     # await websocket.send_json({"file_len":len(file)})
     for i in range(0, len(message_str), chunk_size):
         await websocket.send_text(message_str[i:i + chunk_size])