Update app.py

app.py

@@ -10,11 +10,13 @@ transcript = st.text_area("Enter the transcript:", height=200)
 if st.button("Analyze"):
     st.subheader("Sentiment Analysis")
     sentiment = sentiment_model(transcript)[0]['label']
-    st.write(sentiment)
+    sentiment_color = "green" if sentiment == "POSITIVE" else "red"
+    st.markdown(f'<span style="color:{sentiment_color}">{sentiment}</span>', unsafe_allow_html=True)
 
     st.subheader("Increase/Decrease Prediction")
     increase_decrease = increase_decrease_model(transcript)[0]['label']
-    st.write(increase_decrease)
+    increase_decrease_color = "green" if increase_decrease == "INCREASE" else "red"
+    st.markdown(f'<span style="color:{increase_decrease_color}">{increase_decrease}</span>', unsafe_allow_html=True)
 
     st.subheader("NER Metrics")
     ner_result = ner_model(transcript)
@@ -23,3 +25,12 @@ if st.button("Analyze"):
         st.write(f"Revenue: {revenue}")
     else:
         st.write("Revenue not found.")
+
+    show_details = st.checkbox("Show Detailed Predictions")
+    if show_details:
+        st.subheader("Detailed Predictions")
+        st.json({
+            "Sentiment Analysis": sentiment,
+            "Increase/Decrease Prediction": increase_decrease,
+            "NER Metrics": ner_result
+        })