Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,40 +19,47 @@ st.markdown(
     <style>
     /* Main app background and text color */
     .stApp {
-        background-color: #FFF8F0; /* A very light, creamy orange */
+        background-color: #FFE5E5; /* A very light red */
         color: #000000; /* Black for text */
     }
+
     /* Sidebar background color */
     .css-1d36184 {
-        background-color: #FFC080; /* A soft orange for the sidebar */
-        secondary-background-color: #FFC080;
+        background-color: #FF6B6B; /* A soft red for the sidebar */
+        secondary-background-color: #FF6B6B;
     }
+
     /* Expander background color */
     .streamlit-expanderContent {
-        background-color: #FFF8F0;
+        background-color: #FFE5E5;
     }
+
     /* Expander header background color */
     .streamlit-expanderHeader {
-        background-color: #FFF8F0;
+        background-color: #FFE5E5;
     }
+
     /* Text Area background and text color */
     .stTextArea textarea {
-        background-color: #FFDDAA; /* A light, soft orange */
+        background-color: #FF9999; /* A light red */
         color: #000000; /* Black for text */
     }
+
     /* Button background and text color */
     .stButton > button {
-        background-color: #FFDDAA;
+        background-color: #FF9999;
         color: #000000;
     }
+
     /* Warning box background and text color */
     .stAlert.st-warning {
-        background-color: #FFBB88; /* A slightly darker orange for warnings */
+        background-color: #FF4D4D; /* A slightly darker red for warnings */
         color: #000000;
     }
+
     /* Success box background and text color */
     .stAlert.st-success {
-        background-color: #FFBB88; /* A slightly darker orange for success boxes */
+        background-color: #FF4D4D; /* A slightly darker red for success boxes */
         color: #000000;
     }
     </style>
@@ -63,24 +70,27 @@ st.markdown(
 
 
 
-
-
-
 # --- Page Configuration and UI Elements ---
 st.set_page_config(layout="wide", page_title="Named Entity Recognition App")
-st.subheader("Uncover", divider="orange")
+st.subheader("Uncover", divider="red")
 st.link_button("by nlpblogs", "https://nlpblogs.com", type="tertiary")
 expander = st.expander("**Important notes**")
-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"
+expander.write("""**Named Entities:** This Uncover predicts twenty-eight (28) labels: "Names", "Aliases", "Identifiers", "Roles", "Government_agencies", "Businesses", "Criminal_groups", "Financial_institutions", "Addresses", "Geographic_coordinates", "Landmarks", "Jurisdictions", "Dates", "Timestamps", "Time_ranges", "Weapons", "Vehicles", "Financial_information", "Evidence", "Relationships", "Demographics", "Biometrics", "Psychological_states", "Software_types", "Hardware_components", "Equipment", "Events", "Activities"
+
 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. 
+
 For any errors or inquiries, please contact us at [email protected]""")
 
 with st.sidebar:
-    st.write("Use the following code to embed the Business Core web app on your website. Feel free to adjust the width and height values to fit your page.")
+    st.write("Use the following code to embed the Uncover web app on your website. Feel free to adjust the width and height values to fit your page.")
     code = '''
     <iframe
 	src="https://aiecosystem-business-core.hf.space"
@@ -93,7 +103,7 @@ with st.sidebar:
     st.text("")
     st.text("")
     st.divider()
-    st.subheader("🚀 Ready to build your own NER Web App?", divider="orange")
+    st.subheader("🚀 Ready to build your own NER Web App?", divider="red")
     st.link_button("NER Builder", "https://nlpblogs.com", type="primary")
 
 # --- Comet ML Setup ---
@@ -207,7 +217,7 @@ if st.button("Results"):
                     experiment.log_parameter("input_text", text)
                     experiment.log_table("predicted_entities", df)
                 
-                st.subheader("Grouped Entities by Category", divider = "orange")
+                st.subheader("Grouped Entities by Category", divider = "red")
                 
                 # Create tabs for each category
                 category_names = sorted(list(category_mapping.keys()))
@@ -235,9 +245,10 @@ if st.button("Results"):
                 st.divider()
                                 
                 # Tree map
-                st.subheader("Tree map", divider = "orange")
+                st.subheader("Tree map", divider = "red")
                 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='#FFF8F0', plot_bgcolor='#FFF8F0')
+                fig_treemap.update_layout(margin=dict(t=50, l=25, r=25, b=25), paper_bgcolor='#FFE5E5', plot_bgcolor='#FFE5E5')
+                
                 st.plotly_chart(fig_treemap)
                 
                 # Pie and Bar charts
@@ -246,26 +257,26 @@ if st.button("Results"):
                 col1, col2 = st.columns(2)
                 
                 with col1:
-                    st.subheader("Pie chart", divider = "orange")
+                    st.subheader("Pie chart", divider = "red")
                     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='#FFF8F0',
-                        plot_bgcolor='#FFF8F0'
+                        paper_bgcolor='#FFE5E5',
+                        plot_bgcolor='#FFE5E5'
                     )
                     st.plotly_chart(fig_pie)
                 
                 with col2:
-                    st.subheader("Bar chart", divider = "orange")
+                    st.subheader("Bar chart", divider = "red")
                     fig_bar = px.bar(grouped_counts, x="count", y="category", color="category", text_auto=True, title='Occurrences of predicted categories')
                     fig_bar.update_layout(
-                        paper_bgcolor='#FFF8F0',
-                        plot_bgcolor='#FFF8F0'
+                        paper_bgcolor='#FFE5E5',
+                        plot_bgcolor='#FFE5E5'
                     )
                     st.plotly_chart(fig_bar)
                 
                 # Most Frequent Entities
-                st.subheader("Most Frequent Entities", divider="orange")
+                st.subheader("Most Frequent Entities", divider="red")
                 word_counts = df['text'].value_counts().reset_index()
                 word_counts.columns = ['Entity', 'Count']
                 repeating_entities = word_counts[word_counts['Count'] > 1]
@@ -273,8 +284,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='#FFF8F0',
-                        plot_bgcolor='#FFF8F0')
+                        paper_bgcolor='#FFE5E5',
+                        plot_bgcolor='#FFE5E5')
                     st.plotly_chart(fig_repeating_bar)
                 else:
                     st.warning("No entities were found that occur more than once.")