Update app.py

app.py

@@ -42,75 +42,139 @@ def process_nlp(system_message):
     colorized_text = colorize_text(system_message['content'])
     return colorized_text
 
-# Define color combinations for different parts of speech
+# define color combinations for different parts of speech
 COLORS = {
-    "NOUN": "#FF3300",
-    "VERB": "#FFFF00",
-    "ADJ": "#00CC00",
-    "ADV": "#FF6600",
-    "digit": "#9900CC",
-    "punct": "#8B4513",
-    "quote": "#008080",
+    "NOUN": "#5e5e5e",  # Dark gray
+    "VERB": "#ff6936",  # Orange
+    "ADJ": "#4363d8",   # Blue
+    "ADV": "#228b22",   # Green
+    "digit": "#9a45d6", # Purple
+    "punct": "#ffcc00", # Yellow
+    "quote": "#b300b3"  # Magenta
 }
 
-# Define color combinations for individuals with dyslexia
+# define color combinations for individuals with dyslexia
 DYSLEXIA_COLORS = {
-    "NOUN": "#1E90FF",
-    "VERB": "#FFFF00",
-    "ADJ": "#00CED1",
-    "ADV": "#FF8C00",
-    "digit": "#800080",
-    "punct": "#A0522D",
-    "quote": "#ADFF2F",
+    "NOUN": "#5e5e5e",
+    "VERB": "#ff6936",
+    "ADJ": "#4363d8",
+    "ADV": "#228b22",
+    "digit": "#9a45d6",
+    "punct": "#ffcc00",
+    "quote": "#b300b3"
 }
 
-# Define a dark background color
-DARK_BG = "#212121"
+# define a muted background color
+BACKGROUND_COLOR = "#f5f5f5"  # Light gray
 
-def colorize_text(text, colors=None):
+# define font and size
+FONT = "Arial"
+FONT_SIZE = "14px"
+
+# load the English language model
+nlp = spacy.load('en_core_web_sm')
+
+def colorize_text(text, colors=None, background_color=None):
     if colors is None:
         colors = COLORS
-
     colorized_text = ""
     lines = text.split("\n")
-
+    
+    # set background color
+    if background_color is None:
+        background_color = BACKGROUND_COLOR
+    
+    # iterate over the lines in the text
     for line in lines:
+        # parse the line with the language model
         doc = nlp(line)
-
+        # iterate over the tokens in the line
         for token in doc:
+            # check if the token is an entity
             if token.ent_type_:
-                colorized_text += f"**{token.text_with_ws}**"
+                # use dyslexia colors for entity if available
+                if colors == COLORS:
+                    color = DYSLEXIA_COLORS.get(token.pos_, None)
+                else:
+                    color = colors.get(token.pos_, None)
+                # check if a color is available for the token
+                if color is not None:
+                    colorized_text += (
+                        f'<span style="color: {color}; '
+                        f'background-color: {background_color}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
+                    )
+                else:
+                    colorized_text += (
+                        f'<span style="font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
+                    )
             else:
+                # check if a color is available for the token
                 color = colors.get(token.pos_, None)
                 if color is not None:
                     colorized_text += (
                         f'<span style="color: {color}; '
-                        f'background-color: transparent;">'
-                        f"{token.text_with_ws}</span>"
+                        f'background-color: {background_color}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
                     )
                 elif token.is_digit:
                     colorized_text += (
                         f'<span style="color: {colors["digit"]}; '
-                        f'background-color: transparent;">'
-                        f"{token.text_with_ws}</span>"
+                        f'background-color: {background_color}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
                     )
                 elif token.is_punct:
                     colorized_text += (
                         f'<span style="color: {colors["punct"]}; '
-                        f'background-color: transparent;">'
-                        f"{token.text_with_ws}</span>"
+                        f'background-color: {background_color}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
                     )
                 elif token.is_quote:
                     colorized_text += (
                         f'<span style="color: {colors["quote"]}; '
-                        f'background-color: transparent;">'
-                        f"{token.text_with_ws}</span>"
+                        f'background-color: {background_color}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
                     )
                 else:
-                    colorized_text += token.text_with_ws
-
+                    colorized_text += (
+                        f'<span style="font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'  # Add space between tokens
+                        f"{token.text}</span>"
+                    )
         colorized_text += "<br>"
-
+    
     return colorized_text