Update app.py

app.py

@@ -13,7 +13,7 @@ def analyze_email(email_body):
     spam_confidence = spam_result[0]["score"]
 
     if spam_label == "LABEL_1":
-        return f"This is a spam email (Confidence: {spam_confidence:.2f}). No follow-up needed."
+        return "spam", f"This is a spam email (Confidence: {spam_confidence:.2f}). No follow-up needed."
     else:
         inputs = tokenizer(email_body, padding=True, truncation=True, return_tensors='pt')
         outputs = sentiment_model(**inputs)
@@ -24,12 +24,12 @@ def analyze_email(email_body):
         sentiment = "Positive" if sentiment_index == 1 else "Negative"
 
         if sentiment == "Positive":
-            return (f"This email is not spam (Confidence: {spam_confidence:.2f}).\n"
-                    f"Sentiment: {sentiment} (Confidence: {sentiment_confidence:.2f}). No follow-up needed.")
+            return "positive", (f"This email is not spam (Confidence: {spam_confidence:.2f}).\n"
+                                f"Sentiment: {sentiment} (Confidence: {sentiment_confidence:.2f}). No follow-up needed.")
         else:
-            return (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.**")
+            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.**")
 
 def main():
     st.title("Email Analysis Tool")
@@ -40,6 +40,17 @@ def main():
         st.session_state.email_body = ""
     if "result" not in st.session_state:
         st.session_state.result = ""
+    if "result_type" not in st.session_state:
+        st.session_state.result_type = ""
+
+    # Collapsible instructions
+    with st.expander("How to Use", expanded=False):
+        st.write("""
+        - Type or paste an email into the text box.
+        - Alternatively, click one of the sample buttons to load a predefined email.
+        - Press 'Analyze Email' to check if it’s spam and analyze its sentiment.
+        - Use 'Clear' to reset the input and result.
+        """)
 
     # Text area for email input
     email_body = st.text_area("Email Body", value=st.session_state.email_body, height=200, key="email_input")
@@ -74,63 +85,115 @@ Sarah
     """
     negative_snippet = "Subject: Issue with Recent Delivery Dear Support, I received my package today, but..."
 
-    # Custom CSS for buttons
+    # Custom CSS for buttons and result boxes
     st.markdown("""
         <style>
-        /* Style for sample buttons (smaller text) */
+        /* Sample buttons (smaller text) */
         div.stButton > button[kind="secondary"] {
-            font-size: 12px; /* Smaller text */
-            padding: 5px 10px; /* Smaller padding */
-            background-color: #f0f0f0; /* Light gray background */
-            color: #333333; /* Darker text */
+            font-size: 12px;
+            padding: 5px 10px;
+            background-color: #f0f0f0;
+            color: #333333;
             border: 1px solid #cccccc;
             border-radius: 3px;
         }
-        /* Style for Analyze Email button (larger, colored) */
+        /* Analyze Email button (larger, orange) */
         div.stButton > button[kind="primary"] {
-            background-color: #FF5733; /* Orange color */
+            background-color: #FF5733;
             color: white;
-            font-size: 18px; /* Larger text */
-            padding: 12px 24px; /* Larger padding */
+            font-size: 18px;
+            padding: 12px 24px;
             border: none;
             border-radius: 5px;
-            display: block;
-            margin-top: 15px;
+            margin-right: 10px;
         }
         div.stButton > button[kind="primary"]:hover {
-            background-color: #E74C3C; /* Darker orange on hover */
+            background-color: #E74C3C;
+        }
+        /* Clear button (gray) */
+        div.stButton > button[kind="secondary"][key="clear"] {
+            background-color: #d3d3d3;
+            color: #333333;
+            font-size: 16px;
+            padding: 10px 20px;
+            border: none;
+            border-radius: 5px;
+        }
+        div.stButton > button[kind="secondary"][key="clear"]:hover {
+            background-color: #b0b0b0;
+        }
+        /* Result boxes */
+        .spam-result {
+            background-color: #ffcccc;
+            padding: 10px;
+            border-radius: 5px;
+            border: 1px solid #ff9999;
+        }
+        .positive-result {
+            background-color: #ccffcc;
+            padding: 10px;
+            border-radius: 5px;
+            border: 1px solid #99cc99;
+        }
+        .negative-result {
+            background-color: #fff3cc;
+            padding: 10px;
+            border-radius: 5px;
+            border: 1px solid #ffcc66;
         }
         </style>
     """, unsafe_allow_html=True)
 
-    # Buttons with sample content (in columns)
+    # Sample buttons (in columns)
     col1, col2, col3 = st.columns(3)
     with col1:
         if st.button(spam_snippet, key="spam_sample"):
             st.session_state.email_body = sample_spam
             st.session_state.result = ""
+            st.session_state.result_type = ""
             st.rerun()
     with col2:
         if st.button(positive_snippet, key="positive_sample"):
             st.session_state.email_body = sample_not_spam_positive
             st.session_state.result = ""
+            st.session_state.result_type = ""
             st.rerun()
     with col3:
         if st.button(negative_snippet, key="negative_sample"):
             st.session_state.email_body = sample_not_spam_negative
             st.session_state.result = ""
+            st.session_state.result_type = ""
             st.rerun()
 
-    # Analyze Email button (distinct style)
-    if st.button("Analyze Email", key="analyze", type="primary"):
-        if email_body:
-            st.session_state.result = analyze_email(email_body)
-        else:
-            st.session_state.result = "Please enter an email body or select a sample to analyze."
+    # Analyze and Clear buttons (in a row)
+    col_analyze, col_clear = st.columns([1, 1])
+    with col_analyze:
+        if st.button("Analyze Email", key="analyze", type="primary"):
+            if email_body:
+                with st.spinner("Analyzing email..."):
+                    result_type, result = analyze_email(email_body)
+                    st.session_state.result = result
+                    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
+    # Display result with styled box
     if st.session_state.result:
-        st.write(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()