Update app.py

app.py

@@ -3,7 +3,8 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import io
 from datetime import datetime
-from fpdf import FPDF
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
 
 # ----------------------------
 # Load Logs
@@ -13,7 +14,6 @@ def load_logs(file_obj):
         content = file_obj.read().decode('utf-8')
         df = pd.read_csv(io.StringIO(content), parse_dates=['Timestamp'])
     else:
-        # Sample fallback data
         sample_data = {
             'DeviceID': ['D001', 'D002', 'D003'],
             'Lab': ['Lab A', 'Lab B', 'Lab A'],
@@ -49,36 +49,44 @@ def generate_chart(df):
     return fig
 
 # ----------------------------
-# Export to PDF
+# Export PDF with ReportLab
 # ----------------------------
 def export_pdf(df):
     summary = summarize_logs(df)
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
 
-    pdf.cell(200, 10, txt="LabOps Dashboard Summary Report", ln=True, align='C')
-    pdf.cell(200, 10, txt=f"Generated on {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=True, align='C')
-    pdf.ln(10)
+    buffer = io.BytesIO()
+    pdf = canvas.Canvas(buffer, pagesize=letter)
+    width, height = letter
 
-    headers = ["Lab", "Device", "OK", "DOWN"]
-    col_widths = [40, 40, 30, 30]
-    for header, width in zip(headers, col_widths):
-        pdf.cell(width, 10, header, border=1)
-    pdf.ln()
+    pdf.setFont("Helvetica-Bold", 14)
+    pdf.drawCentredString(width / 2, height - 40, "LabOps Dashboard Summary Report")
+    pdf.setFont("Helvetica", 10)
+    pdf.drawCentredString(width / 2, height - 60, f"Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+
+    y = height - 100
+    pdf.setFont("Helvetica-Bold", 10)
+    pdf.drawString(50, y, "Lab")
+    pdf.drawString(150, y, "Device Type")
+    pdf.drawString(300, y, "OK")
+    pdf.drawString(350, y, "DOWN")
+    y -= 20
+    pdf.setFont("Helvetica", 10)
 
     for (lab, dev_type), row in summary.iterrows():
         ok = int(row.get('OK', 0))
         down = int(row.get('DOWN', 0))
-        pdf.cell(40, 10, lab, border=1)
-        pdf.cell(40, 10, dev_type, border=1)
-        pdf.cell(30, 10, str(ok), border=1)
-        pdf.cell(30, 10, str(down), border=1)
-        pdf.ln()
-
-    # Convert to byte stream for Gradio
-    pdf_output = pdf.output(dest='S').encode('latin1')
-    return ("LabOps_Summary.pdf", pdf_output)
+        pdf.drawString(50, y, str(lab))
+        pdf.drawString(150, y, str(dev_type))
+        pdf.drawString(300, y, str(ok))
+        pdf.drawString(350, y, str(down))
+        y -= 20
+        if y < 50:
+            pdf.showPage()
+            y = height - 50
+
+    pdf.save()
+    buffer.seek(0)
+    return ("LabOps_Summary.pdf", buffer.read())
 
 # ----------------------------
 # Gradio App