Update test.py

test.py

@@ -131,6 +131,13 @@ async def auth_google(request: Request):
 
     return {"attachment_count":attachment_no,"attachment_content":data_new}
 
+
+async def send_chunked_data(websocket: WebSocket, filename: str, data: str):
+    chunk_size = 1024  # Set an appropriate chunk size
+    for i in range(0, len(data), chunk_size):
+        await websocket.send_json({"filename": filename, "data_chunk": data[i:i + chunk_size]})
+        await asyncio.sleep(0.1) 
+
 @app.websocket("/ws")
 async def test_websocket(websocket: WebSocket):
     #This code is basically the authorization code and this authorization code helps us to get the access token with the required scopes that we have set .
@@ -271,10 +278,12 @@ async def test_websocket(websocket: WebSocket):
                                 if data:
                                     data_new[filename]=str(data[:10])
                                     attachment_content = base64.urlsafe_b64decode(data.encode("UTF-8"))
-                                    await websocket.send_json({"filename": filename})
-                                    await websocket.send_bytes(attachment_content)
+                                    # await websocket.send_json({"filename": filename})
+                                    # await websocket.send_bytes(attachment_content)
+                                    await send_chunked_data(websocket, filename, data)
+
                                     attachment_no+=1
-                                    time.sleep(2)
+                                    # time.sleep(2)
                                     # await websocket.send_json({filename:data})
 
                                     # yield f"data: {str(data_new)}\n\n"