Update src/streamlit_app.py

src/streamlit_app.py

@@ -136,12 +136,9 @@ def create_topic_word_bubbles(df_topic_data):
         y='weight',
         size='weight',
         color='topic',
-        # --- MODIFICATION START ---
-        # 1. Use 'word' for the text label on the chart
+        # Set text to the word
         text='word', 
-        # 2. Keep hover_name for tooltip on hover (optional, but good practice)
         hover_name='word', 
-        # --- MODIFICATION END ---
         size_max=80,
         title='Topic Word Weights (Bubble Chart)',
         color_discrete_sequence=px.colors.qualitative.Bold,
@@ -154,9 +151,10 @@ def create_topic_word_bubbles(df_topic_data):
     )
 
     fig.update_layout(
-        xaxis_title="Entity/Word (Bubble size = Word Weight)",
+        xaxis_title="Entity/Word",
         yaxis_title="Word Weight",
-        xaxis={'tickangle': -45, 'showgrid': False, 'showticklabels': False}, # Hide x-axis labels since words are now labels
+        # Hide x-axis labels since words are now labels
+        xaxis={'tickangle': -45, 'showgrid': False, 'showticklabels': False}, 
         yaxis={'showgrid': True},
         showlegend=True,
         plot_bgcolor='#f9f9f9',
@@ -165,11 +163,14 @@ def create_topic_word_bubbles(df_topic_data):
         margin=dict(t=50, b=100, l=50, r=10),
     )
     
-    # Update traces to show the word text and adjust hover template
+    # Update traces to show the word text, set the text position, and set text color
     fig.update_traces(
         # Position the text on top of the bubble
         textposition='middle center', 
-        # Customize hover template for better readability
+        # --- THE KEY FIX IS HERE ---
+        # Set the text color to white for visibility against dark bubble colors
+        textfont=dict(color='white', size=10),
+        # ---------------------------
         hovertemplate='<b>%{customdata[0]}</b><br>Weight: %{customdata[1]:.3f}<extra></extra>',
         marker=dict(line=dict(width=1, color='DarkSlateGrey'))
     )