Correct app.py

app.py

@@ -30,6 +30,7 @@ def get_highlighted_text(doc):
     """
     Wraps detected ASCs in each sentence with a span tag containing an inline style.
     The inline style sets a custom background color based on the tag.
+    Additionally, the entity's tag name is appended as a small superscript to make it always visible.
     """
     highlighted_sentences = []
     for sent in doc.sents:
@@ -37,17 +38,19 @@ def get_highlighted_text(doc):
         # Find all entities fully contained within this sentence.
         ents_in_sent = [ent for ent in doc.ents if ent.start_char >= sent.start_char and ent.end_char <= sent.end_char]
         if ents_in_sent:
-            # Process entities in reverse order so the character offsets remain valid.
+            # Process entities in reverse order so character offsets remain valid.
             ents_in_sent = sorted(ents_in_sent, key=lambda x: x.start_char, reverse=True)
             for ent in ents_in_sent:
                 ent_start = ent.start_char - sent.start_char
                 ent_end = ent.end_char - sent.start_char
                 # Get the color from our scheme; use default if not defined.
                 color = color_scheme.get(ent.label_, "#cccccc")
+                # Wrap the entity with inline style and append its label as a superscript.
                 s = (
                     s[:ent_start]
                     + f'<span style="background-color: {color}; font-weight: bold;" title="{ent.label_}">'
                     + s[ent_start:ent_end]
+                    + f'<sup style="font-size:0.6em; margin-left:2px;">{ent.label_}</sup>'
                     + '</span>'
                     + s[ent_end:]
                 )
@@ -79,39 +82,28 @@ sample_text = (
     "But this wasn't just any baker—Oliver had a magical secret. Every loaf of bread he baked had the power to grant a wish, "
     "but only if it was made with true joy in his heart. The villagers adored Oliver’s bread, though they had no idea about its magic. "
     "They just knew it tasted unlike anything else—warm, soft, and with a hint of something... extraordinary. Only a few people had ever discovered "
-    "its secret, and even fewer had made wishes, because the key was in asking for something truly selfless.\n\n"
-    
+    "its secret, and even fewer had made wishes, because the key was in asking for something truly selfless. "
     "One crisp autumn morning, a traveling musician named Lyla passed through the village. She had a worn-out lute, tattered clothes, "
-    "and a bright spark of hope in her eyes despite her hardships. Drawn by the smell of freshly baked bread, she wandered into Oliver’s bakery.\n\n"
-    
-    "Oliver, busy dusting flour off his hands, smiled warmly. “What can I do for you?” he asked.\n\n"
-    
+    "and a bright spark of hope in her eyes despite her hardships. Drawn by the smell of freshly baked bread, she wandered into Oliver’s bakery. "
+    "Oliver, busy dusting flour off his hands, smiled warmly. “What can I do for you?” he asked. "
     "Lyla’s stomach growled in response, and she laughed, a sound as melodic as her lute playing. “Just one loaf of bread, please,” she said, "
-    "placing a single coin on the counter, all she had left.\n\n"
-    
+    "placing a single coin on the counter, all she had left. "
     "Oliver, feeling the joy in the simple act of serving someone in need, baked the loaf with care. As the bread came out of the oven, a golden glow "
-    "shimmered around it for just a moment. Lyla didn’t notice, but Oliver did, and he smiled to himself.\n\n"
-    
+    "shimmered around it for just a moment. Lyla didn’t notice, but Oliver did, and he smiled to himself. "
     "As Lyla took her first bite outside the shop, she felt warmth spread through her whole being. She thought of how she missed playing her lute to "
     "cheer others up, but it was broken and beyond repair. Without knowing the magic at work, she whispered under her breath, 'I just wish I could bring "
-    "music to everyone again.'\n\n"
-    
+    "music to everyone again.' "
     "That evening, as Lyla set up by the village fountain to play, something miraculous happened. Her old, broken lute began to transform, glowing just "
     "like the bread had. In her hands now was the most beautiful lute she had ever seen, shimmering with gold and silver strings. When she strummed it, "
-    "the music that flowed out wasn’t just sound—it was pure joy.\n\n"
-    
+    "the music that flowed out wasn’t just sound—it was pure joy."
     "The village gathered around her, drawn by the enchanting melodies that seemed to fill their hearts with warmth and happiness. People laughed, danced, "
-    "and even sang along, forgetting their worries.\n\n"
-    
+    "and even sang along, forgetting their worries. "
     "Oliver watched from his bakery window, smiling to himself. He had no idea what Lyla had wished for, but he could see its magic at work. He knew his "
-    "bread had once again brought a little bit of magic to the world.\n\n"
-    
+    "bread had once again brought a little bit of magic to the world. "
     "And so, every day after that, Lyla would play her magical lute in the village square, bringing joy to everyone who heard her music. People came from "
-    "distant lands just to hear her play, and with every note, the village grew a little brighter, a little happier.\n\n"
-    
+    "distant lands just to hear her play, and with every note, the village grew a little brighter, a little happier. "
     "As for Oliver, he continued to bake with joy in his heart, knowing that sometimes the smallest acts of kindness—like baking a loaf of bread—could "
-    "make the biggest difference in someone’s life.\n\n"
-    
+    "make the biggest difference in someone’s life. "
     "And so, the village became known not just for its delicious bread, but for its music, magic, and the happiness that lingered in every corner. The end. "
     "(Generated by ChatGPT 4o 🤖)"
 )