Update app.py

app.py

@@ -255,14 +255,17 @@ app.layout = dbc.Container([
     Output("progress-bar", "value"),
     Output("progress-message", "children"),
     Output("pdf-store", "data"),
+    Output("download-pdf", "data"),
     Input("submit-button", "n_clicks"),
+    Input("download-button", "n_clicks"),
     Input("progress-interval", "n_intervals"),
     State("url-input", "value"),
     State("depth-slider", "value"),
     State("progress-store", "data"),
+    State("pdf-store", "data"),
     prevent_initial_call=True
 )
-def update_output(n_clicks, n_intervals, url, depth, progress_data):
+def update_and_download(submit_clicks, download_clicks, n_intervals, url, depth, progress_data, pdf_data):
     ctx = dash.callback_context
     if not ctx.triggered:
         raise PreventUpdate
@@ -271,26 +274,33 @@ def update_output(n_clicks, n_intervals, url, depth, progress_data):
 
     if triggered_id == "submit-button":
         if not url:
-            return False, 0, "Please enter a URL", None
+            return False, 0, "Please enter a URL", None, None
 
         # Start the background task
         task_id = str(uuid.uuid4())
         executor.submit(background_task, url, depth, task_id)
         
-        return True, 0, "Processing... Please wait.", None
+        return True, 0, "Processing... Please wait.", None, None
 
     elif triggered_id == "progress-interval":
         progress = progress_data['progress']
         message = progress_data['message']
 
         if isinstance(message, str) and message.startswith("Error"):
-            return False, 100, message, None
+            return False, 100, message, None, None
         elif progress == 100 and generated_file:
-            return False, 100, "PDF ready for download!", generated_file
+            return False, 100, "PDF ready for download!", generated_file, None
         elif progress > 0:
-            return True, progress, message, None
+            return True, progress, message, None, None
+
+    elif triggered_id == "download-button":
+        if pdf_data is None:
+            # If PDF is not ready, return None to prevent download
+            return dash.no_update, dash.no_update, dash.no_update, dash.no_update, None
+        
+        return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dcc.send_bytes(pdf_data, f"website_content_{int(time.time())}.pdf")
 
-    return False, 0, "", None
+    return False, 0, "", None, None
 
 @app.callback(
     Output("download-pdf", "data"),