Update app.py

app.py

@@ -29,11 +29,11 @@ def analyze_email(email_body):
         else:
             return "negative", (f"This email is not spam (Confidence: {spam_confidence:.2f}).\n"
                                 f"Sentiment: {sentiment} (Confidence: {sentiment_confidence:.2f}).\n"
-                                "**This email needs follow-up as it is not spam and has negative sentiment.**")
+                                "**Need to Follow-Up**: This email is not spam and has negative sentiment.")
 
 def main():
-    st.title("Email Analysis Tool")
-    st.write("Enter an email body below or click a sample to analyze its spam status and sentiment.")
+    st.title("EmailSentry")
+    st.write("Aims to perform analysis on incoming emails and to determine whether there is urgency or higher priority for the company to follow-up.")
 
     # Initialize session state
     if "email_body" not in st.session_state:
@@ -68,13 +68,13 @@ The Security Team
     spam_snippet = "Subject: Urgent: Verify Your Account Now! Dear Customer, We have detected unusual activity..."
 
     sample_not_spam_positive = """
-Subject: Great News About Your Project!
-Hi Team,
-I just wanted to let you know that the project is progressing wonderfully! Everyone’s efforts are paying off, and we’re ahead of schedule. Keep up the fantastic work!
-Best,
-Alex
+Subject: Great Experience with HKTV Mall!
+Dear HKTV Mall Team,
+I just received my order #HKTV-123456, and I’m really impressed with the fast delivery and quality of the products! Thanks for making my shopping experience so smooth. Keep up the great work!
+Best regards,
+Emily
     """
-    positive_snippet = "Subject: Great News About Your Project! Hi Team, I just wanted to let you know..."
+    positive_snippet = "Subject: Great Experience with HKTV Mall! Dear HKTV Mall Team, I just received my order..."
 
     sample_not_spam_negative = """
 Subject: Issue with Recent Delivery
@@ -176,24 +176,4 @@ Sarah
                     st.session_state.result_type = result_type
             else:
                 st.session_state.result = "Please enter an email body or select a sample to analyze."
-                st.session_state.result_type = ""
-    with col_clear:
-        if st.button("Clear", key="clear"):
-            st.session_state.email_body = ""
-            st.session_state.result = ""
-            st.session_state.result_type = ""
-            st.rerun()
-
-    # Display result with styled box
-    if st.session_state.result:
-        if st.session_state.result_type == "spam":
-            st.markdown(f'<div class="spam-result">{st.session_state.result}</div>', unsafe_allow_html=True)
-        elif st.session_state.result_type == "positive":
-            st.markdown(f'<div class="positive-result">{st.session_state.result}</div>', unsafe_allow_html=True)
-        elif st.session_state.result_type == "negative":
-            st.markdown(f'<div class="negative-result">{st.session_state.result}</div>', unsafe_allow_html=True)
-        else:
-            st.write(st.session_state.result)  # For error messages
-
-if __name__ == "__main__":
-    main()
+                st