Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from analyze_email_main import analyze
+import pdfkit
+import tempfile
+import os
 
-def analyze_email(file_obj):
+def analyze_email(file_obj, export_pdf=False):
     if file_obj is None:
-        return "<p style='color:red;'>Please upload a .eml file to analyze.</p>"
+        return "<p style='color:red;'>Please upload a .eml file to analyze.</p>", None
 
     results = analyze(file_obj.name)
 
@@ -32,7 +35,8 @@ def analyze_email(file_obj):
         findings_index = results.index("---- Detailed Findings ----")
         body_index = results.index("---- Highlighted Body ----")
     except ValueError:
-        return "<pre>" + "\n".join(results) + "</pre>"
+        html = "<pre>" + "\n".join(results) + "</pre>"
+        return html, None
 
     tags = results[tags_index + 1 : findings_index]
     findings = results[findings_index + 1 : body_index]
@@ -53,7 +57,7 @@ def analyze_email(file_obj):
 
       <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.replace('Final Verdict:', '').strip()}
       </span></p>
 
       <h3>🔖 Attack Analysis Tags</h3>
@@ -74,16 +78,29 @@ def analyze_email(file_obj):
     </div>
     """
 
-    return html
+    pdf_path = None
+    if export_pdf:
+        # Generate temporary PDF file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
+            pdf_path = tmp_pdf.name
+        pdfkit.from_string(html, pdf_path)
+
+    return html, pdf_path
 
 # Gradio Interface
 demo = gr.Interface(
     fn=analyze_email,
-    inputs=gr.File(label="Upload .eml File", type="file"),
-    outputs=gr.HTML(label="Analysis Report"),
+    inputs=[
+        gr.File(label="Upload .eml File", type="file"),
+        gr.Checkbox(label="Export PDF Report")
+    ],
+    outputs=[
+        gr.HTML(label="Analysis Report"),
+        gr.File(label="Download PDF Report")
+    ],
     title="📧 Email Security Analyzer",
     description="Upload an .eml file to detect phishing, spam, or malicious emails. "
-                "This analyzer combines AI, heuristics, and threat intelligence feeds."
+                "Check 'Export PDF Report' to download a professional PDF report."
 )
 
 if __name__ == "__main__":