Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 from analyze_email_main import analyze
 from weasyprint import HTML
 import tempfile
+import os
 
 def html_to_pdf(html_content):
     """Convert HTML string to a temporary PDF file and return its path."""
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
-        HTML(string=html_content).write_pdf(tmp_pdf.name)
-        return tmp_pdf.name
+    tmp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    tmp_pdf.close()  # Close before writing
+    HTML(string=html_content).write_pdf(tmp_pdf.name)
+    return tmp_pdf.name
 
 def analyze_email(file_path, export_pdf=False):
     if file_path is None:
@@ -25,69 +27,54 @@ def analyze_email(file_path, export_pdf=False):
         score_val = 100
 
     # Verdict color
-    if "Malicious" in verdict:
-        verdict_color = "#e74c3c"  # red
-    elif "Suspicious" in verdict:
-        verdict_color = "#e67e22"  # orange
-    elif "Spam" in verdict:
-        verdict_color = "#f1c40f"  # yellow
-    else:
-        verdict_color = "#2ecc71"  # green
+    verdict_color = (
+        "#e74c3c" if "Malicious" in verdict else
+        "#e67e22" if "Suspicious" in verdict else
+        "#f1c40f" if "Spam" in verdict else
+        "#2ecc71"
+    )
 
     # Extract sections
     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 :]
     except ValueError:
         html = "<pre>" + "\n".join(results) + "</pre>"
         return html, None
 
-    tags = results[tags_index + 1 : findings_index]
-    findings = results[findings_index + 1 : body_index]
-    highlighted_body = results[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>
-
       <h3>Attack Score</h3>
       <div style="background:#ecf0f1; border-radius:10px; overflow:hidden; height:24px; width:100%;">
         <div style="width:{score_val}%; background:#3498db; height:100%; text-align:center; color:white; line-height:24px;">
           {score_val}/100
         </div>
       </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()}
       </span></p>
-
       <h3>🔖 Attack Analysis Tags</h3>
-      <ul>
-        {''.join(f'<li>{tag}</li>' for tag in tags)}
-      </ul>
-
+      <ul>{''.join(f'<li>{tag}</li>' for tag in tags)}</ul>
       <h3>🕵️ Detailed Findings</h3>
-      <ul>
-        {''.join(f'<li>{finding}</li>' for finding in findings)}
-      </ul>
-
+      <ul>{''.join(f'<li>{finding}</li>' for finding in findings)}</ul>
       <h3>📩 Highlighted Email Body</h3>
       <div style='border:1px solid #ccc; padding:15px; background:#fdfdfd; border-radius:8px;'>
         {"<br>".join(highlighted_body)}
       </div>
-
     </div>
     """
 
-    pdf_path = None
-    if export_pdf:
-        pdf_path = html_to_pdf(html)
+    pdf_path = html_to_pdf(html) if export_pdf else None
 
-    return html, pdf_path
+    # Ensure Gradio doesn't break when PDF is not generated
+    return html, pdf_path if pdf_path and os.path.exists(pdf_path) else None
 
 # Gradio Interface
 demo = gr.Interface(
@@ -101,8 +88,10 @@ demo = gr.Interface(
         gr.File(label="Download PDF Report")
     ],
     title="📧 Email Security Analyzer",
-    description="Upload an .eml file to detect phishing, spam, or malicious emails. "
-                "Check 'Export PDF Report' to download a professional PDF report."
+    description=(
+        "Upload an .eml file to detect phishing, spam, or malicious emails. "
+        "Check 'Export PDF Report' to download a professional PDF report."
+    )
 )
 
 if __name__ == "__main__":