Update pdf_utils.py

pdf_utils.py

@@ -1,20 +1,32 @@
 from reportlab.lib.pagesizes import A4, landscape
-from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer
 from reportlab.lib import colors
+from reportlab.lib.styles import getSampleStyleSheet
 
 def export_pdf(df, filename="timetable.pdf"):
     pdf = SimpleDocTemplate(filename, pagesize=landscape(A4))
-    data = [df.columns.tolist()] + df.values.tolist()
+    elements = []
+    styles = getSampleStyleSheet()
 
-    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])
+    current_class = None
+    for idx, row in df.iterrows():
+        if row['Class/Program'] != current_class:
+            current_class = row['Class/Program']
+            elements.append(Spacer(1,12))
+            elements.append(Paragraph(f"<b>Class/Program: {current_class}</b>", styles['Heading2']))
+            class_df = df[df['Class/Program']==current_class]
+            data = [class_df.columns.tolist()] + class_df.values.tolist()
+            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), 9)
+            ])
+            table.setStyle(style)
+            elements.append(table)
+
+    pdf.build(elements)
     return filename