Update app.py

app.py

@@ -4,12 +4,10 @@ import torch
 import numpy as np
 
 def analyze_email(email_body):
-    # Load models (ideally cached, but kept here for simplicity)
     spam_pipeline = pipeline("text-classification", model="cybersectony/phishing-email-detection-distilbert_v2.4.1")
     sentiment_model = AutoModelForSequenceClassification.from_pretrained("ISOM5240GP4/email_sentiment", num_labels=2)
     tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
-    # Step 1: Check if the email is spam
     spam_result = spam_pipeline(email_body)
     spam_label = spam_result[0]["label"]
     spam_confidence = spam_result[0]["score"]
@@ -17,7 +15,6 @@ def analyze_email(email_body):
     if spam_label == "LABEL_1":
         return f"This is a spam email (Confidence: {spam_confidence:.2f}). No follow-up needed."
     else:
-        # Step 2: Analyze sentiment for non-spam emails
         inputs = tokenizer(email_body, padding=True, truncation=True, return_tensors='pt')
         outputs = sentiment_model(**inputs)
         predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
@@ -77,12 +74,41 @@ Sarah
     """
     negative_snippet = "Subject: Issue with Recent Delivery Dear Support, I received my package today, but..."
 
+    # Custom CSS for buttons
+    st.markdown("""
+        <style>
+        /* Style for 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 */
+            border: 1px solid #cccccc;
+            border-radius: 3px;
+        }
+        /* Style for Analyze Email button (larger, colored) */
+        div.stButton > button[kind="primary"] {
+            background-color: #FF5733; /* Orange color */
+            color: white;
+            font-size: 18px; /* Larger text */
+            padding: 12px 24px; /* Larger padding */
+            border: none;
+            border-radius: 5px;
+            display: block;
+            margin-top: 15px;
+        }
+        div.stButton > button[kind="primary"]:hover {
+            background-color: #E74C3C; /* Darker orange on hover */
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
     # Buttons with sample content (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 = ""  # Clear result until Analyze is clicked
+            st.session_state.result = ""
             st.rerun()
     with col2:
         if st.button(positive_snippet, key="positive_sample"):
@@ -95,27 +121,8 @@ Sarah
             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"):
+    # 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: