Update app.py

app.py

@@ -300,33 +300,26 @@ def update_output(n_clicks, n_intervals, progress_data, url, depth):
             return progress_data, False, {"visibility": "visible"}
         
         # PDF generation complete
-        try:
-            encoded = base64.b64encode(progress_data).decode()
-            return html.Div([
-                html.H4("PDF Generated Successfully"),
-                html.A(
-                    dbc.Button("Download PDF", color="success", className="mt-2"),
-                    href=f"data:application/pdf;base64,{encoded}",
-                    download="website_content.pdf"
-                )
-            ]), True, {"visibility": "hidden"}
-        except Exception as e:
-            logger.error(f"Error creating download link: {str(e)}")
-            return f"An error occurred while creating the download link: {str(e)}", True, {"visibility": "hidden"}
+        if isinstance(progress_data, bytes):
+            try:
+                encoded = base64.b64encode(progress_data).decode()
+                filename = f"website_content_{int(time.time())}.pdf"
+                return html.Div([
+                    html.H4("PDF Generated Successfully"),
+                    html.P("Click the link below to download your PDF:"),
+                    html.A(
+                        filename,
+                        href=f"data:application/pdf;base64,{encoded}",
+                        download=filename,
+                        className="btn btn-success mt-2"
+                    )
+                ]), True, {"visibility": "hidden"}
+            except Exception as e:
+                logger.error(f"Error creating download link: {str(e)}")
+                return f"An error occurred while creating the download link: {str(e)}", True, {"visibility": "hidden"}
 
     raise PreventUpdate
 
-@app.callback(
-    Output('progress-store', 'data'),
-    Input('progress-interval', 'n_intervals'),
-    prevent_initial_call=True
-)
-def update_progress(n):
-    # This function will be called every second to update the progress
-    # You can implement a mechanism to check the actual progress of the PDF generation
-    # For now, we'll just return a placeholder message
-    return "Processing... Please wait."
-
 def background_task(url, depth, task_id):
     def progress_callback(message):
         # Update progress in the progress-store
@@ -336,7 +329,7 @@ def background_task(url, depth, task_id):
         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
+        # Store the PDF content directly in the progress-store
         app.layout.children[1].data = pdf_content
     except Exception as e:
         logger.error(f"Error in background task: {str(e)}")