Update app.py

app.py

@@ -44,9 +44,153 @@ def process_nlp(system_message):
 
 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_:
+#                 # 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:
+#                 # 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
+
 # define color combinations for different parts of speech
 COLORS = {
-    "NOUN": "#000000",  # Black
+    "NOUN": "#5e5e5e",  # Dark gray
     "VERB": "#ff6936",  # Orange
     "ADJ": "#4363d8",   # Blue
     "ADV": "#228b22",   # Green
@@ -55,34 +199,28 @@ COLORS = {
     "quote": "#b300b3"  # Magenta
 }
 
-# define color combinations for individuals with dyslexia and color vision deficiencies
+# define color combinations for individuals with dyslexia
 DYSLEXIA_COLORS = {
-    "NOUN": "#000000",
+    "NOUN": "#5e5e5e",
     "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
+    "quote": "#b300b3"
 }
 
 # define a muted background color
-BACKGROUND_COLOR = "#ffffff"  # White
+BACKGROUND_COLOR = "#f5f5f5"  # Light gray
 
 # define font and size
-FONT = "OpenDyslexic"
-FONT_SIZE = "18px"
+FONT = "Arial"
+FONT_SIZE = "14px"
 
-def colorize_text(text, colors=DYSLEXIA_COLORS, background_color=None, font=FONT, font_size=FONT_SIZE):
+# load the English language model
+nlp = spacy.load('en_core_web_sm')
+
+def colorize_text(text, colors=DYSLEXIA_COLORS, background_color=None):
     if colors is None:
         colors = COLORS
     colorized_text = ""
@@ -110,20 +248,20 @@ def colorize_text(text, colors=DYSLEXIA_COLORS, background_color=None, font=FONT
                     colorized_text += (
                         f'<span style="color: {color}; '
                         f'background-color: {background_color}; '
-                        f'font-family: {font}; '
-                        f'font-size: {font_size}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'font-weight: bold; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        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'<span style="font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'font-weight: bold; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        f'padding-right: 0.5em;">'  # Add space between tokens
                         f"{token.text}</span>"
                     )
             else:
@@ -133,59 +271,56 @@ def colorize_text(text, colors=DYSLEXIA_COLORS, background_color=None, font=FONT
                     colorized_text += (
                         f'<span style="color: {color}; '
                         f'background-color: {background_color}; '
-                        f'font-family: {font}; '
-                        f'font-size: {font_size}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'font-weight: bold; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        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: {background_color}; '
-                        f'font-family: {font}; '
-                        f'font-size: {font_size}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'font-weight: bold; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        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: {background_color}; '
-                        f'font-family: {font}; '
-                        f'font-size: {font_size}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'font-weight: bold; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        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: {background_color}; '
-                        f'font-family: {font}; '
-                        f'font-size: {font_size}; '
+                        f'font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        f'padding-right: 0.5em;">'  # Add space between tokens
                         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'<span style="font-family: {FONT}; '
+                        f'font-size: {FONT_SIZE}; '
                         f'font-weight: bold; '
                         f'text-decoration: none; '
-                        f'padding-right: 0.5em;">'
+                        f'padding-right: 0.5em;">'  # Add space between tokens
                         f"{token.text}</span>"
                     )
         colorized_text += "<br>"
+    
     return colorized_text