Simplify HTML cleaning to fix 500 error - use basic string replacement instead of BeautifulSoup

app.py

@@ -67,29 +67,8 @@ def render_preview(code: str, width: int, height: int, scale: float) -> str:
     """
     Preview renderer with both width and height control for the inner canvas.
     """
-    # 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 or iframeResizer
-    for script in soup.find_all('script'):
-        src = script.get('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
-    for link in soup.find_all('link'):
-        href = link.get('href', '')
-        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("'", "'")
+    # Simple HTML escaping for now to avoid potential issues
+    safe_code = html.escape(code).replace("'", "'")
     
     iframe_html = f"""
     <div style="width: 100%; max-width: 1920px; margin: 0 auto; overflow-x: auto; overflow-y: hidden;">
@@ -155,24 +134,10 @@ 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 or 'iframeResizer' 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()
-    
-    # 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()
+    # Simple cleanup for now - just remove obvious problematic references
+    html_content = html_content.replace('index-BQPjLIsY.js', '')
+    html_content = html_content.replace('index-BOW2xVAS.css', '')
+    html_content = html_content.replace('iframeResizer', '')
     
     for img in soup.find_all('img'):
         if img.get('src') and not img['src'].startswith(('http', 'data:')):