Update app.py via AI Editor

app.py

@@ -192,6 +192,7 @@ app.layout = dbc.Container(
     [
         dcc.Store(id='session-id-store', storage_type='session'),
         dcc.Store(id='session-store', storage_type='session'),
+        dcc.Download(id="download-zip-dcc"),
         dbc.Row(
             [
                 dbc.Col(
@@ -379,7 +380,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, de
                 html.Div(
                     dbc.Button(
                         "Download All (ZIP)", color="primary", size="lg", className='mb-3 mt-4',
-                        href=f"/download_zip/{session_id}/split_files.zip"
+                        id="download-zip-btn"
                     ),
                     style={'marginTop': '30px', 'display': 'block' if split_files else 'none'}
                 ),
@@ -456,7 +457,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, de
             split_files_list = get_split_files_ui(split_files, session_id)
             download_zip_btn = dbc.Button(
                 "Download All (ZIP)", color="primary", size="lg", className='mb-3 mt-4',
-                href=f"/download_zip/{session_id}/split_files.zip"
+                id="download-zip-btn"
             )
             results = html.Div([
                 html.H5("Split Files:"),
@@ -483,7 +484,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, de
         split_files_list = get_split_files_ui(split_files, session_id)
         download_zip_btn = dbc.Button(
             "Download All (ZIP)", color="primary", size="lg", className='mb-3 mt-4',
-            href=f"/download_zip/{session_id}/split_files.zip"
+            id="download-zip-btn"
         )
         results = html.Div([
             html.H5("Split Files:"),
@@ -508,6 +509,24 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, de
     logging.info(f"handle_upload: No action taken, returning current session_data for session {session_id}")
     return "", True, get_split_results_placeholder(), session_data
 
+@app.callback(
+    Output("download-zip-dcc", "data"),
+    Input("download-zip-btn", "n_clicks"),
+    State('session-id-store', 'data'),
+    prevent_initial_call=True
+)
+def trigger_download_zip(n_clicks, session_id):
+    if not n_clicks or not session_id:
+        return dash.no_update
+    session_dir = get_session_dir(session_id)
+    zip_path = os.path.join(session_dir, "split_files.zip")
+    if os.path.exists(zip_path):
+        logging.info(f"trigger_download_zip: Sending zip {zip_path} for session {session_id}")
+        return dcc.send_file(zip_path)
+    else:
+        logging.error(f"trigger_download_zip: Zip file not found for session {session_id}")
+        return dash.no_update
+
 @app.server.route('/download_zip/<session_id>/<filename>')
 def download_zip_file(session_id, filename):
     session_dir = get_session_dir(session_id)