Update app.py

app.py

@@ -80,9 +80,9 @@ def generate_pastel_colors(n):
         return []
 
     # To generate colors, we use the HSL colorspace (see http://en.wikipedia.org/wiki/HSL_color_space)
-    start_hue = 0.6  # 0=red    1/3=0.333=green   2/3=0.666=blue
+    start_hue = 0.0  # 0=red    1/3=0.333=green   2/3=0.666=blue
     saturation = 1.0
-    lightness = 0.8
+    lightness = 0.9
     # We take points around the chromatic circle (hue):
     # (Note: we generate n+1 colors, then drop the last one ([:-1]) because
     # it equals the first one (hue 0 = hue 1))
@@ -122,12 +122,13 @@ def set_sidebar(css):
 
 
 def get_el_annotations(response):
+    el_link_wrapper = "<link rel='stylesheet' href='https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.2/css/all.min.css'><a href='https://en.wikipedia.org/wiki/{}' style='color: #414141'><i class='fa-brands fa-wikipedia-w fa-xs'></i> <span style='font-size: 1.0em; font-family: monospace'> {}</span></a>"
     # swap labels key with ents
     ents = [
         {
             "start": l.start,
             "end": l.end,
-            "label": f"<a href='https://en.wikipedia.org/wiki/{l.label.replace(' ', '_')}' style='color: #222222'>{l.label}</a>",
+            "label": el_link_wrapper.format(l.label.replace(" ", "_"), l.label),
         }
         for l in response.labels
     ]
@@ -217,33 +218,14 @@ def run_client():
                 # else:
                 #     response = response.json()
 
-                # Entity Linking
-                # with stylable_container(
-                #     key="container_with_border",
-                #     css_styles="""
-                #         {
-                #             border: 1px solid rgba(49, 51, 63, 0.2);
-                #             border-radius: 0.5rem;
-                #             padding: 0.5rem;
-                #             padding-bottom: 2rem;
-                #         }
-                #         """,
-                # ):
                 # st.markdown("##")
                 dict_of_ents, options = get_el_annotations(response=response)
                 display = displacy.render(
                     dict_of_ents, manual=True, style="ent", options=options
                 )
-                display = display.replace("\n", " ")
                 with st.container():
                     st.write(display, unsafe_allow_html=True)
 
-                # st.markdown("####")
-                # st.markdown("#### Relation Extraction")
-
-                # with st.container():
-                #     st.write("Coming :)", unsafe_allow_html=True)
-
         else:
             st.error("Please enter some text.")