Update app.py

app.py

@@ -188,9 +188,11 @@ def remove_file(n_clicks, existing_files):
     if not ctx.triggered:
         raise dash.exceptions.PreventUpdate
     removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
-    uploaded_files.pop(removed_file, None)
-    shredded_document = None  # Reset shredded document when a file is removed
-    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file], "Document removed. Please upload a document and click 'Shred' to begin."
+    if removed_file in uploaded_files:
+        uploaded_files.pop(removed_file, None)
+        shredded_document = None  # Reset shredded document when a file is removed
+        return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file], f"Document '{removed_file}' removed."
+    return dash.no_update, dash.no_update
 
 def generate_document(document_type, file_contents):
     prompt = f"""Generate a {document_type} based on the following project artifacts:
@@ -226,7 +228,6 @@ Now, generate the {document_type}:
     response = model.generate_content(prompt)
     return response.text
 
-
 @app.callback(
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
@@ -253,11 +254,11 @@ def generate_document_preview(*args):
     if triggered_id.startswith('btn-'):
         document_type = triggered_id.replace('btn-', '').replace('-', ' ').title()
         if document_type == "Shred":
-            return html.Div("Ready to shred. Click 'Generate Document' to proceed."), "", "Ready to shred.", {'display': 'none'}, {'display': 'none'}, {'display': 'block'}
+            return html.Div("Ready to shred. Click 'Generate Document' to proceed."), "", dash.no_update, {'display': 'none'}, {'display': 'none'}, {'display': 'block'}
         elif document_type in ["Pink", "Red", "Gold"]:
-            return html.Div(f"Ready to generate {document_type} document."), "", f"Ready to generate {document_type} document.", {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
+            return html.Div(f"Ready to generate {document_type} document."), "", dash.no_update, {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
         elif document_type in ["Pink Review", "Red Review", "Gold Review", "Virtual Board", "Loe"]:
-            return html.Div(f"Ready to generate {document_type}."), "", f"Ready to generate {document_type}.", {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
+            return html.Div(f"Ready to generate {document_type}."), "", dash.no_update, {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
         return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update
 
     if triggered_id == 'btn-generate-document':