Update app.py

app.py

@@ -325,7 +325,7 @@ def perplexity_to_color(perplexity, min_perp=1, max_perp=1000):
         max_perp: Maximum perplexity (maps to red)
 
     Returns:
-        Hex color string (e.g., "#00FF00")
+        Tuple of (r, g, b) values as integers (0-255)
     """
     # Clamp perplexity to range
     perp = max(min_perp, min(max_perp, perplexity))
@@ -356,7 +356,7 @@ def perplexity_to_color(perplexity, min_perp=1, max_perp=1000):
         g = int(255 * (1 - factor))
         b = 0
 
-    return f"#{r:02x}{g:02x}{b:02x}"
+    return (r, g, b)
 
 
 def create_visualization(tokens, perplexities):
@@ -390,7 +390,9 @@ def create_visualization(tokens, perplexities):
             continue
 
         # Clean token for display
-        clean_token = token.replace("</w>", "").replace("##", "").strip()
+        clean_token = (
+            token.replace("</w>", "").replace("##", "").replace("Ġ", "").strip()
+        )
         if not clean_token:
             continue
 
@@ -439,15 +441,20 @@ def create_visualization(tokens, perplexities):
         bg_alpha = VIZ_SETTINGS["color_scheme"].get("background_alpha", 0.7)
         border_alpha = VIZ_SETTINGS["color_scheme"].get("border_alpha", 0.9)
 
+        # Get RGB color from perplexity
+        r, g, b = perplexity_to_color(
+            perp, min_perp=1, max_perp=VIZ_SETTINGS["max_perplexity_display"]
+        )
+
         # Create colored span with tooltip
         html_parts.append(
             f'<span style="'
-            f"background-color: rgba({perplexity_to_color(perp, min_perp=1, max_perp=VIZ_SETTINGS['max_perplexity_display'])}, {bg_alpha}); "
+            f"background-color: rgba({r}, {g}, {b}, {bg_alpha}); "
             f"color: #000; "
             f"padding: 2px 4px; "
             f"margin: 1px; "
             f"border-radius: 3px; "
-            f"border: 1px solid rgba({perplexity_to_color(perp, min_perp=1, max_perp=VIZ_SETTINGS['max_perplexity_display'])}, {border_alpha}); "
+            f"border: 1px solid rgba({r}, {g}, {b}, {border_alpha}); "
             f"font-weight: 500; "
             f"cursor: help; "
             f"display: inline-block;"