Update app.py via AI Editor

app.py

@@ -1,6 +1,6 @@
 import dash
 import dash_bootstrap_components as dbc
-from dash import html, dcc, Input, Output, State, ctx
+from dash import html, dcc, Input, Output, State, ctx, ALL
 import flask
 import uuid
 import os
@@ -14,7 +14,6 @@ import re
 import zipfile
 import base64
 
-# Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
 
 SESSION_DATA = {}
@@ -196,12 +195,12 @@ app.layout = dbc.Container(
     Input('upload-pdf', 'contents'),
     State('upload-pdf', 'filename'),
     Input('clear-session', 'n_clicks'),
-    Input({'type': 'delete-upload-btn', 'index': 0}, 'n_clicks'),
+    Input({'type': 'delete-upload-btn', 'index': ALL}, 'n_clicks'),
     Input('split-btn', 'n_clicks'),
     State('session-store', 'data'),
     prevent_initial_call='initial_duplicate'
 )
-def handle_upload(contents, filename, clear_n, delete_upload_n, split_n, session_data):
+def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, session_data):
     trigger = ctx.triggered_id
     session_id = get_session_id()
     flask.g.session_id = session_id
@@ -211,12 +210,14 @@ def handle_upload(contents, filename, clear_n, delete_upload_n, split_n, session
     if session_data is None:
         session_data = {}
 
+    # Handle Clear Session
     if trigger == 'clear-session':
         clean_session(session_id)
         resp_data = {}
         return "", "", resp_data
 
-    if trigger == {'type': 'delete-upload-btn', 'index': 0}:
+    # Handle Delete Upload (detect ANY delete button press)
+    if isinstance(trigger, dict) and trigger.get('type') == 'delete-upload-btn':
         orig_filename = session_data.get('orig_filename', '')
         pdf_path = os.path.join(session_dir, orig_filename)
         if os.path.exists(pdf_path):
@@ -227,6 +228,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n, split_n, session
                 os.remove(os.path.join(session_dir, file))
         return "", "", {}
 
+    # Handle Upload
     if trigger == 'upload-pdf':
         if not contents:
             return "", "", {}
@@ -265,8 +267,8 @@ def handle_upload(contents, filename, clear_n, delete_upload_n, split_n, session
             logging.error(f"Error processing PDF: {e}")
             return html.Div(f"Error: {e}", style={'color': 'red'}), "", {}
 
+    # Restore after upload (before split)
     if session_data.get('orig_filename') and not session_data.get('split_files'):
-        # If user refreshes after upload but before split, restore file info and split button
         file_info = dbc.Row([
             dbc.Col(html.Div(f"Uploaded: {session_data['orig_filename']}"), width=9, style={'display': 'flex', 'alignItems': 'center'}),
             dbc.Col(
@@ -282,6 +284,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n, split_n, session
         ])
         return file_info, split_results, session_data
 
+    # Handle Split
     if trigger == 'split-btn':
         orig_filename = session_data.get('orig_filename')
         if not orig_filename:
@@ -322,7 +325,7 @@ def handle_upload(contents, filename, clear_n, delete_upload_n, split_n, session
             logging.error(f"Error splitting PDF: {e}")
             return html.Div(f"Error: {e}", style={'color': 'red'}), "", session_data
 
-    # Restore split results if user refreshes after splitting
+    # Restore after split
     if session_data.get('split_files'):
         split_files = session_data['split_files']
         orig_filename = session_data.get('orig_filename', '')