Update app.py

app.py

@@ -59,15 +59,31 @@ if st.button("Classify Feedback"):
                 if score >= CONFIDENCE_THRESHOLD:
                     # Perform sentiment analysis on the sentence
                     sentiment_result = sentiment_analyzer(sentence)
-                    sentiment_label = sentiment_result[0]["label"]
+                    raw_label = sentiment_result[0]["label"]
                     sentiment_score = round(sentiment_result[0]["score"], 4)
 
+                    # Map the raw label to NEGATIVE or POSITIVE
+                    if raw_label == "LABEL_0":
+                        sentiment_label = "NEGATIVE"
+                        sentiment_icon = "👎"  # Thumbs-down icon for negative
+                        sentiment_color = "red"  # Red color for negative
+                    elif raw_label == "LABEL_1":
+                        sentiment_label = "POSITIVE"
+                        sentiment_icon = "👍"  # Thumbs-up icon for positive
+                        sentiment_color = "green"  # Green color for positive
+                    else:
+                        sentiment_label = raw_label  # Fallback in case of unexpected label
+                        sentiment_icon = "❓"  # Question mark for unknown
+                        sentiment_color = "gray"  # Gray color for unknown
+
                     # Store the result for the category
                     category_results[label].append({
                         "sentence": sentence,
                         "confidence": round(score, 4),
                         "sentiment": sentiment_label,
-                        "sentiment_score": sentiment_score
+                        "sentiment_score": sentiment_score,
+                        "sentiment_icon": sentiment_icon,
+                        "sentiment_color": sentiment_color
                     })
 
         # Check if there are any relevant categories
@@ -81,7 +97,13 @@ if st.button("Classify Feedback"):
                 for result in results:
                     st.write(f"- **Sentence**: {result['sentence']}")
                     st.write(f"  - Confidence: {result['confidence']}")
-                    st.write(f"  - Sentiment: {result['sentiment']} (Score: {result['sentiment_score']})")
+                    # Use st.markdown with HTML to display the sentiment with icon and color
+                    st.markdown(
+                        f"  - Sentiment: {result['sentiment_icon']} "
+                        f"<span style='color:{result['sentiment_color']}'>{result['sentiment']}</span> "
+                        f"(Score: {result['sentiment_score']})",
+                        unsafe_allow_html=True
+                    )
                 st.write("")  # Add a blank line for readability
 
         if not found_categories: