Update app.py

app.py

@@ -74,13 +74,13 @@ def main(prob1):
     
 # Define HTML for the legend
 legend_html = """
-<div style='margin-top: 20px;'>
+<div style='margin-top: 20px; color: white;'>  <!-- Ensure the legend text is white for visibility -->
     <h3>NER Legend</h3>
-    <ul style='list-style-type:none;'>
+    <ul style='list-style-type:none; padding-left: 0;'>  <!-- Remove padding from the list -->
 """
 
 for entity, color in entity_colors.items():
-    legend_html += f"<li><span style='color:{color}; font-weight:bold;'>▉ </span>{entity}</li>"
+    legend_html += f"<li><span style='color: white; background-color: {color}; padding: 5px 10px; margin-right: 5px; border-radius: 5px;'>{entity}</span></li>"
 
 legend_html += "</ul></div>"