Update app.py

app.py

@@ -156,23 +156,63 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
 # Generate PDF report with signature slots using in-memory buffer
 def generate_pdf(score, checklist_summary, missing_summary, status):
     try:
+        # Sanitize inputs to ensure they are strings and handle None
+        score = str(float(score)) if score is not None else "0"
+        checklist_summary = str(checklist_summary) if checklist_summary is not None else ""
+        missing_summary = str(missing_summary) if missing_summary is not None else ""
+        status = str(status) if status is not None else ""
+
         # Use an in-memory buffer instead of writing to disk
         buffer = io.BytesIO()
         c = canvas.Canvas(buffer, pagesize=letter)
+        
+        # Set initial font and draw title
         c.setFont("Helvetica-Bold", 16)
         c.drawString(50, 750, "Project Closure Readiness Report")
+        
+        # Draw basic fields
         c.setFont("Helvetica", 12)
         c.drawString(50, 720, f"Readiness Score: {score}%")
         c.drawString(50, 700, f"Status: {status}")
         c.drawString(50, 680, "Checklist Summary:")
+
+        # Draw checklist summary with Y-position overflow handling
         y = 660
+        BOTTOM_MARGIN = 50  # Minimum Y position before creating a new page
         for line in checklist_summary.split("\n"):
+            if y < BOTTOM_MARGIN:
+                c.showPage()  # Create a new page
+                c.setFont("Helvetica", 12)  # Reset font for new page
+                y = 750  # Reset Y position
             c.drawString(50, y, line)
             y -= 20
-        c.drawString(50, y-20, "Missing Items:")
-        c.drawString(50, y-40, missing_summary)
-        c.drawString(50, y-80, "Stakeholder Signature: ____________________")
-        c.drawString(50, y-100, "Date: ____________________")
+
+        # Draw remaining fields with Y-position checks
+        if y < BOTTOM_MARGIN:
+            c.showPage()
+            c.setFont("Helvetica", 12)
+            y = 750
+        c.drawString(50, y - 20, "Missing Items:")
+
+        if y - 40 < BOTTOM_MARGIN:
+            c.showPage()
+            c.setFont("Helvetica", 12)
+            y = 750
+        c.drawString(50, y - 40, missing_summary)
+
+        if y - 80 < BOTTOM_MARGIN:
+            c.showPage()
+            c.setFont("Helvetica", 12)
+            y = 750
+        c.drawString(50, y - 80, "Stakeholder Signature: ____________________")
+
+        if y - 100 < BOTTOM_MARGIN:
+            c.showPage()
+            c.setFont("Helvetica", 12)
+            y = 750
+        c.drawString(50, y - 100, "Date: ____________________")
+
+        # Finalize the PDF
         c.save()
         buffer.seek(0)
         return buffer, "readiness_report.pdf"
@@ -197,9 +237,7 @@ with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-
             submit_btn = gr.Button("Evaluate Readiness")
         with gr.Column(scale=3):
             score_output = gr.Number(label="Readiness Score (%)")
-            progress_output = gr.HTML(
-                label="Alert Indicator: Progress"
-            )
+            progress_output = gr.HTML(label="Alert Indicator: Progress")
             gr.Markdown("Color-coded readiness: Red (<70%), Yellow (70-90%), Green (>90%)")
             status_output = gr.Textbox(label="Overall Status")
             with gr.Group():