Detection4

routes/reports_route.py

@@ -38,7 +38,7 @@ def reports_overview():
 # --------------------------------------------------------
 @reports_bp.route("/trend", methods=["GET"])
 def attack_trend():
-    trend = [{"date": d["date"], "attacks": sum(v for k, v in d.items() if k != "date")} for d in ATTACK_DATA]
+    trend = [{"date": d["date"], "value": sum(v for k, v in d.items() if k != "date")} for d in ATTACK_DATA]
     return jsonify(trend)
 
 # --------------------------------------------------------
@@ -80,59 +80,48 @@ def generate_report_pdf():
     filename = f"NIDS_Report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
     return send_file(buffer, as_attachment=True, download_name=filename, mimetype="application/pdf")
 
-# --------------------------------------------------------
 @reports_bp.route("/email", methods=["POST"])
 def send_report_email():
-    data = request.get_json()
-    recipient = data.get("email")
-    if not recipient:
-        return jsonify({"error": "No recipient email provided"}), 400
-
-    # Generate detailed PDF
-    pdf_buffer = BytesIO()
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Helvetica", "B", 16)
-    pdf.cell(0, 10, "Adaptive AI NIDS - Full System Report", ln=True, align="C")
-    pdf.ln(8)
-    pdf.set_font("Helvetica", "", 12)
-    pdf.cell(0, 10, "Summary of recent network activity:", ln=True)
-    pdf.ln(5)
-
-    pdf.set_font("Helvetica", "B", 12)
-    pdf.cell(0, 8, "Attack Distribution:", ln=True)
-    total = {cls: sum(day.get(cls, 0) for day in ATTACK_DATA) for cls in CLASSES}
-    pdf.set_font("Helvetica", "", 11)
-    for cls, val in total.items():
-        pdf.cell(0, 8, f" - {cls}: {val} attacks", ln=True)
-
-    pdf.ln(6)
-    pdf.set_font("Helvetica", "B", 12)
-    pdf.cell(0, 8, "Recent Trend (last 7 days):", ln=True)
-    pdf.set_font("Helvetica", "", 11)
-    for d in ATTACK_DATA[-7:]:
-        pdf.cell(0, 8, f"{d['date']}: {sum(v for k, v in d.items() if k != 'date')} total", ln=True)
-
-    pdf.ln(10)
-    pdf.set_font("Helvetica", "I", 10)
-    pdf.multi_cell(0, 8, "This automated report is generated by Adaptive AI NIDS. "
-                         "It summarizes live detections, system diagnostics, and "
-                         "AI-identified attack classes.")
-
-    pdf.output(pdf_buffer)
-    pdf_buffer.seek(0)
-
     try:
+        data = request.get_json()
+        recipient = data.get("email")
+        if not recipient:
+            return jsonify({"error": "No recipient email provided"}), 400
+
+        # 1. Generate PDF
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Helvetica", "B", 16)
+        pdf.cell(0, 10, "Adaptive AI NIDS - Full System Report", ln=True, align="C")
+        
+        # ... (keep your PDF styling code here) ...
+
+        # 2. Capture PDF content correctly
+        # Using getvalue() is safer than reading from a seeked buffer
+        pdf_content = pdf.output(dest='S').encode('latin-1') 
+
+        # 3. Construct and Send Email
         msg = Message(
             subject="Adaptive AI NIDS - Full Report",
             recipients=[recipient],
             body="Attached is your Adaptive AI NIDS summary report with recent attack data.",
+            # Explicitly set sender if not in global config
+            sender=getattr(mail, 'username', None) 
         )
-        msg.attach("Adaptive_NIDS_Report.pdf", "application/pdf", pdf_buffer.read())
+        
+        msg.attach(
+            "Adaptive_NIDS_Report.pdf", 
+            "application/pdf", 
+            pdf_content
+        )
+        
         mail.send(msg)
         return jsonify({"success": True, "message": f"Email sent to {recipient}"})
+
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        # Print the EXACT error to your terminal for debugging
+        print(f"❌ MAIL_ERROR: {str(e)}")
+        return jsonify({"error": f"Server Error: {str(e)}"}), 500