style update

app.py

@@ -1,6 +1,6 @@
+from turtle import color
 import pandas as pd
 import streamlit as st
-import streamlit.components.v1 as components
 from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
 
 from src import GBRT, wikipedia_search, google_search
@@ -67,11 +67,9 @@ def get_candidates(mentions_tags):
 # ---------------------------------------------------------------------------
 def display_tag(text, typ, label):
     if label != 'NIL':
-        label = "https://en.wikipedia.org/wiki/" + label
-        return f'<a style="margin: 0px 5px 0px 0px; padding: 2px 4px; border-radius: 4px; text-decoration:none; background-color:{COLOR[typ]}; color: white; cursor:pointer; white-space: nowrap; font-weight:400" href={label} target="_blank"><span style="margin-right:2px;">{text}</span><span style="background-color: white; color: {COLOR[typ]}; padding: 1px 2px; border-radius: 4px; font-size: 0.8em">{typ}</span></a>'
+        return f'<span style="color: {COLOR[typ]}" class="tag">[**{text}** <small>{typ}</small>](https://en.wikipedia.org/wiki/{label})</span>'
     else:
-        return f'<span style="margin: 0px 5px 0px 0px; padding: 2px 4px; border-radius: 4px; text-decoration:none; background-color: #EF4444; color: white; cursor:not-allowed; white-space: nowrap;"><span style="margin-right:2px">{text}</span><span style="background-color: white; color: #EF4444; padding: 1px 2px; border-radius: 4px; font-size: 0.8em">{typ}</span></span>'
-
+        return f'<span style="color: #EF4444" class="tag">**{text}** <small>{typ}</small></span>'
 
 
 # ---------------------------------------------------------------------------
@@ -97,8 +95,8 @@ def main(text):
 
     with col2:
         st.write("### Disambiguated Text")
-        components.html(f'<div style="white-space: pre-line; line-height: 1.65; font-family: sans-serif">{tagged}</div>', 
-                        height=700, scrolling=True)
+        st.markdown("<style>a {color: inherit !important; text-decoration: none;} .tag {border: 1px solid; border-radius: 5px; padding: 0px 3px;}</style>", unsafe_allow_html=True)
+        st.write(tagged, unsafe_allow_html=True)
 
     df = pd.DataFrame(data=preditions, columns=['Mention', 'Prediction', 'Confidence'])
     st.write("**Additional Information**")
@@ -114,5 +112,3 @@ if __name__ == '__main__':
                                   'George Washington went to Washington.', height=350)
         if user_input:
             main(user_input)
-
-