Update app.py

app.py

@@ -168,10 +168,13 @@ def update_output(list_of_contents, list_of_names, existing_files):
             ]))
         if existing_files is None:
             existing_files = []
+        elif not isinstance(existing_files, list):
+            existing_files = [existing_files]
         shredded_document = None  # Reset shredded document when new files are uploaded
-        return existing_files + new_files, "Document uploaded. Please click 'Shred' to proceed."
-    return existing_files, "Please upload a document and click 'Shred' to begin."
-
+        updated_files = existing_files + new_files
+        return updated_files, f"{len(new_files)} document(s) uploaded. Please click 'Shred' to proceed."
+    return dash.no_update, "Please upload a document and click 'Shred' to begin."
+    
 @app.callback(
     Output('file-list', 'children', allow_duplicate=True),
     Output('status-bar', 'children', allow_duplicate=True),
@@ -223,19 +226,7 @@ Now, generate the {document_type}:
     response = model.generate_content(prompt)
     return response.text
 
-@app.callback(
-    Output('document-preview', 'children'),
-    Output('loading-output', 'children'),
-    Output('status-bar', 'children', allow_duplicate=True),
-    Output('document-upload', 'style'),
-    Output('document-choice', 'style'),
-    Output('btn-generate-document', 'style'),
-    [Input(f'btn-{doc_type.lower().replace("_", "-")}', 'n_clicks') for doc_type in document_types.keys()],
-    Input('btn-generate-document', 'n_clicks'),
-    State('document-source', 'value'),
-    State('specific-document-name', 'children'),
-    prevent_initial_call=True
-)
+
 @app.callback(
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),