Update app.py

app.py

@@ -13,15 +13,17 @@ def analyze_email(file_path, export_pdf=False):
     if file_path is None:
         return "<p style='color:red;'>Please upload a .eml file to analyze.</p>", None
 
-    results = analyze(file_path)
+    # Updated analyzer returns summary (dict) and details (dict)
+    summary, details = analyze(file_path)
 
-    # Extract basic verdict info
-    attack_score = results[0]
-    attack_type = results[1]
-    verdict = results[2]
+    # Extract score, attack type, verdict
+    attack_score = summary.get("Attack Score", "0")
+    attack_type = summary.get("Attack Type", "Unknown")
+    verdict = summary.get("Final Verdict", "Safe")
 
+    # Cap score at 100
     score_val = int("".join(filter(str.isdigit, attack_score)))
-    if score_val > 100:  
+    if score_val > 100:
         score_val = 100
 
     # Verdict color
@@ -34,19 +36,11 @@ def analyze_email(file_path, export_pdf=False):
         verdict_color = "#f1c40f"
 
     # Extract sections
-    try:
-        tags_index = results.index("---- Attack Analysis Tags ----")
-        findings_index = results.index("---- Detailed Findings ----")
-        body_index = results.index("---- Highlighted Body ----")
-    except ValueError:
-        html = "<pre>" + "\n".join(results) + "</pre>"
-        return html, None
+    tags = details.get("Attack Analysis Tags", [])
+    findings = details.get("Detailed Findings", [])
+    highlighted_body = details.get("Highlighted Body", [])
 
-    tags = results[tags_index + 1 : findings_index]
-    findings = results[findings_index + 1 : body_index]
-    highlighted_body = results[body_index + 1 :]
-
-    # HTML Report with collapsible sections
+    # HTML Report
     html = f"""
     <div style='font-family: Arial, sans-serif; padding: 20px; max-width: 800px; margin:auto;'>
       <h2 style="color:#2c3e50;">📧 Email Security Report</h2>
@@ -58,7 +52,7 @@ def analyze_email(file_path, export_pdf=False):
       </div>
       <p><b>Attack Type:</b> {attack_type}</p>
       <p><b>Final Verdict:</b> <span style="background:{verdict_color}; color:white; padding:4px 10px; border-radius:6px;">
-        {verdict.replace('Final Verdict:', '').strip()}
+        {verdict}
       </span></p>
       <details style="margin-top:15px;">
         <summary>🔖 Attack Analysis Tags ({len(tags)})</summary>