Update app.py via AI Editor

app.py

@@ -64,12 +64,10 @@ def process_document(action, selected_filename=None, chat_input=None):
     global shredded_document, generated_response
     logging.info(f"Process document called with action: {action}")
 
-    # Select document content to use, if needed
     doc_content = None
     if selected_filename and selected_filename in uploaded_documents:
         doc_content = uploaded_documents[selected_filename]
     elif uploaded_documents:
-        # Default to the first document if not specified
         doc_content = next(iter(uploaded_documents.values()))
         selected_filename = next(iter(uploaded_documents.keys()))
     else:
@@ -261,13 +259,11 @@ def update_uploaded_docs(content, filename, delete_clicks, children, selected_do
     if delete_clicks:
         for i, n_click in enumerate(delete_clicks):
             if n_click:
-                # The delete button's id is a dict with 'index' = filename
                 btn_id = ctx.inputs_list[2][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in uploaded_documents:
                     del uploaded_documents[del_filename]
                     logging.info(f"Document deleted: {del_filename}")
-                    # If the selected doc is deleted, choose another
                     if selected_doc == del_filename:
                         selected_doc = next(iter(uploaded_documents), None)
                     break
@@ -290,7 +286,6 @@ def handle_upload(content, filename):
         if text is None:
             return html.Div("Error: Could not decode document. Please upload a valid text file.", style={"wordWrap": "break-word"})
         else:
-            # Already handled and added in update_uploaded_docs
             return html.Div(f"Document '{filename}' uploaded successfully.", style={"wordWrap": "break-word"})
     return ""
 
@@ -312,22 +307,29 @@ def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_cli
     ctx = callback_context
     if not ctx.triggered:
         logging.info("No action triggered yet.")
-        return "No action taken yet."
+        return html.Div("No action taken yet.", style={"wordWrap": "break-word"})
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
     logging.info(f"Button pressed: {button_id}")
+    result = ""
     if button_id == 'shred-action-btn':
-        return process_document('shred', selected_filename, chat_input)
+        result = process_document('shred', selected_filename, chat_input)
     elif button_id == 'generate-action-btn':
-        return process_document('generate', selected_filename, chat_input)
+        result = process_document('generate', selected_filename, chat_input)
     elif button_id == 'compliance-action-btn':
-        return process_document('compliance', selected_filename, chat_input)
+        result = process_document('compliance', selected_filename, chat_input)
     elif button_id == 'recover-action-btn':
-        return process_document('recover', selected_filename, chat_input)
+        result = process_document('recover', selected_filename, chat_input)
     elif button_id == 'board-action-btn':
-        return process_document('board', selected_filename, chat_input)
+        result = process_document('board', selected_filename, chat_input)
     elif button_id == 'loe-action-btn':
-        return process_document('loe', selected_filename, chat_input)
-    return "Action not implemented yet."
+        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"})
+    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"})
+    return dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
 
 if __name__ == '__main__':
     print("Starting the Dash application...")