Update app.py via AI Editor

app.py

@@ -158,6 +158,36 @@ app.title = "Intelligent PDF Splitter"
 def get_split_results_placeholder():
     return html.Div("", id="split-results-inner")
 
+def get_split_files_ui(split_files, session_id):
+    items = []
+    for i, fi in enumerate(split_files):
+        fname = fi['filename']
+        size = fi['size']
+        download_link = html.A(
+            f"Download {fname} ({size:.2f} MB)",
+            href=f"/download_split/{session_id}/{fname}",
+            target="_blank",
+            style={'marginRight': '16px'}
+        )
+        delete_btn = dbc.Button(
+            "Delete",
+            id={'type': 'delete-split-btn', 'index': i},
+            color='danger',
+            size='sm',
+            className='ms-3',
+            n_clicks=0
+        )
+        items.append(
+            html.Li(
+                dbc.Row([
+                    dbc.Col(download_link, width=9, style={'display': 'flex', 'alignItems': 'center'}),
+                    dbc.Col(delete_btn, width=3, style={'display': 'flex', 'justifyContent': 'end'})
+                ], align='center'),
+                style={'marginBottom': '10px'}
+            )
+        )
+    return html.Ul(items)
+
 app.layout = dbc.Container(
     [
         dcc.Store(id='session-id-store', storage_type='session'),
@@ -253,11 +283,12 @@ def ensure_session_id(session_id):
     Input('clear-session', 'n_clicks'),
     Input({'type': 'delete-upload-btn', 'index': ALL}, 'n_clicks'),
     Input('split-btn', 'n_clicks'),
+    Input({'type': 'delete-split-btn', 'index': ALL}, 'n_clicks'),
     State('session-store', 'data'),
     State('session-id-store', 'data'),
     prevent_initial_call=True
 )
-def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, session_data, session_id):
+def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, delete_split_n_list, session_data, session_id):
     trigger = ctx.triggered_id
     logging.info(f"handle_upload: Triggered by {trigger}, session_id={session_id}")
 
@@ -300,6 +331,63 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
         logging.info(f"Session files deleted for {session_id}")
         return "", True, get_split_results_placeholder(), {}
 
+    # Handle Delete Split File
+    delete_split_pressed = False
+    delete_split_idx = None
+    if isinstance(trigger, dict) and trigger.get('type') == 'delete-split-btn':
+        delete_split_pressed = True
+        delete_split_idx = trigger.get('index')
+    if not delete_split_pressed and delete_split_n_list is not None and len(delete_split_n_list) > 0:
+        for i, n in enumerate(delete_split_n_list):
+            if n is not None and n > 0:
+                delete_split_pressed = True
+                delete_split_idx = i
+                break
+    if delete_split_pressed and session_data.get('split_files'):
+        split_files = session_data['split_files']
+        if 0 <= delete_split_idx < len(split_files):
+            del_file = split_files[delete_split_idx]
+            file_path = del_file['path']
+            if os.path.exists(file_path):
+                os.remove(file_path)
+                logging.info(f"Deleted split file: {file_path} for session {session_id}")
+            split_files = [f for i, f in enumerate(split_files) if i != delete_split_idx]
+            session_data['split_files'] = split_files
+            # Recreate ZIP if any splits remain, else remove ZIP
+            zip_path = os.path.join(session_dir, "split_files.zip")
+            if split_files:
+                make_zip_of_splits(split_files, session_dir)
+                session_data['zip_ready'] = True
+                logging.info(f"Regenerated ZIP file after split delete for session {session_id}")
+            else:
+                if os.path.exists(zip_path):
+                    os.remove(zip_path)
+                    logging.info(f"Deleted ZIP file as no splits remain for session {session_id}")
+                session_data['zip_ready'] = False
+            # Build UI
+            orig_filename = session_data.get('orig_filename', '')
+            file_info = dbc.Row([
+                dbc.Col(html.Div(f"Uploaded: {orig_filename}"), width=9, style={'display': 'flex', 'alignItems': 'center'}),
+                dbc.Col(
+                    dbc.Button("Delete", id={'type': 'delete-upload-btn', 'index': 0}, color='danger', n_clicks=0, className='ms-5'),
+                    width=3, style={'display': 'flex', 'justifyContent': 'end'}
+                )
+            ], className='mb-3', align='center', style={'marginTop': "15px", 'marginBottom': '25px'})
+            results = html.Div([
+                html.H5("Split Files:"),
+                get_split_files_ui(split_files, session_id) if split_files else html.Div("No split files remain."),
+                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"
+                    ),
+                    style={'marginTop': '30px', 'display': 'block' if split_files else 'none'}
+                ),
+            ], id="split-results-inner")
+            return file_info, False, results, session_data
+        else:
+            logging.warning(f"Split file delete index {delete_split_idx} invalid for session {session_id}")
+
     # Handle Upload
     if trigger == 'upload-pdf':
         logging.info(f"handle_upload: Upload triggered for filename={filename}, session_id={session_id}")
@@ -363,10 +451,11 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
         try:
             logging.info(f"Splitting PDF for session {session_id}. File: {pdf_path}")
             with lock:
-                logging.info(f"handle_upload: Acquired lock for session {session_id}, starting split.")
                 split_files = intelligent_pdf_split(pdf_path, session_dir)
+                for fi in split_files:
+                    logging.info(f"Split file saved: {fi['path']} ({fi['size']:.2f} MB)")
                 zip_path = make_zip_of_splits(split_files, session_dir)
-                logging.info(f"handle_upload: Split/ZIP finished for {session_id}, zip_path={zip_path}")
+                logging.info(f"Split/ZIP finished for {session_id}, zip_path={zip_path}")
             session_data['split_files'] = split_files
             session_data['zip_ready'] = True
             file_info = dbc.Row([
@@ -376,11 +465,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
                     width=3, style={'display': 'flex', 'justifyContent': 'end'}
                 )
             ], className='mb-3', align='center', style={'marginTop': "15px", 'marginBottom': '25px'})
-            split_files_list = html.Ul([
-                html.Li([
-                    f"{fi['filename']} ({fi['size']:.2f} MB)"
-                ]) for fi in split_files
-            ])
+            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"
@@ -407,11 +492,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
                 width=3, style={'display': 'flex', 'justifyContent': 'end'}
             )
         ], className='mb-3', align='center', style={'marginTop': "15px", 'marginBottom': '25px'})
-        split_files_list = html.Ul([
-            html.Li([
-                f"{fi['filename']} ({fi['size']:.2f} MB)"
-            ]) for fi in split_files
-        ])
+        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"
@@ -438,6 +519,17 @@ def download_zip_file(session_id, filename):
         logging.error(f"ZIP file not found for download: {file_path}")
         return "File not found", 404
 
+@app.server.route('/download_split/<session_id>/<filename>')
+def download_split_file(session_id, filename):
+    session_dir = get_session_dir(session_id)
+    file_path = os.path.join(session_dir, filename)
+    if os.path.exists(file_path):
+        logging.info(f"Serving split file {file_path} for session {session_id}")
+        return send_file(file_path, mimetype='application/pdf', as_attachment=True, download_name=filename)
+    else:
+        logging.error(f"Split file not found for download: {file_path}")
+        return "File not found", 404
+
 if __name__ == '__main__':
     print("Starting the Dash application...")
     app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)