Update app.py

app.py

@@ -51,12 +51,17 @@ def inline_resources(soup, base_url):
 
     return soup
 
-async def send_html_chunks(websocket, soup):
-    for tag in soup.descendants:
-        if isinstance(tag, str):
-            continue
-        await websocket.send(str(tag))
-        await asyncio.sleep(0.1)  # Add a small delay if needed
+async def send_html_chunks(websocket,html):
+    
+    if html is not None:
+        
+        chunk_size = 1024
+        for i in range(0, len(html), chunk_size):
+            chunk = html[i:i + chunk_size]
+            await websocket.send(json.dumps({"type": "chunk", "content": chunk}))
+        await websocket.send(json.dumps({"type": "end", "url": url}))
+    else:
+        await websocket.send(json.dumps({"type": "error", "message": "Failed to fetch URL"}))
 
 # async def send_html_chunks(ws, html_data, chunk_size=1024):