Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -61,11 +61,6 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
         gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token}"})
         # print(gmail_response.text)
         end2 = time.time()
-
-        print("End 2 ")
-        print(end2-start2)
-        print("response length")
-        
         gmail_data = gmail_response.json()
         # print(gmail_data)
         # print(len(gmail_data['messages']))
@@ -94,6 +89,7 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
                         print("time 1")
                         print("sending the message")
                         await websocket.send_text("{"+message+"}")
+                        await websocket.send_text("NEXT_MESSAGE")
                         # await process_message(message, websocket, 100000)
                         
                       
@@ -110,6 +106,7 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
             page_token = gmail_data["nextPageToken"]
         else:
             break
+    await websocket.send_text(f"Total messages processed: {total_processed}")
     print(f"Total messages processed: {total_processed}")
     logging.info(f"Total Processed Messages : {total_processed}")
     print("printing messages")
@@ -120,7 +117,7 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
 
 async def process_message(message:Message, websocket:WebSocket, chunk_size:int):
     logging.info("process_message")
-    print(message)
+    # print(message)
     if message:
         # message_json = message.to_json()
         # logging.info(f"{message_json}")