Update app.py via AI Editor

app.py

@@ -425,18 +425,6 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, de
             logging.error(f"Error processing PDF: {e}")
             return html.Div(f"Error: {e}", style={'color': 'red'}), True, get_split_results_placeholder(), {}
 
-    # Restore after upload (before split)
-    if session_data.get('orig_filename') and not session_data.get('split_files'):
-        file_info = dbc.Row([
-            dbc.Col(html.Div(f"Uploaded: {session_data['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'})
-        logging.info(f"handle_upload: Restoring view after upload, split button enabled.")
-        return file_info, False, get_split_results_placeholder(), session_data
-
     # Handle Split
     if trigger == 'split-btn':
         orig_filename = session_data.get('orig_filename')
@@ -505,6 +493,18 @@ def handle_upload(contents, filename, clear_n, delete_upload_n_list, split_n, de
         logging.info(f"handle_upload: Restoring split results for session {session_id}, {len(split_files)} files.")
         return file_info, False, results, session_data
 
+    # Restore after upload (before split)
+    if session_data.get('orig_filename') and not session_data.get('split_files'):
+        file_info = dbc.Row([
+            dbc.Col(html.Div(f"Uploaded: {session_data['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'})
+        logging.info(f"handle_upload: Restoring view after upload, split button enabled.")
+        return file_info, False, get_split_results_placeholder(), session_data
+
     logging.info(f"handle_upload: No action taken, returning current session_data for session {session_id}")
     return "", True, get_split_results_placeholder(), session_data