Update app.py

app.py

@@ -92,16 +92,6 @@ def predict_category(text):
     category_idx = np.argmax(prediction, axis=1)[0]
     return label_encoder.inverse_transform([category_idx])[0]
 
-# ✅ Category Color Mapping
-category_colors = {
-    "Sports": "#27ae60",  # Green
-    "Politics": "#2980b9",  # Blue
-    "Entertainment": "#8e44ad",  # Purple
-    "Technology": "#e67e22",  # Orange
-    "Business": "#c0392b",  # Red
-    "Default": "#ffffff"  # White
-}
-
 # ✅ Streamlit UI Design
 st.markdown("""
     <style>
@@ -123,6 +113,7 @@ st.markdown("""
         padding: 10px;
     }
     .result-box {
+        background-color: #000000;  /* Black background */
         padding: 25px;
         border-radius: 12px;
         text-align: center;
@@ -131,12 +122,15 @@ st.markdown("""
         font-weight: 900;
         text-shadow: 3px 3px 10px rgba(0, 0, 0, 0.5);
     }
+    .result-text {
+        color: #27ae60; /* Green text */
+    }
     </style>
     """, unsafe_allow_html=True)
 
 # ✅ Page Title
 st.markdown("<div class='title'>🧠 Intelligent News Classifier</div>", unsafe_allow_html=True)
-st.markdown("<div class='subtitle'>Find out what type of news you're reading!.</div>", unsafe_allow_html=True)
+st.markdown("<div class='subtitle'>Find out what type of news you're reading!</div>", unsafe_allow_html=True)
 
 # ✅ User Input
 user_input = st.text_area("Enter text here:", height=150, placeholder="Type your news text here...")
@@ -145,11 +139,10 @@ user_input = st.text_area("Enter text here:", height=150, placeholder="Type your
 if st.button("Analyze 🧐"): 
     if user_input.strip():
         category = predict_category(user_input)
-        color = category_colors.get(category, category_colors["Default"])
         st.markdown(
             f"""
-            <div class='result-box' style='color: {color};'>
-                🗂️ Predicted Category: <strong>{category}</strong>
+            <div class='result-box'>
+                <span class='result-text'>🗂️ Predicted Category: <strong>{category}</strong></span>
             </div>
             """,
             unsafe_allow_html=True