Update controllers/ws_controller_1.py

controllers/ws_controller_1.py

@@ -131,10 +131,10 @@ async def fetch_message_data(session: aiohttp.ClientSession, access_token: str,
         structured_data=structured_attachment_data
     )
 
-async def process_message(message: Message, websocket: WebSocket, chunk_size: int):
-    message_json = message.to_json()
-    await send_message_in_chunks(websocket, message_json, chunk_size)
-    await websocket.send_text("NEXT_MESSAGE")
+# async def process_message(message: Message, websocket: WebSocket, chunk_size: int):
+#     message_json = message.to_json()
+#     await send_message_in_chunks(websocket, message_json, chunk_size)
+#     await websocket.send_text("NEXT_MESSAGE")
 
 async def get_messages(code: str, websocket: WebSocket,start, brand_name: Optional[str] = None):
     access_token = code
@@ -165,18 +165,19 @@ async def get_messages(code: str, websocket: WebSocket,start, brand_name: Option
             
             for future in asyncio.as_completed(processing_tasks):
                 processed_message = await future
+                # await websocket.send_text("MESSAGE_START")
                 await websocket.send_text(json.dumps(
-                    # "status": "processing",
-                    processed_message.to_json()
-                    # "processed_count": processed_count + 1
+                   
+                   processed_message.to_json(),
+                    
                 ))
+                await websocket.send_text("NEXT_MESSAGE")
+                
                 processed_count += 1
 
             if "nextPageToken" in gmail_data:
                 page_token = gmail_data["nextPageToken"]
-                # await websocket.send_text(json.dumps({"status": "next_page"}))
             else:
-                # await websocket.send_text(json.dumps({"status": "completed", "total_processed": processed_count}))
                 break
 
 async def send_message_in_chunks(websocket: WebSocket, message_json: dict, chunk_size: int):
@@ -252,4 +253,4 @@ async def process_message(session: aiohttp.ClientSession, access_token: str, mes
         company=company_from_mail,
         high_level_company_type=high_level_company_type,
         structured_data=structured_data
-    )
+    )