Update app.py

app.py

@@ -55,7 +55,13 @@ 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
+        
 def auto_detect_mode(content):
     if isinstance(content, str):
         return 'w'  # If content is a string, it's text
@@ -581,7 +587,12 @@ async def handle_user_input(websocket,key, process, connected,process_ids):
                             "type": "web-data",
                             "data": str(soup),
                         }
-                        await websocket.send(json.dumps(response_data))
+                        #await websocket.send(json.dumps(response_data))
+                        # Convert the final soup to a string
+                        final_html = str(soup)
+                        
+                        # Send the HTML in chunks
+                        await send_html_chunked(websocket, final_html)
 
                     elif event["command"]["type"]=="create":
                         if event["item"]=="folder":
@@ -972,7 +983,12 @@ 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))
+                # Convert the final soup to a string
+                final_html = str(soup)
+                
+                # Send the HTML in chunks
+                await send_html_chunked(websocket, final_html)
 
             elif event["command"]["type"]=="create":
                 if event["item"]=="folder":