Update app.py

app.py

@@ -11,6 +11,7 @@ from dash.exceptions import PreventUpdate
 import PyPDF2
 import docx
 import chardet
+import json
 
 # Initialize the Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
@@ -124,6 +125,32 @@ 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 = json.loads(button_id)['index']
+    
+    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'),
@@ -138,7 +165,8 @@ 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'}),
+                html.Button('×', id={'type': 'remove-file', 'index': name}, 
+                            style={'marginRight': '5px', 'fontSize': '10px', 'border': 'none', 'background': 'none', 'cursor': 'pointer'}),
                 html.Span(name)
             ]))
         if existing_files is None:
@@ -146,21 +174,6 @@ 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)}