Update analyze_email_main.py

analyze_email_main.py

@@ -4,19 +4,14 @@ from body_analyzer import analyze_body
 from url_analyzer import analyze_urls
 
 def analyze(file_path):
-    # --- Parse email ---
     headers, body, urls = parse_email(file_path)
 
-    # --- Analyze different components ---
     header_findings, header_score = analyze_headers(headers)
     body_findings, body_score, highlighted_body = analyze_body(body)
     url_findings, url_score = analyze_urls(urls)
 
-    # --- Weighted total score ---
-    # Give extra weight to URLs and phishing indicators
     total_score = header_score + body_score + (url_score * 1.5)
-    if total_score > 100:
-        total_score = 100
+    total_score = min(total_score, 100)
 
     # --- Final verdict ---
     if total_score >= 70:
@@ -43,7 +38,7 @@ def analyze(file_path):
     else:
         attack_type = "General Phishing"
 
-    # --- Collect tags for detailed analysis ---
+    # --- Collect tags ---
     tags = []
     for finding in header_findings + body_findings + url_findings:
         f_lower = finding.lower()
@@ -56,24 +51,37 @@ def analyze(file_path):
         if "spam" in f_lower or "marketing" in f_lower:
             tags.append("Spam Tone")
 
-    # --- Build final report ---
-    report = [
-        f"Attack Score: {total_score}",
-        f"Attack Type: {attack_type}",
-        f"Final Verdict: {verdict}",
-        "---- Attack Analysis Tags ----",
-        ", ".join(sorted(set(tags))) if tags else "No special tags",
-        "---- Detailed Findings ----",
-    ]
-    report.extend(header_findings + body_findings + url_findings)
-    report.append("---- Highlighted Body ----")
-    report.append(highlighted_body)
+    # --- Compact report ---
+    summary = {
+        "Final Verdict": verdict,
+        "Attack Type": attack_type,
+        "Attack Score": total_score,
+        "Main Tags": ", ".join(sorted(set(tags))) if tags else "No special tags"
+    }
 
-    return report
+    details = {
+        "Header Findings": header_findings,
+        "Body Findings": body_findings,
+        "URL Findings": url_findings,
+        "Highlighted Body": highlighted_body
+    }
 
-# --- For testing locally ---
+    return summary, details
+
+# --- Local testing ---
 if __name__ == "__main__":
     file_path = "sample.eml"
-    findings = analyze(file_path)
-    for f in findings:
-        print(f)
+    summary, details = analyze(file_path)
+    
+    print("==== SUMMARY ====")
+    for k, v in summary.items():
+        print(f"{k}: {v}")
+    
+    print("\n==== DETAILS ====")
+    for section, findings in details.items():
+        print(f"\n-- {section} --")
+        if isinstance(findings, list):
+            for f in findings:
+                print(f)
+        else:
+            print(findings)