Update app.py

app.py

@@ -124,32 +124,6 @@ def parse_file_content(contents, filename):
         print(f"Error processing file {filename}: {str(e)}")
         return "Error processing file"
 
-@app.callback(
-    Output('file-list', 'children', allow_duplicate=True),
-    Input({'type': 'remove-file', 'index': dash.ALL}, 'n_clicks'),
-    State('file-list', 'children'),
-    prevent_initial_call=True
-)
-def remove_file(n_clicks, existing_files):
-    global uploaded_files
-    ctx = dash.callback_context
-    if not ctx.triggered:
-        raise PreventUpdate
-    
-    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
-    removed_file = button_id.split('"index":"')[1].split('"')[0]
-    
-    if removed_file in uploaded_files:
-        del uploaded_files[removed_file]
-    
-    updated_files = [
-        file for file in existing_files 
-        if file['props']['children'][1]['props']['children'] != removed_file
-    ]
-    
-    return updated_files
-
-# Modify the update_output function to include the correct button structure
 @app.callback(
     Output('file-list', 'children'),
     Input('upload-files', 'contents'),
@@ -164,8 +138,7 @@ def update_output(list_of_contents, list_of_names, existing_files):
             file_content = parse_file_content(content, name)
             uploaded_files[name] = file_content
             new_files.append(html.Div([
-                html.Button('×', id={'type': 'remove-file', 'index': name}, 
-                            style={'marginRight': '5px', 'fontSize': '10px', 'border': 'none', 'background': 'none', 'cursor': 'pointer'}),
+                html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
                 html.Span(name)
             ]))
         if existing_files is None:
@@ -173,6 +146,21 @@ def update_output(list_of_contents, list_of_names, existing_files):
         return existing_files + new_files
     return existing_files
 
+@app.callback(
+    Output('file-list', 'children', allow_duplicate=True),
+    Input({'type': 'remove-file', 'index': dash.ALL}, 'n_clicks'),
+    State('file-list', 'children'),
+    prevent_initial_call=True
+)
+def remove_file(n_clicks, existing_files):
+    global uploaded_files
+    ctx = dash.callback_context
+    if not ctx.triggered:
+        raise PreventUpdate
+    removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
+    uploaded_files.pop(removed_file, None)
+    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file]
+
 def generate_matrix_with_gpt(matrix_type, file_contents):
     prompt = f"""Generate a {matrix_type} based on the following project artifacts:
 {' '.join(file_contents)}