Update app.py

app.py

@@ -228,8 +228,7 @@ async def process_url(url, depth, progress_callback):
 
 # App layout
 app.layout = dbc.Container([
-    dcc.Store(id='pdf-store'),
-    dcc.Store(id='progress-store', data={'progress': 0, 'message': ''}),
+    dcc.Store(id='pdf-store', storage_type='memory'),
     dbc.Card(
         dbc.CardBody([
             html.H1("Website to PDF Converter", className="text-center mb-4"),
@@ -238,11 +237,7 @@ app.layout = dbc.Container([
             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),
-            html.Div([
-                dbc.Progress(id="progress-bar", value=0, className="mb-3"),
-                html.Div(id="progress-message", className="text-center"),
-            ], className="mb-3"),
-            dcc.Interval(id="progress-interval", interval=1000, n_intervals=0),
+            html.Div(id="status-message", className="text-center mb-3"),
             dcc.Download(id="download-pdf")
         ]),
         className="mt-4"
@@ -250,47 +245,31 @@ app.layout = dbc.Container([
 ], fluid=True)
 
 @app.callback(
-    Output("submit-button", "disabled"),
-    Output("download-button", "disabled"),
-    Output("progress-bar", "value"),
-    Output("progress-message", "children"),
     Output("pdf-store", "data"),
+    Output("status-message", "children"),
+    Output("submit-button", "disabled"),
     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 update_progress(submit_clicks, n_intervals, url, depth, progress_data):
-    ctx = dash.callback_context
-    if not ctx.triggered:
-        raise PreventUpdate
-
-    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:
-            return False, False, 100, "PDF ready for download!", generated_file
-        elif progress > 0:
-            return True, True, progress, message, None
+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
 
-    return False, True, 0, "", None
+@app.callback(
+    Output("download-button", "disabled"),
+    Input("pdf-store", "data")
+)
+def update_download_button(pdf_data):
+    return pdf_data is None
 
 @app.callback(
     Output("download-pdf", "data"),
@@ -301,7 +280,6 @@ def update_progress(submit_clicks, n_intervals, url, depth, progress_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")
 
 def background_task(url, depth, task_id):