Update app.py via AI Editor

app.py

@@ -56,7 +56,11 @@ def get_right_col_content(selected_type):
     controls.append(dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"))
     controls.append(dcc.Download(id="download-document"))
     controls.append(html.Hr())
-    if selected_type != "Shred":
+    if selected_type == "Shred":
+        controls.append(
+            dbc.Button("Generate Document", id={'type': 'btn-generate-doc', 'index': selected_type}, color="primary", className="mb-3")
+        )
+    elif selected_type != "Shred":
         controls.append(html.Div([
             html.Label(f"Upload {selected_type} Document"),
             dcc.Upload(
@@ -248,12 +252,10 @@ def remove_file(n_clicks, existing_files):
     ctx = dash.callback_context
     if not ctx.triggered:
         raise dash.exceptions.PreventUpdate
-    # Only proceed if a remove button was actually clicked
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
     if not triggered_id.startswith("{\"type\":\"remove-file\""):
         raise dash.exceptions.PreventUpdate
     try:
-        # Parse the file name from the triggered id
         import json
         triggered_dict = json.loads(triggered_id)
         removed_file = triggered_dict['index']
@@ -263,11 +265,9 @@ def remove_file(n_clicks, existing_files):
     uploaded_files.pop(removed_file, None)
     shredded_document = None
     logging.info(f"Removed file: {removed_file}")
-    # Filter out the removed file from the list
     filtered_files = []
     if existing_files:
         for file in existing_files:
-            # Defensive: file['props']['children'][1]['props']['children'] is the filename
             try:
                 file_name = file['props']['children'][1]['props']['children']
                 if file_name != removed_file: