Update pdf_generator.py

pdf_generator.py

@@ -48,12 +48,15 @@ def create_page_template(canvas, doc):
     canvas.drawString(30, 20, f"Page {doc.page}")
     canvas.restoreState()
 
-def generate_pdf(pages, answers):
+def generate_pdf(session_state):
     buffer = BytesIO()
     doc = SimpleDocTemplate(buffer, pagesize=A4, rightMargin=20*mm, leftMargin=20*mm, topMargin=20*mm, bottomMargin=20*mm)
     styles = create_styles()
     story = [Paragraph("AI Trust and Opacity Evaluation", styles['Title'])]
 
+    pages = session_state.pages
+    answers = session_state.answers
+
     for page in pages[:-1]:  # Skip the last page
         if 'input_key' in page and page['input_key'] is not None:
             story.append(Paragraph(page['title'], styles['Heading1']))