Update app.py

app.py

@@ -343,13 +343,24 @@ def visualize_molecule_2d_3d(smiles: str, name: str):
         svg_2d = svg_2d.replace('stroke:rgb(0,0,0)', 'stroke:rgb(255,255,255)')
         svg_2d = svg_2d.replace('fill:rgb(0,0,0)', 'fill:rgb(255,255,255)')
         
-        # Fix text/font colors for atom labels
+        # Fix text/font colors for atom labels and stereochemistry labels
         svg_2d = svg_2d.replace('<text', '<text fill="white"')
         svg_2d = svg_2d.replace('fill="white" fill="white"', 'fill="white"')  # Remove duplicates
         svg_2d = svg_2d.replace('color="black"', 'color="white"')
         svg_2d = svg_2d.replace('color:#000000', 'color:#FFFFFF')
         svg_2d = svg_2d.replace('color:#000', 'color:#FFF')
         
+        # Additional fixes for stereochemistry labels (R) and (S) that might be rendered differently
+        # These labels are often in separate text elements
+        import re
+        # Find all text elements and ensure they have white fill
+        svg_2d = re.sub(r'<text([^>]*?)(?:fill="[^"]*")?([^>]*?)>', r'<text\1 fill="white"\2>', svg_2d)
+        # Fix any remaining black text in style attributes
+        svg_2d = re.sub(r'style="([^"]*?)fill:\s*black([^"]*?)"', r'style="\1fill:white\2"', svg_2d)
+        svg_2d = re.sub(r'style="([^"]*?)fill:\s*#000000([^"]*?)"', r'style="\1fill:#FFFFFF\2"', svg_2d)
+        svg_2d = re.sub(r'style="([^"]*?)fill:\s*#000([^"]*?)"', r'style="\1fill:#FFF\2"', svg_2d)
+        svg_2d = re.sub(r'style="([^"]*?)fill:\s*rgb\(0,0,0\)([^"]*?)"', r'style="\1fill:rgb(255,255,255)\2"', svg_2d)
+        
         # Embed the SVG within a div with a dark background for consistency
         svg_2d = f'<div style="background-color: #1C1C1C; padding: 10px; border-radius: 5px;">{svg_2d}</div>'