Update app.py

app.py

@@ -50,12 +50,29 @@ def inline_resources(soup, base_url):
 
     return soup
 
-async def send_html_chunked(websocket, html):
-    chunk_size = 1024  # Define your chunk size here
-    for i in range(0, len(html), chunk_size):
-        chunk = html[i:i + chunk_size]
-        await websocket.send(chunk)
-        await asyncio.sleep(0.1)  # Add a small delay if needed
+# async def send_html_chunked(websocket, html):
+#     chunk_size = 1024  # Define your chunk size here
+#     for i in range(0, len(html), chunk_size):
+#         chunk = html[i:i + chunk_size]
+#         await websocket.send(chunk)
+#         await asyncio.sleep(0.1)  # Add a small delay if needed
+
+async def send_html_chunks(ws, html_data, chunk_size=1024):
+   
+
+    # Split the HTML data into chunks
+    total_chunks = math.ceil(len(html_data) / chunk_size)
+    for i in range(total_chunks):
+        start = i * chunk_size
+        end = min(start + chunk_size, len(html_data))
+        chunk = html_data[start:end]
+
+        # Send the chunk with chunk number and total chunks information
+        chunk_message = f"chunk_{i+1}/{total_chunks}:{chunk}"
+        await ws.send(chunk_message.encode())
+
+        # Wait for a short interval before sending the next chunk
+        await asyncio.sleep(0.1)  # Adjust as needed
         
 def auto_detect_mode(content):
     if isinstance(content, str):
@@ -975,7 +992,8 @@ async def exe(websocket,connected,key,process_ids):
                     "type": "web-data",
                     "data": str(soup),
                 }
-                await websocket.send(json.dumps(response_data))
+                # await websocket.send(json.dumps(response_data))
+                await send_html_chunks(websocket, soup)
                 
 
             elif event["command"]["type"]=="create":