Update app.py

app.py

@@ -245,31 +245,50 @@ app.layout = dbc.Container([
 ], fluid=True)
 
 @app.callback(
-    Output("pdf-store", "data"),
-    Output("status-message", "children"),
     Output("submit-button", "disabled"),
+    Output("download-button", "disabled"),
+    Output("progress-bar", "value"),
+    Output("progress-message", "children"),
+    Output("pdf-store", "data"),
     Input("submit-button", "n_clicks"),
+    Input("progress-interval", "n_intervals"),
     State("url-input", "value"),
     State("depth-slider", "value"),
+    State("progress-store", "data"),
     prevent_initial_call=True
 )
-def generate_pdf(n_clicks, url, depth):
-    if not url:
-        return None, "Please enter a URL", False
-    
-    try:
-        pdf_content = asyncio.run(process_url(url, depth, lambda x: None))
-        return pdf_content, "PDF generated successfully. Ready for download!", False
-    except Exception as e:
-        logger.error(f"Error generating PDF: {str(e)}")
-        return None, f"Error: {str(e)}", False
+def update_progress(submit_clicks, n_intervals, url, depth, progress_data):
+    global generated_file
+    ctx = dash.callback_context
+    if not ctx.triggered:
+        raise PreventUpdate
 
-@app.callback(
-    Output("download-button", "disabled"),
-    Input("pdf-store", "data")
-)
-def update_download_button(pdf_data):
-    return pdf_data is None
+    triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
+
+    if triggered_id == "submit-button":
+        if not url:
+            return False, True, 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
+
+    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
+        elif progress == 100 and generated_file:
+            # Encode the PDF file as base64
+            encoded_pdf = base64.b64encode(generated_file).decode('utf-8')
+            return False, False, 100, "PDF ready for download!", {'content': encoded_pdf, 'filename': f"website_content_{int(time.time())}.pdf"}
+        elif progress > 0:
+            return True, True, progress, message, None
+
+    return False, True, 0, "", None
 
 @app.callback(
     Output("download-pdf", "data"),
@@ -280,7 +299,8 @@ def update_download_button(pdf_data):
 def download_pdf(n_clicks, pdf_data):
     if pdf_data is None:
         raise PreventUpdate
-    return dcc.send_bytes(pdf_data, f"website_content_{int(time.time())}.pdf")
+    
+    return dcc.send_bytes(base64.b64decode(pdf_data['content']), pdf_data['filename'])
 
 def background_task(url, depth, task_id):
     global generated_file