Update app.py

app.py

@@ -88,10 +88,11 @@ app.layout = dbc.Container([
         ], width=3),
         dbc.Col([
             html.Div(style={"height": "20px"}),  # Small gap
+            html.Div(id="loading-output"),  # Text status box
             dcc.Loading(
                 id="loading-indicator",
                 type="dot",
-                children=[html.Div(id="loading-output")]
+                children=[html.Div()]  # Empty div for the blinking dots
             ),
             html.Div(id='matrix-preview', className="border p-3 mb-3"),
             dbc.Button("Download Matrix", id="btn-download", color="success", className="mt-3"),
@@ -213,24 +214,26 @@ Now, generate the {matrix_type}:
 @app.callback(
     Output('matrix-preview', 'children'),
     Output('loading-output', 'children'),
-    [Input(f'btn-{matrix_type.lower().replace(" ", "-")}', 'n_clicks') for matrix_type in matrix_types.keys()],
+    Input({'type': 'matrix-button', 'index': dash.ALL}, 'n_clicks'),
+    State('uploaded-files-state', 'children'),
     prevent_initial_call=True
 )
-def generate_matrix(*args):
+def generate_matrix(n_clicks, uploaded_files_json):
     global current_matrix, matrix_type
     ctx = dash.callback_context
     if not ctx.triggered:
         raise PreventUpdate
-    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
-    matrix_type = button_id.replace('btn-', '').replace('-', ' ').title()
     
-    if not uploaded_files:
+    button_id = ctx.triggered[0]['prop_id']
+    matrix_type = json.loads(button_id.split('.')[0])['index']
+    
+    if not uploaded_files_json:
         return html.Div("Please upload project artifacts before generating a matrix."), ""
 
-    file_contents = list(uploaded_files.values())
+    uploaded_files = json.loads(uploaded_files_json)
     
     try:
-        current_matrix = generate_matrix_with_gpt(matrix_type, file_contents)
+        current_matrix = generate_matrix_with_gpt(matrix_type, uploaded_files)
         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)}")