Update app.py

app.py

@@ -227,24 +227,6 @@ async def process_url(url, depth, progress_callback):
 app.layout = dbc.Container([
     dcc.Store(id='pdf-store'),
     dcc.Store(id='progress-store'),
-    dbc.Navbar(
-        dbc.Container([
-            html.A(
-                dbc.Row([
-                    dbc.Col(html.Img(src="/assets/logo.png", height="30px")),
-                    dbc.Col(dbc.NavbarBrand("Website to PDF Converter", className="ms-2")),
-                ],
-                align="center",
-                className="g-0",
-                ),
-                href="/",
-                style={"textDecoration": "none"},
-            )
-        ]),
-        color="#116F70",
-        dark=True,
-    ),
-    
     dbc.Card(
         dbc.CardBody([
             html.H1("Website to PDF Converter", className="text-center mb-4"),
@@ -252,18 +234,20 @@ 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.Progress(id="progress-bar", style={"visibility": "hidden"}),
+            dbc.Button("Download PDF", id="download-button", color="secondary", className="mb-3 w-100", disabled=True),
             dbc.Spinner(html.Div(id="output-area"), color="primary", type="grow"),
             dcc.Interval(id="progress-interval", interval=1000, n_intervals=0, disabled=True),
+            dcc.Download(id="download-pdf")
         ]),
         className="mt-4"
     )
 ], fluid=True)
 
 @app.callback(
-    Output("output-area", "children"),
+    Output("download-button", "disabled"),
+    Output("download-button", "color"),
     Output("progress-interval", "disabled"),
-    Output("progress-bar", "style"),
+    Output("pdf-store", "data"),
     Input("submit-button", "n_clicks"),
     Input("progress-interval", "n_intervals"),
     Input("progress-store", "data"),
@@ -280,45 +264,39 @@ def update_output(n_clicks, n_intervals, progress_data, url, depth):
 
     if triggered_id == "submit-button":
         if not url:
-            return "Please enter a valid URL.", True, {"visibility": "hidden"}
+            return True, "secondary", True, None
 
         # Start the background task
         task_id = str(uuid.uuid4())
         executor.submit(background_task, url, depth, task_id)
         
-        return "Processing... Please wait.", False, {"visibility": "visible"}
+        return True, "secondary", False, None
 
     elif triggered_id == "progress-interval" or triggered_id == "progress-store":
-        # Check progress
         if progress_data is None:
-            return "Processing... Please wait.", False, {"visibility": "visible"}
+            return True, "secondary", False, None
         
         if isinstance(progress_data, str) and progress_data.startswith("Error"):
-            return progress_data, True, {"visibility": "hidden"}
-        
-        if isinstance(progress_data, str) and progress_data.startswith("Processing"):
-            return progress_data, False, {"visibility": "visible"}
+            return True, "secondary", True, None
         
-        # PDF generation complete
         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"}
+            encoded = base64.b64encode(progress_data).decode()
+            return False, "primary", True, encoded
+
+    return True, "secondary", False, None
 
-    raise PreventUpdate
+@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:
+        raise PreventUpdate
+    
+    decoded = base64.b64decode(pdf_data)
+    return dcc.send_bytes(decoded, f"website_content_{int(time.time())}.pdf")
 
 def background_task(url, depth, task_id):
     def progress_callback(message):