Update app.py

app.py

@@ -9,7 +9,7 @@ def html_to_pdf(html_content):
         HTML(string=html_content).write_pdf(tmp_pdf.name)
         return tmp_pdf.name
 
-def analyze_email(file_path, export_pdf=False):
+def analyze_email(file_path):
     if file_path is None:
         return "<p style='color:red;'>Please upload a .eml file to analyze.</p>", None
 
@@ -52,7 +52,7 @@ def analyze_email(file_path, export_pdf=False):
     else:
         auth_html = "<p>No auth results found.</p>"
 
-    # Tags (from summary Main Tags if present)
+    # Tags
     main_tags_str = summary.get("Main Tags", "")
     tags = [t.strip() for t in main_tags_str.split(",")] if main_tags_str else []
 
@@ -70,14 +70,17 @@ def analyze_email(file_path, export_pdf=False):
       <p><b>Final Verdict:</b> <span style="background:{verdict_color}; color:white; padding:4px 10px; border-radius:6px;">
         {verdict}
       </span></p>
+
       <details style="margin-top:15px;" open>
         <summary>🔖 Attack Analysis Tags ({len(tags)})</summary>
         {"<ul>" + "".join(f"<li>{tag}</li>" for tag in tags if tag) + "</ul>" if tags else "<p>No special tags</p>"}
       </details>
+
       <details style="margin-top:15px;" open>
         <summary>🛡 Authentication Results</summary>
         {auth_html}
       </details>
+
       <details style="margin-top:15px;" open>
         <summary>🕵️ Detailed Findings</summary>
         <h4>Headers ({len(header_findings)})</h4>
@@ -87,6 +90,7 @@ def analyze_email(file_path, export_pdf=False):
         <h4>URLs ({len(url_findings)})</h4>
         {"<ul>" + "".join(f"<li>{f}</li>" for f in url_findings) + "</ul>" if url_findings else "<p>No URL findings.</p>"}
       </details>
+
       <details style="margin-top:15px;">
         <summary>📩 Highlighted Email Body</summary>
         <div style='border:1px solid #ccc; padding:15px; background:#fdfdfd; border-radius:8px; white-space:pre-wrap;'>
@@ -96,29 +100,27 @@ def analyze_email(file_path, export_pdf=False):
     </div>
     """
 
-    pdf_path = None
-    if export_pdf:
-        try:
-            pdf_path = html_to_pdf(html)
-        except Exception as e:
-            html += f"<p style='color:#c0392b;'><b>PDF export failed:</b> {e}</p>"
+    # ALWAYS generate PDF
+    try:
+        pdf_path = html_to_pdf(html)
+    except Exception as e:
+        pdf_path = None
+        html += f"<p style='color:#c0392b;'><b>PDF export failed:</b> {e}</p>"
 
     return html, pdf_path
 
-# Gradio Interface
+# Gradio Interface (checkbox removed)
 demo = gr.Interface(
     fn=analyze_email,
     inputs=[
-        gr.File(label="Upload .eml File", type="filepath"),
-        gr.Checkbox(label="Export PDF Report")
+        gr.File(label="Upload .eml File", type="filepath")
     ],
     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. "
-                "Check 'Export PDF Report' to download a professional PDF report."
+    description="Upload an .eml file to detect phishing, spam, or malicious emails. A PDF report will be generated automatically."
 )
 
 demo.launch()