Restore BeautifulSoup HTML cleaning with error handling to fix CSS loading issues

app.py

@@ -67,7 +67,30 @@ def render_preview(code: str, width: int, height: int, scale: float) -> str:
     """
     Preview renderer with both width and height control for the inner canvas.
     """
-    safe_code = html.escape(code).replace("'", "'")
+    try:
+        # Clean up the HTML code to remove problematic script and CSS references
+        soup = BeautifulSoup(code, 'html.parser')
+        
+        # Remove any script tags that reference problematic files
+        for script in soup.find_all('script'):
+            src = script.get('src', '')
+            if src and any(pattern in src for pattern in ['assets/', 'index-', 'iframeResizer']):
+                script.decompose()
+        
+        # Remove any link tags that reference problematic CSS files
+        for link in soup.find_all('link'):
+            href = link.get('href', '')
+            if href and any(pattern in href for pattern in ['assets/', 'index-']):
+                link.decompose()
+        
+        # Get the cleaned HTML
+        cleaned_code = str(soup)
+    except Exception as e:
+        print(f"Error cleaning HTML in render_preview: {e}")
+        # Fallback to original code if cleaning fails
+        cleaned_code = code
+    
+    safe_code = html.escape(cleaned_code).replace("'", "'")
     iframe_html = f"""
     <div style="width: 100%; max-width: 1920px; margin: 0 auto; overflow-x: auto; overflow-y: hidden;">
         <div style="
@@ -132,6 +155,22 @@ def process_and_generate(image_np, image_path_from_state, sidebar_prompt, header
     
     print(f"Processing HTML for run_id: {run_id}")
     
+    # Clean up problematic script and CSS references with error handling
+    try:
+        for script in soup.find_all('script'):
+            src = script.get('src', '')
+            if src and any(pattern in src for pattern in ['assets/', 'index-', 'iframeResizer']):
+                print(f"Removing problematic script: {src}")
+                script.decompose()
+        
+        for link in soup.find_all('link'):
+            href = link.get('href', '')
+            if href and any(pattern in href for pattern in ['assets/', 'index-']):
+                print(f"Removing problematic CSS link: {href}")
+                link.decompose()
+    except Exception as e:
+        print(f"Error cleaning HTML in process_and_generate: {e}")
+    
     for img in soup.find_all('img'):
         if img.get('src') and not img['src'].startswith(('http', 'data:')):
             original_src = img['src']