Update app.py

app.py

@@ -15,6 +15,7 @@ app = Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 uploaded_files = {}
 converted_files = {}
 conversion_progress = {}
+conversion_complete = False
 
 def convert_pdf_to_docx(pdf_path, docx_path):
     cv = Converter(pdf_path)
@@ -27,7 +28,7 @@ def process_contents(contents, filename):
     return io.BytesIO(decoded)
 
 def convert_files(filenames):
-    global conversion_progress, converted_files
+    global conversion_progress, converted_files, conversion_complete
     total_files = len(filenames)
     for i, filename in enumerate(filenames):
         pdf_file = uploaded_files[filename]
@@ -50,6 +51,7 @@ def convert_files(filenames):
         conversion_progress[filename] = (i + 1) / total_files * 100
 
     conversion_progress['overall'] = 100
+    conversion_complete = True
 
 app.layout = dbc.Container([
     dbc.Card(
@@ -76,7 +78,8 @@ app.layout = dbc.Container([
             html.Div(id='upload-output'),
             dbc.Button("Convert and Download", id="convert-button", color="primary", className="mt-3 mb-3", disabled=True),
             html.Div(id='conversion-output'),
-            dcc.Download(id="download-zip")
+            dcc.Download(id="download-zip"),
+            dcc.Interval(id='interval-component', interval=500, n_intervals=0, disabled=True)
         ]),
         className="mt-3"
     )
@@ -104,36 +107,50 @@ def update_output(list_of_contents, list_of_names):
     return [], True
 
 @app.callback(
-    Output('conversion-output', 'children'),
-    Output('download-zip', 'data'),
+    Output('interval-component', 'disabled'),
     Input('convert-button', 'n_clicks'),
     prevent_initial_call=True
 )
-def convert_and_download(n_clicks):
+def start_conversion(n_clicks):
     if n_clicks is None:
-        return [], None
+        return True
 
-    global conversion_progress, converted_files
+    global conversion_progress, converted_files, conversion_complete
     conversion_progress.clear()
     converted_files.clear()
     conversion_progress['overall'] = 0
+    conversion_complete = False
 
     threading.Thread(target=convert_files, args=(list(uploaded_files.keys()),)).start()
+    return False
 
-    while conversion_progress.get('overall', 0) < 100:
-        time.sleep(0.1)
-        progress_bars = [
-            dbc.Progress(value=conversion_progress.get(filename, 0), label=f"{filename}: {conversion_progress.get(filename, 0):.0f}%", className="mb-3")
-            for filename in uploaded_files.keys()
-        ]
-        yield progress_bars, None
-
-    with io.BytesIO() as zip_buffer:
-        with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
-            for filename, file_content in converted_files.items():
-                zip_file.writestr(filename, file_content.getvalue())
-        
-        return [html.Div("Conversion complete! Downloading ZIP file...")], dcc.send_bytes(zip_buffer.getvalue(), "converted_files.zip")
+@app.callback(
+    Output('conversion-output', 'children'),
+    Input('interval-component', 'n_intervals'),
+    prevent_initial_call=True
+)
+def update_progress(n):
+    progress_bars = [
+        dbc.Progress(value=conversion_progress.get(filename, 0), label=f"{filename}: {conversion_progress.get(filename, 0):.0f}%", className="mb-3")
+        for filename in uploaded_files.keys()
+    ]
+    return progress_bars
+
+@app.callback(
+    Output('download-zip', 'data'),
+    Output('interval-component', 'disabled', allow_duplicate=True),
+    Input('interval-component', 'n_intervals'),
+    prevent_initial_call=True
+)
+def check_conversion_complete(n):
+    if conversion_complete:
+        with io.BytesIO() as zip_buffer:
+            with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
+                for filename, file_content in converted_files.items():
+                    zip_file.writestr(filename, file_content.getvalue())
+            
+            return dcc.send_bytes(zip_buffer.getvalue(), "converted_files.zip"), True
+    return None, False
 
 if __name__ == '__main__':
     print("Starting the Dash application...")