Update app.py

app.py

@@ -12,6 +12,9 @@ import PyPDF2
 
 app = Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 
+# Global variable to track conversion progress
+conversion_progress = 0
+
 app.layout = dbc.Container([
     html.H1("Auto-Wiki", className="my-4"),
     dcc.Upload(
@@ -37,7 +40,8 @@ app.layout = dbc.Container([
     dbc.Progress(id="upload-progress", label="Upload Progress", style={"visibility": "hidden"}),
     dbc.Progress(id="conversion-progress", label="Conversion Progress", style={"visibility": "hidden"}),
     dbc.Button("Convert and Download", id="convert-button", color="primary", className="mt-3", disabled=True),
-    dcc.Download(id="download-zip")
+    dcc.Download(id="download-zip"),
+    dcc.Interval(id='interval-component', interval=500, n_intervals=0)
 ])
 
 def process_docx(contents, filename):
@@ -59,6 +63,28 @@ def process_pdf(contents, filename):
         full_text.append(page.extract_text())
     return '\n\n'.join(full_text)
 
+def process_files(contents, filenames):
+    global conversion_progress
+    processed_files = []
+    for i, (c, n) in enumerate(zip(contents, filenames)):
+        if n.lower().endswith('.docx'):
+            text = process_docx(c, n)
+        elif n.lower().endswith('.pdf'):
+            text = process_pdf(c, n)
+        else:
+            continue  # Skip unsupported file types
+        md = markdown.markdown(text)
+        processed_files.append((n.replace('.docx', '.md').replace('.pdf', '.md'), md))
+        conversion_progress = (i + 1) / len(contents) * 100
+        time.sleep(0.1)  # Simulate processing time
+
+    zip_buffer = io.BytesIO()
+    with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
+        for name, content in processed_files:
+            zip_file.writestr(name, content)
+    
+    return zip_buffer.getvalue()
+
 @app.callback(
     [Output('upload-output', 'children'),
      Output('convert-button', 'disabled'),
@@ -69,11 +95,13 @@ def process_pdf(contents, filename):
      Output('download-zip', 'data')],
     [Input('upload-data', 'contents'),
      Input('upload-data', 'filename'),
-     Input('convert-button', 'n_clicks')],
+     Input('convert-button', 'n_clicks'),
+     Input('interval-component', 'n_intervals')],
     [State('upload-data', 'contents'),
      State('upload-data', 'filename')]
 )
-def update_output(list_of_contents, list_of_names, n_clicks, contents, filenames):
+def update_output(list_of_contents, list_of_names, n_clicks, n_intervals, contents, filenames):
+    global conversion_progress
     ctx = callback_context
     if not ctx.triggered:
         return no_update
@@ -96,31 +124,17 @@ def update_output(list_of_contents, list_of_names, n_clicks, contents, filenames
         if not contents:
             return no_update
 
-        def process_files():
-            processed_files = []
-            for i, (c, n) in enumerate(zip(contents, filenames)):
-                if n.lower().endswith('.docx'):
-                    text = process_docx(c, n)
-                elif n.lower().endswith('.pdf'):
-                    text = process_pdf(c, n)
-                else:
-                    continue  # Skip unsupported file types
-                md = markdown.markdown(text)
-                processed_files.append((n.replace('.docx', '.md').replace('.pdf', '.md'), md))
-                time.sleep(0.1)  # Simulate processing time
-                app.callback_context.response.set_data(f'{{"progress": {(i+1)/len(contents)*100}}}')
-
-            zip_buffer = io.BytesIO()
-            with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
-                for name, content in processed_files:
-                    zip_file.writestr(name, content)
-            
-            return zip_buffer.getvalue()
-
-        thread = threading.Thread(target=process_files)
+        conversion_progress = 0
+        thread = threading.Thread(target=process_files, args=(contents, filenames))
         thread.start()
 
-        return no_update, True, 100, {"visibility": "visible"}, 0, {"visibility": "visible"}, dcc.send_bytes(process_files(), "converted_files.zip")
+        return no_update, True, 100, {"visibility": "visible"}, 0, {"visibility": "visible"}, None
+
+    if ctx.triggered[0]['prop_id'] == 'interval-component.n_intervals':
+        if conversion_progress == 100:
+            return no_update, False, 100, {"visibility": "visible"}, 100, {"visibility": "visible"}, dcc.send_bytes(process_files(contents, filenames), "converted_files.zip")
+        else:
+            return no_update, True, 100, {"visibility": "visible"}, conversion_progress, {"visibility": "visible"}, None
 
     return no_update