Update app.py

app.py

@@ -45,10 +45,9 @@ def main():
         st.session_state.result = ""
 
     # Text area for email input
-    email_body = st.text_area("Email Body", value=st.session_state.email_body, height=200, key="email_input",
-                              on_change=lambda: st.session_state.update({"result": analyze_email(st.session_state.email_input)}))
+    email_body = st.text_area("Email Body", value=st.session_state.email_body, height=200, key="email_input")
 
-    # Sample emails (stored as variables, not button labels)
+    # Sample emails (shortened snippets for button labels)
     sample_spam = """
 Subject: Urgent: Verify Your Account Now!
 Dear Customer,
@@ -58,6 +57,7 @@ Failure to verify within 24 hours will result in your account being locked. This
 Best regards,
 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!
@@ -66,6 +66,7 @@ I just wanted to let you know that the project is progressing wonderfully! Every
 Best,
 Alex
     """
+    positive_snippet = "Subject: Great News About Your Project! Hi Team, I just wanted to let you know..."
 
     sample_not_spam_negative = """
 Subject: Issue with Recent Delivery
@@ -74,25 +75,52 @@ I received my package today, but it was damaged, and two items were missing. Thi
 Thanks,
 Sarah
     """
+    negative_snippet = "Subject: Issue with Recent Delivery Dear Support, I received my package today, but..."
 
-    # Buttons with concise labels (in columns)
+    # Buttons with sample content (in columns)
     col1, col2, col3 = st.columns(3)
     with col1:
-        if st.button("Spam Sample"):
+        if st.button(spam_snippet, key="spam_sample"):
             st.session_state.email_body = sample_spam
-            st.session_state.result = analyze_email(sample_spam)
+            st.session_state.result = ""  # Clear result until Analyze is clicked
             st.rerun()
     with col2:
-        if st.button("Positive Sample"):
+        if st.button(positive_snippet, key="positive_sample"):
             st.session_state.email_body = sample_not_spam_positive
-            st.session_state.result = analyze_email(sample_not_spam_positive)
+            st.session_state.result = ""
             st.rerun()
     with col3:
-        if st.button("Negative Sample"):
+        if st.button(negative_snippet, key="negative_sample"):
             st.session_state.email_body = sample_not_spam_negative
-            st.session_state.result = analyze_email(sample_not_spam_negative)
+            st.session_state.result = ""
             st.rerun()
 
+    # Custom styled Analyze button
+    st.markdown("""
+        <style>
+        .analyze-button {
+            background-color: #4CAF50; /* Green */
+            color: white;
+            padding: 10px 20px;
+            border: none;
+            border-radius: 5px;
+            cursor: pointer;
+            font-size: 16px;
+            display: block;
+            margin-top: 10px;
+        }
+        .analyze-button:hover {
+            background-color: #45a049;
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
+    if st.button("Analyze Email", key="analyze", help="Click to analyze the email"):
+        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."
+
     # Display result
     if st.session_state.result:
         st.write(st.session_state.result)