Update src/streamlit_app.py

src/streamlit_app.py

@@ -76,7 +76,7 @@ st.markdown(
 # --- Page Configuration and UI Elements ---
 st.set_page_config(layout="wide", page_title="Named Entity Recognition App")
 
-st.subheader("ProductTag", divider="red")
+st.subheader("ProductTag", divider="gray")
 st.link_button("by nlpblogs", "https://nlpblogs.com", type="tertiary")
 
 expander = st.expander("**Important notes on the ProductTag**")
@@ -97,7 +97,7 @@ expander.write("""
 """)
 
 with st.sidebar:
-    st.subheader("Build your own NER Web App in a minute without writing a single line of code.", divider="red")
+    st.subheader("Build your own NER Web App in a minute without writing a single line of code.", divider="gray")
     st.link_button("NER File Builder", "https://nlpblogs.com/shop/named-entity-recognition-ner/ner-file-builder/", type="primary")
 
     st.text("")
@@ -196,7 +196,7 @@ if st.button("Results"):
                     experiment.log_parameter("input_text", text)
                     experiment.log_table("predicted_entities", df)
 
-                st.subheader("Extracted Entities", divider = "red")
+                st.subheader("Extracted Entities", divider = "gray")
                 st.dataframe(df.style.set_properties(**{"border": "2px solid gray", "color": "blue", "font-size": "16px"}))
 
                 with st.expander("See Glossary of tags"):
@@ -213,7 +213,7 @@ if st.button("Results"):
                 
 
                 # Tree map
-                st.subheader("Tree map", divider = "red")
+                st.subheader("Tree map", divider = "gray")
                 fig_treemap = px.treemap(df, path=[px.Constant("all"), 'category', 'label', 'text'], values='score', color='category')
                 fig_treemap.update_layout(margin=dict(t=50, l=25, r=25, b=25))
                
@@ -226,20 +226,20 @@ if st.button("Results"):
 
                 col1, col2 = st.columns(2)
                 with col1:
-                    st.subheader("Pie chart", divider = "red")
+                    st.subheader("Pie chart", divider = "gray")
                     fig_pie = px.pie(grouped_counts, values='count', names='category',
                                     hover_data=['count'], labels={'count': 'count'}, title='Percentage of predicted categories')
                     fig_pie.update_traces(textposition='inside', textinfo='percent+label')
                     st.plotly_chart(fig_pie)
 
                 with col2:
-                    st.subheader("Bar chart", divider = "red")
+                    st.subheader("Bar chart", divider = "gray")
                     fig_bar = px.bar(grouped_counts, x="count", y="category", color="category", text_auto=True,
                                     title='Occurrences of predicted categories')
                     st.plotly_chart(fig_bar)
 
                 # Most Frequent Entities
-                st.subheader("Most Frequent Entities", divider="red")
+                st.subheader("Most Frequent Entities", divider="gray")
                 word_counts = df['text'].value_counts().reset_index()
                 word_counts.columns = ['Entity', 'Count']
                 repeating_entities = word_counts[word_counts['Count'] > 1]