Update app.py

app.py

@@ -36,40 +36,159 @@ API_KEY = os.environ["API_KEY"]
 
 
 nlp = spacy.load("en_core_web_sm")
+
 def process_nlp(system_message):
     # Colorize the system message text
     colorized_text = colorize_text(system_message['content'])
     return colorized_text
 
-def colorize_text(text):
+from colour import Color
+
+# define color combinations for different parts of speech
+COLORS = {
+    "NOUN": "#000000",  # Black
+    "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 and color vision deficiencies
+DYSLEXIA_COLORS = {
+    "NOUN": "#000000",
+    "VERB": "#ff6936",
+    "ADJ": "#4363d8",
+    "ADV": "#228b22",
+    "digit": "#9a45d6",
+    "punct": "#ffcc00",
+    "quote": "#b300b3",
+}
+RED_GREEN_COLORS = {
+    "NOUN": "#000000",
+    "VERB": "#fe642e",  # Lighter orange
+    "ADJ": "#2e86c1",   # Lighter blue
+    "ADV": "#82e0aa",   # Lighter green
+    "digit": "#aa6c39", # Brown
+    "punct": "#f0b27a", # Lighter yellow
+    "quote": "#9932cc"  # Darker magenta
+}
+
+# define a muted background color
+BACKGROUND_COLOR = "#ffffff"  # White
+
+# define font and size
+FONT = "OpenDyslexic"
+FONT_SIZE = "18px"
+
+def colorize_text(text, colors=DYSLEXIA_COLORS, background_color=None, font=FONT, font_size=FONT_SIZE):
+    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}**'
-            elif token.pos_ == 'NOUN':
-                colorized_text += f'<span style="color: #FF3300; background-color: transparent;">{token.text_with_ws}</span>'
-            elif token.pos_ == 'VERB':
-                colorized_text += f'<span style="color: #FFFF00; background-color: transparent;">{token.text_with_ws}</span>'
-            elif token.pos_ == 'ADJ':
-                colorized_text += f'<span style="color: #00CC00; background-color: transparent;">{token.text_with_ws}</span>'
-            elif token.pos_ == 'ADV':
-                colorized_text += f'<span style="color: #FF6600; background-color: transparent;">{token.text_with_ws}</span>'
-            elif token.is_digit:
-                colorized_text += f'<span style="color: #9900CC; background-color: transparent;">{token.text_with_ws}</span>'
-            elif token.is_punct:
-                colorized_text += f'<span style="color: #8B4513; background-color: transparent;">{token.text_with_ws}</span>'
-            elif token.is_quote:
-                colorized_text += f'<span style="color: #008080; background-color: transparent;">{token.text_with_ws}</span>'
+                # 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;">'
+                        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;">'
+                        f"{token.text}</span>"
+                    )
             else:
-                colorized_text += token.text_with_ws
+                # 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: {background_color}; '
+                        f'font-family: {font}; '
+                        f'font-size: {font_size}; '
+                        f'font-weight: bold; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'
+                        f"{token.text}</span>"
+                    )
+                elif token.is_digit:
+                    colorized_text += (
+                        f'<span style="color: {colors["digit"]}; '
+                        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;">'
+                        f"{token.text}</span>"
+                    )
+                elif token.is_punct:
+                    colorized_text += (
+                        f'<span style="color: {colors["punct"]}; '
+                        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;">'
+                        f"{token.text}</span>"
+                    )
+                elif token.is_quote:
+                    colorized_text += (
+                        f'<span style="color: {colors["quote"]}; '
+                        f'background-color: {background_color}; '
+                        f'font-family: {font}; '
+                        f'font-size: {font_size}; '
+                        f'text-decoration: none; '
+                        f'padding-right: 0.5em;">'
+                        f"{token.text}</span>"
+                    )
+                else:
+                    # use larger font size for specific parts of speech, such as nouns and verbs
+                    font_size = FONT_SIZE
+                    if token.pos_ in ["NOUN", "VERB"]:
+                        font_size = "22px"
+                    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;">'
+                        f"{token.text}</span>"
+                    )
         colorized_text += "<br>"
-
     return colorized_text
 
+
 def colorize_and_update(system_message, submit_update):
     colorized_system_message = colorize_text(system_message['content'])
     submit_update(None, colorized_system_message)  # Pass the colorized_system_message as the second output