Enhanced HTML cleaning to remove iframeResizer scripts and fix console errors

app.py

@@ -70,10 +70,10 @@ def render_preview(code: str, width: int, height: int, scale: float) -> str:
     # Clean up the HTML code to remove problematic script and CSS references
     soup = BeautifulSoup(code, 'html.parser')
     
-    # Remove any script tags that reference local assets
+    # Remove any script tags that reference local assets or iframeResizer
     for script in soup.find_all('script'):
         src = script.get('src', '')
-        if src and ('assets/' in src or 'index-' in src):
+        if src and ('assets/' in src or 'index-' in src or 'iframeResizer' in src):
             script.decompose()
     
     # Remove any link tags that reference local CSS assets
@@ -82,6 +82,11 @@ def render_preview(code: str, width: int, height: int, scale: float) -> str:
         if href and ('assets/' in href or 'index-' in href):
             link.decompose()
     
+    # Also remove any inline scripts that might contain problematic content
+    for script in soup.find_all('script'):
+        if script.string and ('assets/' in script.string or 'index-' in script.string):
+            script.decompose()
+    
     # Get the cleaned HTML
     cleaned_code = str(soup)
     safe_code = html.escape(cleaned_code).replace("'", "'")
@@ -153,7 +158,7 @@ def process_and_generate(image_np, image_path_from_state, sidebar_prompt, header
     # Clean up problematic script and CSS references
     for script in soup.find_all('script'):
         src = script.get('src', '')
-        if src and ('assets/' in src or 'index-' in src):
+        if src and ('assets/' in src or 'index-' in src or 'iframeResizer' in src):
             print(f"Removing problematic script: {src}")
             script.decompose()
     
@@ -163,8 +168,14 @@ def process_and_generate(image_np, image_path_from_state, sidebar_prompt, header
             print(f"Removing problematic CSS link: {href}")
             link.decompose()
     
+    # Also remove any inline scripts that might contain problematic content
+    for script in soup.find_all('script'):
+        if script.string and ('assets/' in script.string or 'index-' in script.string):
+            print(f"Removing problematic inline script")
+            script.decompose()
+    
     for img in soup.find_all('img'):
-        if img.get('src') and not img['src'].startswith(('http', 'dbata:')):
+        if img.get('src') and not img['src'].startswith(('http', 'data:')):
             original_src = img['src']
             print(f"Processing image: {original_src}")
             
@@ -298,4 +309,10 @@ for path in allowed_paths:
     print(f"  - {path}")
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=allowed_paths)
+    demo.launch(
+        allowed_paths=allowed_paths,
+        show_error=True,
+        quiet=False,
+        favicon_path=None,
+        prevent_thread_lock=True
+    )