Update analyze_email_main.py

analyze_email_main.py

@@ -4,16 +4,23 @@ from body_analyzer import analyze_body
 from url_analyzer import analyze_urls
 
 def analyze(file_path):
+    # --- Parse the email ---
     headers, body, urls = parse_email(file_path)
 
+    # --- Analyze headers ---
     header_findings, header_score = analyze_headers(headers)
-    body_findings, body_score, highlighted_body = analyze_body(body)
+
+    # --- Analyze body (updated analyzer with improvements) ---
+    body_findings, body_score, highlighted_body, body_verdict = analyze_body(body)
+
+    # --- Analyze URLs ---
     url_findings, url_score = analyze_urls(urls)
 
+    # --- Calculate total score ---
     total_score = header_score + body_score + (url_score * 1.5)
     total_score = min(total_score, 100)
 
-    # --- Final verdict ---
+    # --- Determine final verdict ---
     if total_score >= 70:
         verdict = "🚨 Malicious"
     elif 50 <= total_score < 70:
@@ -25,6 +32,7 @@ def analyze(file_path):
 
     # --- Determine attack type ---
     body_lower = body.lower()
+    attack_type = "General Phishing"  # default
     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"]):
@@ -35,8 +43,6 @@ def analyze(file_path):
         attack_type = "Spam / Marketing"
     elif verdict == "✅ Safe":
         attack_type = "Benign / Normal Email"
-    else:
-        attack_type = "General Phishing"
 
     # --- Collect tags ---
     tags = []
@@ -51,7 +57,7 @@ def analyze(file_path):
         if "spam" in f_lower or "marketing" in f_lower:
             tags.append("Spam Tone")
 
-    # --- Compact report ---
+    # --- Summary report ---
     summary = {
         "Final Verdict": verdict,
         "Attack Type": attack_type,
@@ -59,6 +65,7 @@ def analyze(file_path):
         "Main Tags": ", ".join(sorted(set(tags))) if tags else "No special tags"
     }
 
+    # --- Detailed findings ---
     details = {
         "Header Findings": header_findings,
         "Body Findings": body_findings,