Update app.py

app.py

@@ -21,6 +21,8 @@ import ssl
 from io import BytesIO
 from concurrent.futures import ThreadPoolExecutor
 import math
+from PyPDF2 import PdfMerger
+
 
 # Initialize Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
@@ -199,13 +201,16 @@ async def website_to_pdf(all_pages, progress_callback):
             except Exception as e:
                 logger.error(f"Error generating PDF chunk {i}: {str(e)}")
 
-    # Combine PDF chunks
-    combined_pdf = FPDF()
+    # Combine PDF chunks using PyPDF2
+    merger = PdfMerger()
     for chunk in pdf_chunks:
-        combined_pdf.add_page()
-        combined_pdf.put_file(chunk)
+        merger.append(BytesIO(chunk))
+
+    output = BytesIO()
+    merger.write(output)
+    merger.close()
 
-    return combined_pdf.output(dest='S').encode('latin-1')
+    return output.getvalue()
 
 async def process_url(url, depth, progress_callback):
     try: