Update app.py

app.py

@@ -12,10 +12,58 @@ def main():
     tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
     st.title("Email Analysis Tool")
-    st.write("Enter an email body below to check if it's spam and analyze its sentiment.")
+    st.write("Enter an email body below or select a sample to analyze its spam status and sentiment.")
 
-    email_body = st.text_area("Email Body", height=200)
+    # Initialize session state for the text area
+    if "email_body" not in st.session_state:
+        st.session_state.email_body = ""
 
+    # Text area for email input
+    email_body = st.text_area("Email Body", value=st.session_state.email_body, height=200, key="email_input")
+
+    # Sample emails
+    sample_spam = """
+Subject: Urgent: Verify Your Account Now!
+Dear Customer,
+We have detected unusual activity on your account. To prevent suspension, please verify your login details immediately by clicking the link below:
+[Click Here to Verify](http://totally-legit-site.com/verify)
+Failure to verify within 24 hours will result in your account being locked. This is for your security.
+Best regards,
+The Security Team
+    """
+
+    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
+    """
+
+    sample_not_spam_negative = """
+Subject: Issue with Recent Delivery
+Dear Support,
+I received my package today, but it was damaged, and two items were missing. This is really frustrating—please let me know how we can resolve this as soon as possible.
+Thanks,
+Sarah
+    """
+
+    # Buttons for sample emails (in columns for better layout)
+    col1, col2, col3 = st.columns(3)
+    with col1:
+        if st.button("Spam Email"):
+            st.session_state.email_body = sample_spam
+            st.rerun()  # Rerun to update the text area
+    with col2:
+        if st.button("Not Spam, Positive"):
+            st.session_state.email_body = sample_not_spam_positive
+            st.rerun()
+    with col3:
+        if st.button("Not Spam, Negative"):
+            st.session_state.email_body = sample_not_spam_negative
+            st.rerun()
+
+    # Button to trigger analysis
     if st.button("Analyze Email"):
         if email_body:
             # Step 1: Check if the email is spam
@@ -31,21 +79,4 @@ def main():
                 inputs = tokenizer(email_body, padding=True, truncation=True, return_tensors='pt')
                 outputs = sentiment_model(**inputs)
                 predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
-                predictions = predictions.cpu().detach().numpy()
-                sentiment_index = np.argmax(predictions)
-                sentiment_confidence = predictions[0][sentiment_index]
-
-                sentiment = "Positive" if sentiment_index == 1 else "Negative"
-
-                if sentiment == "Positive":
-                    st.write(f"This email is not spam (Confidence: {spam_confidence:.2f}).")
-                    st.write(f"Sentiment: {sentiment} (Confidence: {sentiment_confidence:.2f}). No follow-up needed.")
-                else:
-                    st.write(f"This email is not spam (Confidence: {spam_confidence:.2f}).")
-                    st.write(f"Sentiment: {sentiment} (Confidence: {sentiment_confidence:.2f}).")
-                    st.write("**This email needs follow-up as it is not spam and has negative sentiment.**")
-        else:
-            st.write("Please enter an email body to analyze.")
-
-if __name__ == "__main__":
-    main()
+                predictions = predictions.cpu().detach