Update app.py via AI Editor

app.py

@@ -198,7 +198,7 @@ app.layout = dbc.Container(
     Input({'type': 'delete-upload-btn', 'index': ALL}, 'n_clicks'),
     Input('split-btn', 'n_clicks'),
     State('session-store', 'data'),
-    prevent_initial_call='initial_duplicate'
+    prevent_initial_call=True
 )
 def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, session_data):
     trigger = ctx.triggered_id
@@ -217,7 +217,15 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, se
         return "", "", resp_data
 
     # Handle Delete Upload (detect ANY delete button press)
+    delete_pressed = False
     if isinstance(trigger, dict) and trigger.get('type') == 'delete-upload-btn':
+        delete_pressed = True
+    # Also handle if delete_upload_n_list is not None and has a value >0
+    if not delete_pressed and delete_upload_n_list is not None and len(delete_upload_n_list) > 0:
+        if any(n is not None and n > 0 for n in delete_upload_n_list):
+            delete_pressed = True
+
+    if delete_pressed:
         orig_filename = session_data.get('orig_filename', '')
         pdf_path = os.path.join(session_dir, orig_filename)
         if os.path.exists(pdf_path):