Update app.py via AI Editor

app.py

@@ -352,15 +352,15 @@ def store_session_id(session_id):
     Input('upload-file', 'contents'),
     Input('generate-content-button', 'n_clicks'),
     Input('generate-ppt-button', 'n_clicks'),
-    Input({'type': 'file-download', 'index': dash.ALL}, 'n_clicks'),
-    Input({'type': 'file-delete', 'index': dash.ALL}, 'n_clicks'),
+    Input({'type': 'file-download', 'filename': dash.ALL}, 'n_clicks'),
+    Input({'type': 'file-delete', 'filename': dash.ALL}, 'n_clicks'),
     Input('slides-content', 'value'),
-    Input('session-id', 'data'), # This input is used to restore state on refresh
+    Input('session-id', 'data'),
     State('upload-file', 'filename'),
     State('document-text', 'value'),
     State('current-slide-index', 'data'),
-    State({'type': 'file-download', 'index': dash.ALL}, 'id'),
-    State({'type': 'file-delete', 'index': dash.ALL}, 'id'),
+    State({'type': 'file-download', 'filename': dash.ALL}, 'id'),
+    State({'type': 'file-delete', 'filename': dash.ALL}, 'id'),
     State('slides-content-store', 'data'),
     prevent_initial_call=False
 )
@@ -369,7 +369,6 @@ def unified_callback(
     upload_filename, document_text, current_slide_idx, download_ids, delete_ids, slides_content_store_data
 ):
     ctx = callback_context
-    # On initial app load or refresh, handle state restoration
     if not ctx.triggered or not session_id:
         session = get_or_create_session_data(session_id)
         session['files_box'] = list_session_files(session_id)
@@ -401,14 +400,14 @@ def unified_callback(
 
     def build_files_box(session):
         files_box = []
-        for idx, f in enumerate(session['files_box']):
+        for f in session['files_box']:
             files_box.append(
                 html.Div([
                     html.Div(
                         html.A(
                             f['name'],
                             href="#",
-                            id={'type': 'file-download', 'index': idx},
+                            id={'type': 'file-download', 'filename': f['name']},
                             n_clicks=0,
                             style={
                                 'color': '#116F70',
@@ -430,12 +429,12 @@ def unified_callback(
                     ),
                     dbc.Button(
                         "Delete",
-                        id={'type': 'file-delete', 'index': idx},
+                        id={'type': 'file-delete', 'filename': f['name']},
                         size="sm",
                         color="secondary",
                         style={'marginLeft': '8px', 'marginBottom': '2px', 'flex': 'none'}
                     )
-                ], style={'display': 'flex', 'alignItems': 'center', 'marginBottom': '8px', 'width': '100%', 'gap': '0.2em'}, id={'type': 'file-row', 'index': idx})
+                ], style={'display': 'flex', 'alignItems': 'center', 'marginBottom': '8px', 'width': '100%', 'gap': '0.2em'}, id={'type': 'file-row', 'filename': f['name']})
             )
         if not files_box:
             return html.Div("No files uploaded or generated yet.", className="text-muted")
@@ -527,37 +526,34 @@ def unified_callback(
     elif button_id.startswith("{"):
         triggered_id = eval(button_id)
         if triggered_id['type'] == 'file-download':
-            idx = triggered_id['index']
-            with session['lock']:
-                if 0 <= idx < len(session['files_box']):
-                    file_path = session['files_box'][idx]['path']
-                    file_name = session['files_box'][idx]['name']
-                    file_bytes = read_session_temp_file(file_path)
-                    add_log(f"Download link generated for file '{file_name}' at '{file_path}'", session_id)
-                    logger.info(f"Download triggered for file '{file_name}' at '{file_path}' session '{session_id}'")
-                    log_str = '\n'.join(session['log_messages'][:100])
-                    return build_files_box(session), slides_content, log_str, slide_idx, dcc.send_bytes(file_bytes, file_name), slides_content
+            filename = triggered_id['filename']
+            file_entry = next((f for f in session['files_box'] if f['name'] == filename), None)
+            if file_entry:
+                file_path = file_entry['path']
+                file_name = file_entry['name']
+                file_bytes = read_session_temp_file(file_path)
+                add_log(f"Download link generated for file '{file_name}' at '{file_path}'", session_id)
+                logger.info(f"Download triggered for file '{file_name}' at '{file_path}' session '{session_id}'")
+                log_str = '\n'.join(session['log_messages'][:100])
+                return build_files_box(session), slides_content, log_str, slide_idx, dcc.send_bytes(file_bytes, file_name), slides_content
         elif triggered_id['type'] == 'file-delete':
-            idx = triggered_id['index']
-            with session['lock']:
-                if 0 <= idx < len(session['files_box']):
-                    deleted = session['files_box'].pop(idx)
-                    if os.path.exists(deleted['path']):
-                        try:
-                            os.remove(deleted['path'])
-                            add_log(f"Deleted file '{deleted['name']}' from '{deleted['path']}'", session_id)
-                            logger.info(f"Deleted file '{deleted['name']}' at '{deleted['path']}' for session '{session_id}'")
-                        except Exception as ex:
-                            add_log(f"Failed to delete file '{deleted['name']}' from '{deleted['path']}': {str(ex)}", session_id)
-                    session['files_box'] = list_session_files(session_id)
-                    session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - File '{deleted['name']}' deleted.")
-                    log_str = '\n'.join(session['log_messages'][:100])
+            filename = triggered_id['filename']
+            file_entry = next((f for f in session['files_box'] if f['name'] == filename), None)
+            if file_entry:
+                try:
+                    os.remove(file_entry['path'])
+                    add_log(f"Deleted file '{file_entry['name']}' from '{file_entry['path']}'", session_id)
+                    logger.info(f"Deleted file '{file_entry['name']}' at '{file_entry['path']}' for session '{session_id}'")
+                except Exception as ex:
+                    add_log(f"Failed to delete file '{file_entry['name']}' from '{file_entry['path']}': {str(ex)}", session_id)
+                session['files_box'] = list_session_files(session_id)
+                session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - File '{file_entry['name']}' deleted.")
+                log_str = '\n'.join(session['log_messages'][:100])
             return build_files_box(session), slides_content, log_str, slide_idx, None, slides_content
         else:
             raise PreventUpdate
 
     else:
-        # If unknown trigger, restore state (should not happen)
         session['files_box'] = list_session_files(session_id)
         session['current_slide_md'] = load_current_slide_md_from_disk(session_id)
         slides_content = session.get('current_slide_md', "")