Update pdf_utils.py

pdf_utils.py

@@ -1,28 +1,20 @@
-from reportlab.lib.pagesizes import A4
-from reportlab.pdfgen import canvas
-import pandas as pd
+from reportlab.lib.pagesizes import A4, landscape
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
+from reportlab.lib import colors
 
-def export_pdf(df: pd.DataFrame, filename="timetable.pdf"):
-    c = canvas.Canvas(filename, pagesize=A4)
-    width, height = A4
+def export_pdf(df, filename="timetable.pdf"):
+    pdf = SimpleDocTemplate(filename, pagesize=landscape(A4))
+    data = [df.columns.tolist()] + df.values.tolist()
 
-    x = 40
-    y = height - 40
-
-    c.setFont("Helvetica-Bold", 14)
-    c.drawString(x, y, "Weekly Time Table")
-    y -= 30
-
-    c.setFont("Helvetica", 10)
-
-    for _, row in df.iterrows():
-        line = f"{row['Day']} | {row['Time']} | {row['Course']} | {row['Teacher']} | {row['Room']}"
-        c.drawString(x, y, line)
-        y -= 18
-
-        if y < 40:
-            c.showPage()
-            y = height - 40
-
-    c.save()
+    table = Table(data)
+    style = TableStyle([
+        ('BACKGROUND', (0,0), (-1,0), colors.lightblue),
+        ('TEXTCOLOR', (0,0), (-1,0), colors.white),
+        ('ALIGN', (0,0), (-1,-1), 'CENTER'),
+        ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
+        ('GRID', (0,0), (-1,-1), 1, colors.black),
+        ('FONTSIZE', (0,0), (-1,-1), 10)
+    ])
+    table.setStyle(style)
+    pdf.build([table])
     return filename