Update analyze_email_main.py

analyze_email_main.py

@@ -6,14 +6,15 @@ from url_analyzer import analyze_urls
 def analyze(file_path):
     headers, body, urls = parse_email(file_path)
 
-    # Run individual analyzers (each returns findings + score)
     header_findings, header_score = analyze_headers(headers)
-    body_findings, body_score = analyze_body(body)
+    body_findings, body_score, highlighted_body = analyze_body(body)
     url_findings, url_score = analyze_urls(urls)
 
     total_score = header_score + body_score + url_score
+    if total_score > 100:
+        total_score = 100
 
-    # --- Determine verdict ---
+    # --- Final verdict ---
     if total_score >= 70:
         verdict = "🚨 Malicious"
     elif 50 <= total_score < 70:
@@ -23,13 +24,18 @@ def analyze(file_path):
     else:
         verdict = "✅ Safe"
 
-    # --- Attack Type (basic heuristic) ---
-    if "invoice" in body.lower() or "payment" in body.lower():
-        attack_type = "Invoice/Payment Fraud"
-    elif "verify" in body.lower() or "password" in body.lower():
-        attack_type = "Credential Harvesting"
-    elif verdict == "📩 Spam":
+    # --- Attack Type ---
+    body_lower = body.lower()
+    if any(word in body_lower for word in ["invoice", "payment", "wire transfer", "bank details"]):
+        attack_type = "Invoice/Payment Fraud (BEC)"
+    elif any(word in body_lower for word in ["password", "verify", "account", "login", "credentials"]):
+        attack_type = "Credential Harvesting (Phishing)"
+    elif any("reply-to mismatch" in f.lower() for f in header_findings):
+        attack_type = "Business Email Compromise (BEC)"
+    elif any("spam" in f.lower() for f in body_findings):
         attack_type = "Spam / Marketing"
+    elif verdict == "✅ Safe":
+        attack_type = "Benign / Normal Email"
     else:
         attack_type = "General Phishing"
 
@@ -56,6 +62,8 @@ def analyze(file_path):
     ]
 
     report.extend(header_findings + body_findings + url_findings)
+    report.append("---- Highlighted Body ----")
+    report.append(highlighted_body)
 
     return report