Update app.py

app.py

@@ -42,6 +42,79 @@ def process_nlp(system_message):
     colorized_text = colorize_text(system_message['content'])
     return colorized_text
 
+# Define color combinations for different parts of speech
+COLORS = {
+    "NOUN": "#FF3300",
+    "VERB": "#FFFF00",
+    "ADJ": "#00CC00",
+    "ADV": "#FF6600",
+    "digit": "#9900CC",
+    "punct": "#8B4513",
+    "quote": "#008080",
+}
+
+# Define color combinations for individuals with dyslexia
+DYSLEXIA_COLORS = {
+    "NOUN": "#1E90FF",
+    "VERB": "#FFFF00",
+    "ADJ": "#00CED1",
+    "ADV": "#FF8C00",
+    "digit": "#800080",
+    "punct": "#A0522D",
+    "quote": "#ADFF2F",
+}
+
+# Define a dark background color
+DARK_BG = "#212121"
+
+def colorize_text(text, colors=None):
+    if colors is None:
+        colors = COLORS
+
+    colorized_text = ""
+    lines = text.split("\n")
+
+    for line in lines:
+        doc = nlp(line)
+
+        for token in doc:
+            if token.ent_type_:
+                colorized_text += f"**{token.text_with_ws}**"
+            else:
+                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>"
+                    )
+                elif token.is_digit:
+                    colorized_text += (
+                        f'<span style="color: {colors["digit"]}; '
+                        f'background-color: transparent;">'
+                        f"{token.text_with_ws}</span>"
+                    )
+                elif token.is_punct:
+                    colorized_text += (
+                        f'<span style="color: {colors["punct"]}; '
+                        f'background-color: transparent;">'
+                        f"{token.text_with_ws}</span>"
+                    )
+                elif token.is_quote:
+                    colorized_text += (
+                        f'<span style="color: {colors["quote"]}; '
+                        f'background-color: transparent;">'
+                        f"{token.text_with_ws}</span>"
+                    )
+                else:
+                    colorized_text += token.text_with_ws
+
+        colorized_text += "<br>"
+
+    return colorized_text
+
+
+
 def colorize_text(text):
     colorized_text = ""
     lines = text.split("\n")