Update app.py

app.py

@@ -12,6 +12,7 @@ import PyPDF2
 import docx
 import chardet
 import json
+from dash.exceptions import PreventUpdate
 
 # Initialize the Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
@@ -147,17 +148,17 @@ def update_output(list_of_contents, list_of_names, existing_files):
     global uploaded_files
     if list_of_contents is not None:
         new_files = []
-        for i, (content, name) in enumerate(zip(list_of_contents, list_of_names)):
+        for content, name in zip(list_of_contents, list_of_names):
             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'}),
+            new_files.append(dbc.ListGroupItem([
+                dbc.Button("×", id={'type': 'remove-file', 'index': name}, 
+                           color="danger", size="sm", className="me-2"),
                 html.Span(name)
             ]))
-        if existing_files is None:
-            existing_files = []
+        existing_files = existing_files or []
         return existing_files + new_files
-    return existing_files
+    return existing_files or []
 
 @app.callback(
     Output('file-list', 'children', allow_duplicate=True),
@@ -170,9 +171,12 @@ def remove_file(n_clicks, existing_files):
     ctx = dash.callback_context
     if not ctx.triggered:
         raise PreventUpdate
-    removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
+    
+    triggered_id = ctx.triggered[0]['prop_id']
+    removed_file = json.loads(triggered_id.split('.')[0])['index']
+    
     uploaded_files.pop(removed_file, None)
-    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file]
+    return [file for file in existing_files if file['props']['children'][1]['children'] != removed_file]
 
 def generate_matrix_with_gpt(matrix_type, file_contents):
     prompt = f"""Generate a {matrix_type} based on the following project artifacts:
@@ -212,6 +216,14 @@ Now, generate the {matrix_type}:
 
     return pd.DataFrame(data, columns=headers)
 
+@app.callback(
+    Output('uploaded-files-state', 'children'),
+    Input('file-list', 'children')
+)
+def update_uploaded_files_state(file_list):
+    global uploaded_files
+    return json.dumps(list(uploaded_files.keys()))
+
 @app.callback(
     Output('matrix-preview', 'children'),
     Output('loading-output', 'children'),
@@ -231,10 +243,13 @@ def generate_matrix(n_clicks, uploaded_files_json):
     if not uploaded_files_json:
         return html.Div("Please upload project artifacts before generating a matrix."), ""
 
-    uploaded_files = json.loads(uploaded_files_json)
+    uploaded_files_list = json.loads(uploaded_files_json)
+    
+    if not uploaded_files_list:
+        return html.Div("Please upload project artifacts before generating a matrix."), ""
     
     try:
-        current_matrix = generate_matrix_with_gpt(matrix_type, uploaded_files)
+        current_matrix = generate_matrix_with_gpt(matrix_type, [uploaded_files[file] for file in uploaded_files_list])
         return dbc.Table.from_dataframe(current_matrix, striped=True, bordered=True, hover=True), f"{matrix_type} generated"
     except Exception as e:
         print(f"Error generating matrix: {str(e)}")