Update app.py

app.py

@@ -51,14 +51,15 @@ def inline_resources(soup, base_url):
 
     return soup
 
-async def send_html_chunks(websocket,html):
-    
+async def send_html_chunks(websocket, html, url):
     if html is not None:
+        soup = BeautifulSoup(html, 'html.parser')
+        tags = soup.find_all(True)
+        
+        for tag in tags:
+            tag_content = str(tag)
+            await websocket.send(json.dumps({"type": "chunk", "content": tag_content}))
         
-        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"}))