Update pdf_utils.py

pdf_utils.py

@@ -1,33 +1,47 @@
-from reportlab.lib.pagesizes import letter, landscape
-from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import A4
 from reportlab.lib import colors
+from reportlab.lib.styles import ParagraphStyle
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph
 
-def export_timetable_pdf(timetable_dict, filename="timetable.pdf"):
-    c = canvas.Canvas(filename, pagesize=landscape(letter))
-    width, height = landscape(letter)
-    y_start = height - 50
-    for cls, df in timetable_dict.items():
-        c.setFont("Helvetica-Bold", 16)
-        c.drawString(50, y_start, cls)
-        y_start -= 30
-        c.setFont("Helvetica", 12)
-        row_height = 50
-        col_width = (width - 100)/len(df.columns)
-        # Draw header
-        for j, day in enumerate(df.columns):
-            c.setFillColor(colors.lightgrey)
-            c.rect(50 + j*col_width, y_start, col_width, row_height, fill=True)
-            c.setFillColor(colors.black)
-            c.drawString(50 + j*col_width + 5, y_start + row_height/2 - 6, day)
-        # Draw rows
-        for i, slot in enumerate(df.index):
-            for j, day in enumerate(df.columns):
-                x = 50 + j*col_width
-                y = y_start - (i+1)*row_height
-                c.rect(x, y, col_width, row_height)
-                text = df.loc[slot, day] if df.loc[slot, day] is not None else ""
-                c.drawString(x + 5, y + row_height/2 - 6, f"{slot}\n{text}")
-        y_start = y - 70  # space before next class
-        c.showPage()
-    c.save()
-    return filename
+TIME_SLOTS = ["9:00-10:30", "10:30-12:00", "12:00-1:30", "1:30-3:00"]
+DAYS = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday"]
+
+def export_timetable_pdf(timetable_dict, output_file="timetable.pdf"):
+    doc = SimpleDocTemplate(output_file, pagesize=A4)
+    elements = []
+    
+    # Multi-line style
+    style = ParagraphStyle(name="Normal", fontSize=9, leading=12)
+    
+    for cls, timetable in timetable_dict.items():
+        # Class/Program Name on top
+        elements.append(Paragraph(f"<b>{cls}</b>", ParagraphStyle(name="Heading", fontSize=14)))
+        elements.append(Paragraph("<br/>", style))
+        
+        # Table header
+        data = [["Day"] + TIME_SLOTS]
+        
+        for day in DAYS:
+            row = [day]
+            for slot in TIME_SLOTS:
+                info = timetable[day][slot]
+                if info:
+                    text = f"{info['Course']}\n{info['Teacher']}\n{info['Room']}"
+                    row.append(Paragraph(text, style))
+                else:
+                    row.append("")
+            data.append(row)
+        
+        table = Table(data, repeatRows=1, colWidths=[60]+[100]*len(TIME_SLOTS))
+        table.setStyle(TableStyle([
+            ('GRID', (0,0), (-1,-1), 1, colors.black),
+            ('BACKGROUND', (0,0), (-1,0), colors.lightblue),
+            ('VALIGN', (0,0), (-1,-1), 'MIDDLE'),
+            ('ALIGN', (0,0), (-1,0), 'CENTER'),
+            ('ALIGN', (0,1), (-1,-1), 'CENTER'),
+        ]))
+        elements.append(table)
+        elements.append(Paragraph("<br/><br/>", style))
+    
+    doc.build(elements)
+    return output_file