Update app.py

app.py

@@ -4,6 +4,7 @@ import zipfile
 import logging
 from threading import Thread
 import json
+import time
 
 import dash
 import dash_bootstrap_components as dbc
@@ -21,6 +22,7 @@ app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 # Global variables
 generated_file = None
 progress = 0
+is_processing = False
 
 # Layout
 app.layout = dbc.Container([
@@ -49,9 +51,13 @@ app.layout = dbc.Container([
     ], className="my-3"),
     dbc.Button("Split PDF", id='split-button', color="primary", className="mt-3", disabled=True),
     dbc.Progress(id='progress-bar', className="my-3"),
-    dbc.Button("Download ZIP", id='download-button', color="success", className="mt-3"),
+    dbc.Spinner(id="processing-spinner", color="primary", type="border", children=[
+        html.Div(id='processing-status')
+    ]),
+    dbc.Button("Download ZIP", id='download-button', color="success", className="mt-3", disabled=True),
     dcc.Download(id="download-zip"),
     html.Div(id='log-output', style={'whiteSpace': 'pre-line'}),
+    dcc.Interval(id='interval-component', interval=1000, n_intervals=0),  # 1 second interval
 ], fluid=True)
 
 @callback(
@@ -96,8 +102,9 @@ def manage_ranges(add_clicks, remove_clicks, existing_ranges):
     return existing_ranges
 
 @callback(
-    Output('progress-bar', 'value'),
-    Output('log-output', 'children'),
+    Output('processing-status', 'children'),
+    Output('split-button', 'disabled', allow_duplicate=True),
+    Output('download-button', 'disabled'),
     Input('split-button', 'n_clicks'),
     State('upload-pdf', 'contents'),
     State('upload-pdf', 'filename'),
@@ -105,7 +112,7 @@ def manage_ranges(add_clicks, remove_clicks, existing_ranges):
     prevent_initial_call=True
 )
 def split_pdf(n_clicks, contents, filename, ranges):
-    global progress
+    global progress, is_processing
     if not contents or not ranges:
         logger.warning("Split PDF clicked but no content or ranges provided")
         raise PreventUpdate
@@ -115,14 +122,15 @@ def split_pdf(n_clicks, contents, filename, ranges):
     logger.info(f"Processing {len(ranges)} ranges")
     
     progress = 0  # Reset progress
+    is_processing = True
     
     thread = Thread(target=process_pdf, args=(contents, filename, ranges))
     thread.start()
     
-    return 0, "PDF splitting process started. Check console for detailed logs."
+    return "Processing started...", True, True
 
 def process_pdf(contents, filename, ranges):
-    global progress, generated_file
+    global progress, generated_file, is_processing
     try:
         # Decode PDF content
         content_type, content_string = contents.split(',')
@@ -149,6 +157,7 @@ def process_pdf(contents, filename, ranges):
                 zf.writestr(f'split_{i+1}.pdf', output.getvalue())
                 
                 progress = (i + 1) / len(ranges) * 100
+                time.sleep(0.1)  # Simulate some processing time
         
         zip_buffer.seek(0)
         generated_file = zip_buffer.getvalue()
@@ -156,23 +165,26 @@ def process_pdf(contents, filename, ranges):
     except Exception as e:
         logger.error(f"Error processing PDF: {str(e)}")
         progress = -1
+    finally:
+        is_processing = False
 
 @callback(
-    Output('progress-bar', 'value', allow_duplicate=True),
-    Output('log-output', 'children', allow_duplicate=True),
-    Input('progress-bar', 'value'),
+    Output('progress-bar', 'value'),
+    Output('processing-status', 'children', allow_duplicate=True),
+    Output('download-button', 'disabled', allow_duplicate=True),
+    Input('interval-component', 'n_intervals'),
     prevent_initial_call=True
 )
-def update_progress(value):
-    global progress
-    if progress == 100:
-        logger.info("PDF splitting completed")
-        return 100, "PDF splitting completed. Click 'Download ZIP' to get your files."
+def update_progress(n):
+    global progress, is_processing, generated_file
+    if is_processing:
+        return progress, f"Processing... {progress:.0f}% complete", True
+    elif progress == 100 and generated_file is not None:
+        return 100, "PDF splitting completed. Click 'Download ZIP' to get your files.", False
     elif progress == -1:
-        logger.error("PDF splitting failed")
-        return 0, "Error occurred during PDF splitting. Check console for details."
+        return 0, "Error occurred during PDF splitting. Please try again.", True
     else:
-        return progress, f"Processing... {progress:.0f}% complete"
+        raise PreventUpdate
 
 @callback(
     Output("download-zip", "data"),