Update app.py

app.py

@@ -13,17 +13,16 @@ 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
 
-    # Updated analyzer returns summary (dict) and details (dict)
-    summary, details = analyze(file_path)
+    results = analyze(file_path)
 
-    # 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")
+    # Extract basic verdict info
+    attack_score = results[0]
+    attack_type = results[1]
+    verdict = results[2]
 
-    # Cap score at 100
-    score_val = int("".join(filter(str.isdigit, attack_score)))
-    if score_val > 100:
+    # Correct handling of attack_score (already int)
+    score_val = int(attack_score)
+    if score_val > 100:  
         score_val = 100
 
     # Verdict color
@@ -36,11 +35,19 @@ def analyze_email(file_path, export_pdf=False):
         verdict_color = "#f1c40f"
 
     # Extract sections
-    tags = details.get("Attack Analysis Tags", [])
-    findings = details.get("Detailed Findings", [])
-    highlighted_body = details.get("Highlighted Body", [])
+    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
 
-    # HTML Report
+    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 = f"""
     <div style='font-family: Arial, sans-serif; padding: 20px; max-width: 800px; margin:auto;'>
       <h2 style="color:#2c3e50;">📧 Email Security Report</h2>
@@ -52,7 +59,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}
+        {verdict.replace('Final Verdict:', '').strip()}
       </span></p>
       <details style="margin-top:15px;">
         <summary>🔖 Attack Analysis Tags ({len(tags)})</summary>