Update src/streamlit_app.py

src/streamlit_app.py

@@ -79,7 +79,9 @@ def highlight_entities(text, df_entities):
         color = entity_color_map.get(label, '#000000')
         # Create a span with background color and tooltip
       # Change the foreground text color to BLACK, and use a white shadow for readability
-        highlight_html = f'<span style="background-color: {color}; color: #000000; padding: 2px 4px; border-radius: 3px; cursor: help; text-shadow: 0.5px 0.5px 0px #FFFFFF;" title="{label}">{entity_text}</span>'
+        # Force text color to BLACK and add a clear, contrasting border
+        # In the 'highlight_entities' function:
+        highlight_html = f'<span style="background-color: {color}; color: #000000; padding: 2px 4px; border-radius: 3px; cursor: help; border: 1px solid #FFFFFF;" title="{label}">{entity_text}</span>'
         
         # Replace the original text segment with the highlighted HTML
         highlighted_text = highlighted_text[:start] + highlight_html + highlighted_text[end:]
@@ -148,12 +150,13 @@ def create_topic_word_bubbles(df_topic_data):
     fig.update_layout(
         xaxis_title="Entity/Word (Bubble size = Word Weight)",
         yaxis_title="Word Weight",
-        xaxis={'tickangle': -45, 'showgrid': False, 'showticklabels': False},
+        xaxis={'tickangle': -45, 'showgrid': False, 'showticklabels': False, 'range': [0, 10]},
         yaxis={'showgrid': True},
         showlegend=True,
         plot_bgcolor='#f9f9f9', # Changed from pink
         paper_bgcolor='#f9f9f9', # Changed from pink
-        height=800,
+        height=1000,
+        autosize=False,
         margin=dict(t=50, b=100, l=50, r=10),
     )
     fig.update_traces(hovertemplate='<b>%{customdata[0]}</b><br>Weight: %{customdata[1]:.3f}<extra></extra>', marker=dict(line=dict(width=1, color='DarkSlateGrey')))