Fix HTML rendering issues by cleaning up problematic script and CSS references

app.py

@@ -67,7 +67,25 @@ 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("'", "'")
+    # 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
+    for script in soup.find_all('script'):
+        src = script.get('src', '')
+        if src and ('assets/' in src or 'index-' in src):
+            script.decompose()
+    
+    # Remove any link tags that reference local CSS assets
+    for link in soup.find_all('link'):
+        href = link.get('href', '')
+        if href and ('assets/' in href or 'index-' in href):
+            link.decompose()
+    
+    # Get the cleaned HTML
+    cleaned_code = str(soup)
+    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 +150,19 @@ 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
+    for script in soup.find_all('script'):
+        src = script.get('src', '')
+        if src and ('assets/' in src or 'index-' in src):
+            print(f"Removing problematic script: {src}")
+            script.decompose()
+    
+    for link in soup.find_all('link'):
+        href = link.get('href', '')
+        if href and ('assets/' in href or 'index-' in href):
+            print(f"Removing problematic CSS link: {href}")
+            link.decompose()
+    
     for img in soup.find_all('img'):
         if img.get('src') and not img['src'].startswith(('http', 'data:')):
             original_src = img['src']