Update app.py

app.py

@@ -226,6 +226,7 @@ async def process_url(url, depth, progress_callback):
         logger.error(f"Error in process_url: {str(e)}")
         return f"An error occurred: {str(e)}"
 
+# App layout
 # App layout
 app.layout = dbc.Container([
     dcc.Store(id='pdf-store'),
@@ -237,7 +238,7 @@ app.layout = dbc.Container([
             dbc.Input(id="url-input", type="text", placeholder="Enter website URL (e.g., https://www.gradio.app/docs)", className="mb-3"),
             dcc.Slider(id="depth-slider", min=1, max=10, step=1, value=3, marks={i: str(i) for i in range(1, 11)}, className="mb-3"),
             dbc.Button("Convert to PDF", id="submit-button", color="primary", className="mb-3 w-100"),
-            dbc.Button("Download PDF", id="download-button", color="secondary", className="mb-3 w-100", disabled=True),
+            dbc.Button("Download PDF", id="download-button", color="secondary", className="mb-3 w-100"),  # Removed disabled property
             html.Div([
                 dbc.Progress(id="progress-bar", value=0, className="mb-3"),
                 html.Div(id="progress-message", className="text-center"),
@@ -251,7 +252,6 @@ app.layout = dbc.Container([
 
 @app.callback(
     Output("submit-button", "disabled"),
-    Output("download-button", "disabled"),
     Output("progress-bar", "value"),
     Output("progress-message", "children"),
     Output("pdf-store", "data"),
@@ -271,26 +271,39 @@ def update_output(n_clicks, n_intervals, url, depth, progress_data):
 
     if triggered_id == "submit-button":
         if not url:
-            return False, True, 0, "Please enter a URL", None
+            return False, 0, "Please enter a URL", None
 
         # Start the background task
         task_id = str(uuid.uuid4())
         executor.submit(background_task, url, depth, task_id)
         
-        return True, True, 0, "Processing... Please wait.", None
+        return True, 0, "Processing... Please wait.", None
 
     elif triggered_id == "progress-interval":
         progress = progress_data['progress']
         message = progress_data['message']
 
         if isinstance(message, str) and message.startswith("Error"):
-            return False, True, 100, message, None
+            return False, 100, message, None
         elif progress == 100 and generated_file:
-            return False, False, 100, "PDF ready for download!", generated_file
+            return False, 100, "PDF ready for download!", generated_file
         elif progress > 0:
-            return True, True, progress, message, None
+            return True, progress, message, None
+
+    return False, 0, "", None
 
-    return False, True, 0, "", None
+@app.callback(
+    Output("download-pdf", "data"),
+    Input("download-button", "n_clicks"),
+    State("pdf-store", "data"),
+    prevent_initial_call=True
+)
+def download_pdf(n_clicks, pdf_data):
+    if pdf_data is None:
+        # If PDF is not ready, return None to prevent download
+        return None
+    
+    return dcc.send_bytes(pdf_data, f"website_content_{int(time.time())}.pdf")
 
 @app.callback(
     Output("download-pdf", "data"),