Update src/streamlit_app.py

src/streamlit_app.py

@@ -75,15 +75,21 @@ st.markdown(
 
 # --- Page Configuration and UI Elements ---
 st.set_page_config(layout="wide", page_title="Named Entity Recognition App")
-st.subheader("Business Core", divider="green")
+st.subheader("Business Core", divider="orange")
 st.link_button("by nlpblogs", "https://nlpblogs.com", type="tertiary")
 expander = st.expander("**Important notes**")
-expander.write("""**Named Entities:** This HR.ai predicts thirty-six (36) labels: "Email", "Phone_number", "Street_address", "City", "Country", "Date_of_birth", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Terminated", "Retired", "Job_title", "Date", "Organization", "Role", "Performance_score", "Leave_of_absence", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance", "Pay_rate", "Annual_salary", "Tax", "Deductions", "Interview_type", "Applicant", "Referral", "Job_board", "Recruiter", "Offer_letter", "Agreement", "Certification", "Skill"
+expander.write("""**Named Entities:** This Business Core predicts twenty-six (26) labels: "Person", "Contact", "Company", "Department", "Vendor", "Client", "Office", "Warehouse", "Address", "City", "State", "Country", "Date", "Time", "Time Period", "Revenue", "Cost", "Budget", "Invoice Number", "Product", "Service", "Task", "Project", "Status", "Asset", "Transaction"
+
 Results are presented in easy-to-read tables, visualized in an interactive tree map, pie chart and bar chart, and are available for download along with a Glossary of tags.      
+
 **How to Use:** Type or paste your text into the text area below, then press Ctrl + Enter. Click the 'Results' button to extract and tag entities in your text data.      
+
 **Usage Limits:** You can request results unlimited times for one (1) month.      
+
 **Supported Languages:** English     
-**Technical issues:** If your connection times out, please refresh the page or reopen the app's URL. 
+
+*Technical issues:** If your connection times out, please refresh the page or reopen the app's URL. 
+
 For any errors or inquiries, please contact us at [email protected]""")
 
 with st.sidebar:
@@ -100,7 +106,7 @@ with st.sidebar:
     st.text("")
     st.text("")
     st.divider()
-    st.subheader("🚀 Ready to build your own NER Web App?", divider="green")
+    st.subheader("🚀 Ready to build your own NER Web App?", divider="orange")
     st.link_button("NER Builder", "https://nlpblogs.com", type="primary")
 
 # --- Comet ML Setup ---
@@ -163,10 +169,6 @@ category_mapping = {
 
 
 
-
-
-
-
 # --- Model Loading ---
 @st.cache_resource
 def load_ner_model():
@@ -212,7 +214,7 @@ if st.button("Results"):
                     experiment.log_parameter("input_text", text)
                     experiment.log_table("predicted_entities", df)
                 
-                st.subheader("Grouped Entities by Category", divider = "green")
+                st.subheader("Grouped Entities by Category", divider = "orange")
                 
                 # Create tabs for each category
                 category_names = sorted(list(category_mapping.keys()))
@@ -240,9 +242,9 @@ if st.button("Results"):
                 st.divider()
                                 
                 # Tree map
-                st.subheader("Tree map", divider = "green")
+                st.subheader("Tree map", divider = "orange")
                 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), paper_bgcolor='#F5FFFA', plot_bgcolor='#F5FFFA')
+                fig_treemap.update_layout(margin=dict(t=50, l=25, r=25, b=25), paper_bgcolor='#FFF8F0', plot_bgcolor='#FFF8F0')
                 st.plotly_chart(fig_treemap)
                 
                 # Pie and Bar charts
@@ -251,26 +253,26 @@ if st.button("Results"):
                 col1, col2 = st.columns(2)
                 
                 with col1:
-                    st.subheader("Pie chart", divider = "green")
+                    st.subheader("Pie chart", divider = "orange")
                     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')
                     fig_pie.update_layout(
-                        paper_bgcolor='#F5FFFA',
-                        plot_bgcolor='#F5FFFA'
+                        paper_bgcolor='#FFF8F0',
+                        plot_bgcolor='#FFF8F0'
                     )
                     st.plotly_chart(fig_pie)
                 
                 with col2:
-                    st.subheader("Bar chart", divider = "green")
+                    st.subheader("Bar chart", divider = "orange")
                     fig_bar = px.bar(grouped_counts, x="count", y="category", color="category", text_auto=True, title='Occurrences of predicted categories')
                     fig_pie.update_layout(
-                        paper_bgcolor='#F5FFFA',
-                        plot_bgcolor='#F5FFFA'
+                        paper_bgcolor='#FFF8F0',
+                        plot_bgcolor='#FFF8F0'
                     )
                     st.plotly_chart(fig_bar)
                 
                 # Most Frequent Entities
-                st.subheader("Most Frequent Entities", divider="green")
+                st.subheader("Most Frequent Entities", divider="orange")
                 word_counts = df['text'].value_counts().reset_index()
                 word_counts.columns = ['Entity', 'Count']
                 repeating_entities = word_counts[word_counts['Count'] > 1]
@@ -278,8 +280,8 @@ if st.button("Results"):
                     st.dataframe(repeating_entities, use_container_width=True)
                     fig_repeating_bar = px.bar(repeating_entities, x='Entity', y='Count', color='Entity')
                     fig_repeating_bar.update_layout(xaxis={'categoryorder': 'total descending'},
-                        paper_bgcolor='#F5FFFA',
-                        plot_bgcolor='#F5FFFA')
+                        paper_bgcolor='#FFF8F0',
+                        plot_bgcolor='#FFF8F0')
                     st.plotly_chart(fig_repeating_bar)
                 else:
                     st.warning("No entities were found that occur more than once.")