Update app.py

app.py

@@ -15,22 +15,17 @@ def analyze_email(file_path, export_pdf=False):
 
     results = analyze(file_path)
 
-    # Ensure results has at least 3 elements
-    if len(results) < 3:
-        html = "<pre>" + "\n".join(results) + "</pre>"
-        return html, None
+    # Ensure all items are strings
+    results_str = [str(r) for r in results]
 
-    # Extract basic verdict info safely
-    attack_score = results[0] if len(results) > 0 else 0
-    attack_type = results[1] if len(results) > 1 else "Unknown"
-    verdict = results[2] if len(results) > 2 else "Safe"
+    # Extract basic verdict info
+    attack_score = results_str[0]
+    attack_type = results_str[1]
+    verdict = results_str[2]
 
     # Correct handling of attack_score
-    try:
-        score_val = int(attack_score)
-    except:
-        score_val = 0
-    if score_val > 100:
+    score_val = int(attack_score)
+    if score_val > 100:  
         score_val = 100
 
     # Verdict color
@@ -43,21 +38,19 @@ def analyze_email(file_path, export_pdf=False):
         verdict_color = "#f1c40f"
 
     # Extract sections safely
-    tags, findings, highlighted_body = [], [], []
     try:
-        tags_index = results.index("---- Attack Analysis Tags ----")
-        findings_index = results.index("---- Detailed Findings ----")
-        body_index = results.index("---- Highlighted Body ----")
-        tags = results[tags_index + 1 : findings_index]
-        findings = results[findings_index + 1 : body_index]
-        highlighted_body = results[body_index + 1 :]
+        tags_index = results_str.index("---- Attack Analysis Tags ----")
+        findings_index = results_str.index("---- Detailed Findings ----")
+        body_index = results_str.index("---- Highlighted Body ----")
     except ValueError:
-        # Sections not found, fallback to full report
-        html = "<pre>" + "\n".join(results) + "</pre>"
-        pdf_path = html_to_pdf(html) if export_pdf else None
-        return html, pdf_path
+        html = "<pre>" + "\n".join(results_str) + "</pre>"
+        return html, None
 
-    # HTML Report with collapsible sections
+    tags = results_str[tags_index + 1 : findings_index]
+    findings = results_str[findings_index + 1 : body_index]
+    highlighted_body = results_str[body_index + 1 :]
+
+    # 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>
@@ -92,7 +85,10 @@ def analyze_email(file_path, export_pdf=False):
     </div>
     """
 
-    pdf_path = html_to_pdf(html) if export_pdf else None
+    pdf_path = None
+    if export_pdf:
+        pdf_path = html_to_pdf(html)
+
     return html, pdf_path
 
 # Gradio Interface