Update src/streamlit_app.py

src/streamlit_app.py

@@ -276,7 +276,7 @@ def generate_html_report(df, text_input, elapsed_time, df_topic_data, entity_col
         df,
         path=[px.Constant("All Entities"), 'category', 'label', 'text'],
         values='score',
-        color='category',
+        color='label',
         title="Entity Distribution by Category and Label",
         color_discrete_sequence=px.colors.qualitative.Bold
     )
@@ -747,7 +747,7 @@ if st.session_state.show_results:
                 df,
                 path=[px.Constant("All Entities"), 'category', 'label', 'text'],
                 values='score',
-                color='category',
+                color='label',
                 color_discrete_sequence=px.colors.qualitative.Bold
             )
             fig_treemap.update_layout(margin=dict(t=10, l=10, r=10, b=10))