Update app.py

app.py

@@ -35,6 +35,7 @@ st.markdown("""
         border-radius: 20px;
         font-weight: bold;
         margin: 0.2rem;
+        display: inline-block;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -68,7 +69,7 @@ if classifier is None:
     st.error("Could not load the model. Please check if the model exists on Hugging Face.")
     st.stop()
 
-# Label colors
+# Label colors and descriptions
 label_colors = {
     "positive": "🟢",
     "negative": "🔴", 
@@ -83,6 +84,13 @@ label_descriptions = {
     "uncertainty": "Uncertain/ambiguous content"
 }
 
+badge_colors = {
+    "positive": "#4CAF50",
+    "negative": "#F44336", 
+    "litigious": "#2196F3",
+    "uncertainty": "#FFC107"
+}
+
 # Input section
 st.markdown("---")
 st.markdown("## 📝 Enter Tweet Text to Analyze")
@@ -123,18 +131,15 @@ if st.button("🔍 Analyze Tweet", type="primary", use_container_width=True):
                 st.markdown("## 📊 Analysis Results")
                 
                 # Result box
-                st.markdown(f'<div class="result-box">', unsafe_allow_html=True)
+                st.markdown('<div class="result-box">', unsafe_allow_html=True)
                 
                 # Label and confidence
                 col1, col2 = st.columns([1, 2])
                 with col1:
                     st.markdown(f"### {label_colors.get(label, '⚪')} **Prediction:**")
-                    st.markdown(f'<span class="label-badge" style="background-color: {{
-                        "positive": "#4CAF50",
-                        "negative": "#F44336", 
-                        "litigious": "#2196F3",
-                        "uncertainty": "#FFC107"
-                    }.get(label, "#9E9E9E")}}; color: white;">{label.upper()}</span>', unsafe_allow_html=True)
+                    # FIXED: Simplified the badge color assignment
+                    color = badge_colors.get(label, "#9E9E9E")
+                    st.markdown(f'<span class="label-badge" style="background-color: {color}; color: white;">{label.upper()}</span>', unsafe_allow_html=True)
                     
                 with col2:
                     st.markdown(f"### 📈 **Confidence:** {confidence:.1%}")