Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,46 +14,7 @@ from typing import Optional
 from gliner import GLiNER
 from comet_ml import Experiment
 
-st.markdown(
-    """
-    <style>
-    /* Main app background and text color */
-    .stApp {
-        background-color: #E8F5E9; /* A very light green */
-        color: #1B5E20; /* Dark green for the text */
-    }
-        /* Sidebar background color */
-    .css-1d36184 {
-        background-color: #A5D6A7; /* A medium light green */
-        secondary-background-color: #A5D6A7;
-    }
-        /* Expander background color and header */
-    .streamlit-expanderContent, .streamlit-expanderHeader {
-        background-color: #E8F5E9;
-    }
-        /* Text Area background and text color */
-    .stTextArea textarea {
-        background-color: #81C784; /* A slightly darker medium green */
-        color: #1B5E20; /* Dark green for text */
-    }
-        /* Button background and text color */
-    .stButton > button {
-        background-color: #81C784;
-        color: #1B5E20;
-    }
-        /* Warning box background and text color */
-    .stAlert.st-warning {
-        background-color: #66BB6A; /* A medium-dark green for the warning box */
-        color: #1B5E20;
-    }
-        /* Success box background and text color */
-    .stAlert.st-success {
-        background-color: #66BB6A; /* A medium-dark green for the success box */
-        color: #1B5E20;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True)
+
 
 # --- Page Configuration and UI Elements ---
 st.set_page_config(layout="wide", page_title="Named Entity Recognition App")
@@ -118,7 +79,6 @@ def load_ner_model():
     except Exception as e:
         st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
         st.stop()
-
 model = load_ner_model()
 # Flatten the mapping to a single dictionary
 reverse_category_mapping = {label: category for category, label_list in category_mapping.items() for label in label_list}
@@ -250,7 +210,7 @@ if st.button("Results"):
                 if comet_initialized:
                     experiment.log_figure(figure=fig_treemap, figure_name="entity_treemap_categories")
                     experiment.end()
-                
+            
                 # Corrected placement for time calculation and display
                 end_time = time.time()
                 elapsed_time = end_time - start_time