Update app.py

app.py

@@ -193,8 +193,9 @@ def website_to_pdf(all_pages, progress_callback):
             
             progress = min((i + chunk_size) / len(all_pages), 1.0)
             progress_callback(f"Processing pages... {progress:.0%}")
+            logger.info(f"Generated PDF chunk {i//chunk_size + 1}/{total_chunks}")
 
-        # Merge PDF chunks
+        logger.info("Merging PDF chunks...")
         output_pdf = os.path.join(temp_dir, "final.pdf")
         merger = PdfMerger()
         for temp_file in temp_files:
@@ -202,6 +203,7 @@ def website_to_pdf(all_pages, progress_callback):
         merger.write(output_pdf)
         merger.close()
 
+        logger.info("PDF generation complete. Reading final PDF...")
         with open(output_pdf, 'rb') as f:
             return f.read()
 
@@ -211,9 +213,11 @@ async def process_url(url, depth, progress_callback):
         if not all_pages:
             return "No pages were successfully crawled. Please check the URL and try again."
         
+        logger.info("Crawling complete. Starting PDF generation...")
         # Use ThreadPoolExecutor to run PDF generation in a separate thread
         loop = asyncio.get_event_loop()
         pdf_content = await loop.run_in_executor(executor, website_to_pdf, all_pages, progress_callback)
+        logger.info("PDF generation complete.")
         return pdf_content
     except Exception as e:
         logger.error(f"Error in process_url: {str(e)}")
@@ -262,7 +266,7 @@ app.layout = dbc.Container([
     Output("progress-bar", "style"),
     Input("submit-button", "n_clicks"),
     Input("progress-interval", "n_intervals"),
-    Input("progress-store", "data"),  # Add this line
+    Input("progress-store", "data"),
     State("url-input", "value"),
     State("depth-slider", "value"),
     prevent_initial_call=True
@@ -329,7 +333,9 @@ def background_task(url, depth, task_id):
         app.layout.children[1].data = message
 
     try:
+        logger.info(f"Starting background task for URL: {url}, depth: {depth}")
         pdf_content = asyncio.run(process_url(url, depth, progress_callback))
+        logger.info("Background task completed successfully")
         # Store the result in the progress-store
         app.layout.children[1].data = pdf_content
     except Exception as e: