Update app.py via AI Editor

app.py

@@ -441,6 +441,7 @@ def update_uploaded_docs(
         Input('recover-action-btn', 'n_clicks'),
         Input('board-action-btn', 'n_clicks'),
         Input('loe-action-btn', 'n_clicks'),
+        Input('select-generated-dropdown', 'value'),
     ],
     State('chat-input', 'value'),
     State('select-document-dropdown', 'value'),
@@ -448,68 +449,74 @@ def update_uploaded_docs(
     State('select-generated-dropdown', 'value'),
     prevent_initial_call=True
 )
-def handle_actions(
-    shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks,
+def handle_all_actions(
+    shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, selected_generated_dropdown,
     chat_input, selected_filename, selected_proposal, selected_generated
 ):
     ctx = callback_context
     if not ctx.triggered:
         logging.info("No action triggered yet.")
         return html.Div("No action taken yet.", style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], selected_generated
-    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
-    logging.info(f"Button pressed: {button_id}")
-
-    result = ""
-    generated_docx_bytes = None
-    generated_docx_name = None
-    new_selected_generated = selected_generated
-
-    if button_id == 'shred-action-btn':
-        result, generated_docx_bytes, generated_docx_name = process_document('shred', selected_filename, chat_input)
-        if generated_docx_bytes and generated_docx_name:
-            generated_documents[generated_docx_name] = generated_docx_bytes
-            logging.info(f"Generated docx saved: {generated_docx_name}")
-            new_selected_generated = generated_docx_name
-    elif button_id == 'generate-action-btn':
-        result, _, _ = process_document('generate', selected_filename, chat_input)
-    elif button_id == 'compliance-action-btn':
-        result, _, _ = process_document('compliance', selected_filename, chat_input)
-    elif button_id == 'recover-action-btn':
-        result, _, _ = process_document('recover', selected_filename, chat_input)
-    elif button_id == 'board-action-btn':
-        result, _, _ = process_document('board', selected_filename, chat_input)
-    elif button_id == 'loe-action-btn':
-        result, _, _ = process_document('loe', selected_filename, chat_input)
-    else:
-        result = "Action not implemented yet."
-
-    if isinstance(result, str) and result.strip().startswith("Error"):
-        return html.Div(result, style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], new_selected_generated
-    if isinstance(result, str) and ("not implemented" in result or "No document uploaded" in result or "Shredding in progress" in result or "Generating response" in result or "Shredded document not available" in result):
-        return html.Div(result, style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], new_selected_generated
-    return dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], new_selected_generated
 
-@app.callback(
-    Output('output-data-upload', 'children'),
-    Input('select-generated-dropdown', 'value'),
-    prevent_initial_call=True
-)
-def display_generated_doc(selected_generated):
-    if not selected_generated or selected_generated not in generated_documents:
-        return html.Div("No generated document selected.", style={"wordWrap": "break-word"})
-    docx_bytes = generated_documents[selected_generated]
-    b64 = base64.b64encode(docx_bytes).decode('utf-8')
-    download_link = html.A(
-        f"Download {selected_generated}",
-        href=f"data:application/vnd.openxmlformats-officedocument.wordprocessingml.document;base64,{b64}",
-        download=selected_generated,
-        target="_blank",
-        style={"wordWrap": "break-word"}
-    )
-    return html.Div([
-        html.Div(download_link, style={"marginBottom": "15px"}),
-        html.Div("Preview not available for docx. Download to view.", style={"wordWrap": "break-word"})
-    ])
+    button_ids = [
+        'shred-action-btn', 'generate-action-btn', 'compliance-action-btn',
+        'recover-action-btn', 'board-action-btn', 'loe-action-btn'
+    ]
+    triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
+    logging.info(f"Trigger: {triggered_id}")
+
+    # if a button is pressed
+    if triggered_id in button_ids:
+        result = ""
+        generated_docx_bytes = None
+        generated_docx_name = None
+        new_selected_generated = selected_generated
+
+        if triggered_id == 'shred-action-btn':
+            result, generated_docx_bytes, generated_docx_name = process_document('shred', selected_filename, chat_input)
+            if generated_docx_bytes and generated_docx_name:
+                generated_documents[generated_docx_name] = generated_docx_bytes
+                logging.info(f"Generated docx saved: {generated_docx_name}")
+                new_selected_generated = generated_docx_name
+        elif triggered_id == 'generate-action-btn':
+            result, _, _ = process_document('generate', selected_filename, chat_input)
+        elif triggered_id == 'compliance-action-btn':
+            result, _, _ = process_document('compliance', selected_filename, chat_input)
+        elif triggered_id == 'recover-action-btn':
+            result, _, _ = process_document('recover', selected_filename, chat_input)
+        elif triggered_id == 'board-action-btn':
+            result, _, _ = process_document('board', selected_filename, chat_input)
+        elif triggered_id == 'loe-action-btn':
+            result, _, _ = process_document('loe', selected_filename, chat_input)
+        else:
+            result = "Action not implemented yet."
+
+        if isinstance(result, str) and result.strip().startswith("Error"):
+            return html.Div(result, style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], new_selected_generated
+        if isinstance(result, str) and ("not implemented" in result or "No document uploaded" in result or "Shredding in progress" in result or "Generating response" in result or "Shredded document not available" in result):
+            return html.Div(result, style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], new_selected_generated
+        return dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], new_selected_generated
+
+    # if the generated doc dropdown is changed
+    if triggered_id == 'select-generated-dropdown':
+        sel_gen = selected_generated_dropdown
+        if not sel_gen or sel_gen not in generated_documents:
+            return html.Div("No generated document selected.", style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], sel_gen
+        docx_bytes = generated_documents[sel_gen]
+        b64 = base64.b64encode(docx_bytes).decode('utf-8')
+        download_link = html.A(
+            f"Download {sel_gen}",
+            href=f"data:application/vnd.openxmlformats-officedocument.wordprocessingml.document;base64,{b64}",
+            download=sel_gen,
+            target="_blank",
+            style={"wordWrap": "break-word"}
+        )
+        return html.Div([
+            html.Div(download_link, style={"marginBottom": "15px"}),
+            html.Div("Preview not available for docx. Download to view.", style={"wordWrap": "break-word"})
+        ]), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], sel_gen
+
+    return html.Div("No action taken.", style={"wordWrap": "break-word"}), get_generated_doc_list(generated_documents), [{'label': fn, 'value': fn} for fn in generated_documents.keys()], selected_generated
 
 if __name__ == '__main__':
     print("Starting the Dash application...")