Upload streamlit_app.py

src/streamlit_app.py

@@ -9,15 +9,15 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification
 
 # Mapping of label to color
 LABEL_COLORS = {
-    'LABEL-0': '#ffffff',  # NONE (no color)
-    'LABEL-1': '#fff4e6',  # B-DATE (creamy orange)
-    'LABEL-2': '#ffe9ec',  # I-DATE (creamy pink)
-    'LABEL-3': '#f3ffe3',  # B-TIME (creamy green)
-    'LABEL-4': '#e6f7ff',  # I-TIME (creamy blue)
-    'LABEL-5': '#f9f7e8',  # B-DURATION (creamy yellow)
-    'LABEL-6': '#f6eaff',  # I-DURATION (creamy purple)
-    'LABEL-7': '#fdf6ec',  # B-SET (creamy beige)
-    'LABEL-8': '#f6fff8',  # I-SET (creamy mint)
+    'LABEL-0': '#cccccc',  # NONE
+    'LABEL-1': '#ffadad',  # B-DATE
+    'LABEL-2': '#ffd6a5',  # I-DATE
+    'LABEL-3': '#fdffb6',  # B-TIME
+    'LABEL-4': '#caffbf',  # I-TIME
+    'LABEL-5': '#9bf6ff',  # B-DURATION
+    'LABEL-6': '#a0c4ff',  # I-DURATION
+    'LABEL-7': '#bdb2ff',  # B-SET
+    'LABEL-8': '#ffc6ff',  # I-SET
 }
 
 LABEL_MEANINGS = {
@@ -76,7 +76,10 @@ def colorize_entities(ner_result: List[Tuple[str, str]]) -> str:
         if norm_label != 'LABEL-0':
             color = LABEL_COLORS.get(norm_label, '#eeeeee')
             label_meaning = LABEL_MEANINGS.get(norm_label, norm_label)
-            html += f'<span style="background-color:{color};padding:2px 4px;border-radius:4px;margin:1px;" title="{label_meaning}">{token}</span> '
+            html += (
+                f'<span class="ner-entity" style="background-color:{color};padding:2px 4px;border-radius:4px;margin:1px;" '
+                f'data-tooltip="{label_meaning}">{token}</span> '
+            )
         else:
             html += f'{token} '
     return html
@@ -116,6 +119,32 @@ def legend_html() -> str:
 
 st.title('LLM-powered Named Entity Recognition (NER)')
 
+st.markdown(
+    '''
+    <style>
+    .ner-entity {
+        position: relative;
+        cursor: pointer;
+    }
+    .ner-entity[data-tooltip]:hover:after {
+        content: attr(data-tooltip);
+        position: absolute;
+        left: 0;
+        top: 100%;
+        background: #222;
+        color: #fff;
+        padding: 2px 8px;
+        border-radius: 4px;
+        white-space: nowrap;
+        z-index: 10;
+        font-size: 0.9em;
+        margin-top: 2px;
+    }
+    </style>
+    ''',
+    unsafe_allow_html=True
+)
+
 st.markdown('**Legend:**')
 st.markdown(legend_html(), unsafe_allow_html=True)