Update app.py via AI Editor

app.py

@@ -26,6 +26,7 @@ CLAUDE3_MAX_OUTPUT_TOKENS = 64_000
 uploaded_documents = {}
 uploaded_proposals = {}
 generated_documents = {}
+shredded_documents = {}
 shredded_document = None
 generated_response = None
 
@@ -112,7 +113,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         if chat_input:
             prompt += f"User additional instructions: {chat_input}\n"
         prompt += f"\nFile Name: {selected_filename}\n\n{doc_content}"
-        
+
         result_holder = {"text": None, "docx_bytes": None, "docx_name": None}
         def thread_shred():
             global shredded_document
@@ -256,6 +257,28 @@ def get_uploaded_doc_list(docdict):
         )
     return dbc.ListGroup(doc_list, flush=True)
 
+def get_shredded_doc_list(shreddict):
+    if not shreddict:
+        return html.Div("No shredded requirements yet.", style={"wordWrap": "break-word"})
+    doc_list = []
+    for filename in shreddict:
+        b64 = base64.b64encode(shreddict[filename]).decode('utf-8')
+        download_link = html.A(
+            f"Download {filename}",
+            href=f"data:application/vnd.openxmlformats-officedocument.wordprocessingml.document;base64,{b64}",
+            download=filename,
+            target="_blank",
+            style={"wordWrap": "break-word", "marginRight": "10px"}
+        )
+        doc_list.append(
+            dbc.ListGroupItem([
+                html.Span(filename, style={"wordWrap": "break-word", "marginRight": "10px"}),
+                download_link,
+                dbc.Button("Delete", id={'type': 'delete-shredded-btn', 'index': filename, 'group': 'shredded'}, size="sm", color="danger", className="float-end ms-2")
+            ], className="d-flex justify-content-between align-items-center")
+        )
+    return dbc.ListGroup(doc_list, flush=True)
+
 def get_uploaded_proposal_list(docdict):
     if not docdict:
         return html.Div("No proposal documents uploaded.", style={"wordWrap": "break-word"})
@@ -313,7 +336,10 @@ app.layout = dbc.Container([
                             'margin': '10px'
                         },
                         multiple=False
-                    )
+                    ),
+                    html.Hr(),
+                    html.H6("Shredded Requirements"),
+                    html.Div(get_shredded_doc_list(shredded_documents), id='shredded-doc-list')
                 ])
             ], className="mb-3"),
 
@@ -401,6 +427,7 @@ app.layout = dbc.Container([
     Output('generated-doc-list', 'children'),
     Output('select-generated-dropdown', 'options'),
     Output('select-generated-dropdown', 'value'),
+    Output('shredded-doc-list', 'children'),
     Output('output-data-upload', 'children'),
     [
         Input('upload-document', 'contents'),
@@ -414,6 +441,8 @@ app.layout = dbc.Container([
         Input({'type': 'delete-generated-btn', 'index': dash.ALL, 'group': 'generated'}, 'n_clicks'),
         State('select-generated-dropdown', 'value'),
         State('select-generated-dropdown', 'options'),
+        Input({'type': 'delete-shredded-btn', 'index': dash.ALL, 'group': 'shredded'}, 'n_clicks'),
+        State('shredded-doc-list', 'children'),
         Input('shred-action-btn', 'n_clicks'),
         Input('proposal-action-btn', 'n_clicks'),
         Input('compliance-action-btn', 'n_clicks'),
@@ -432,6 +461,7 @@ def master_callback(
     rfp_content, rfp_filename, rfp_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
     generated_delete_clicks, selected_generated, generated_options,
+    shredded_delete_clicks, shredded_doc_children,
     shred_clicks, proposal_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks,
     selected_generated_dropdown,
     chat_input, selected_filename, selected_proposal_dropdown, selected_generated_dropdown_state
@@ -502,12 +532,24 @@ def master_callback(
                     upload_triggered = True
                     break
 
+    if triggered_id and isinstance(ctx.inputs_list[12], list):
+        for i, n_click in enumerate(shredded_delete_clicks):
+            if n_click:
+                btn_id = ctx.inputs_list[12][i]['id']
+                del_filename = btn_id['index']
+                if del_filename in shredded_documents:
+                    del shredded_documents[del_filename]
+                    logging.info(f"Shredded doc deleted: {del_filename}")
+                upload_triggered = True
+                break
+
     doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
     doc_value = selected_doc if selected_doc in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
     proposal_options = [{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()]
     proposal_value = selected_proposal if selected_proposal in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
     generated_doc_options = [{'label': fn, 'value': fn} for fn in generated_documents.keys()]
     generated_doc_value = selected_generated if selected_generated in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
+    shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
 
     uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
     uploaded_proposal_list = get_uploaded_proposal_list(uploaded_proposals)
@@ -520,7 +562,6 @@ def master_callback(
         'recover-action-btn', 'board-action-btn', 'loe-action-btn'
     ]
 
-    # Fix: When "Proposal" is clicked, trigger proposal streaming with correct dropdown doc selections
     if triggered_id in action_buttons:
         result = ""
         generated_docx_bytes = None
@@ -528,14 +569,14 @@ def master_callback(
         new_selected_generated = generated_doc_value
 
         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
+            result, shredded_docx_bytes, shredded_docx_name = process_document('shred', selected_filename, chat_input)
+            if shredded_docx_bytes and shredded_docx_name:
+                shredded_documents[shredded_docx_name] = shredded_docx_bytes
+                logging.info(f"Shredded docx saved: {shredded_docx_name}")
+            shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
+            output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
 
         elif triggered_id == 'proposal-action-btn':
-            # Use the currently selected doc and generated doc from dropdowns
             rfp_doc = selected_filename
             gen_doc = selected_generated_dropdown_state
             logging.info(f"Starting proposal streaming with RFP: {rfp_doc}, Generated Doc: {gen_doc}")
@@ -550,28 +591,26 @@ def master_callback(
                 generated_documents[generated_docx_name] = generated_docx_bytes
                 logging.info(f"Generated proposal docx saved: {generated_docx_name}")
                 new_selected_generated = generated_docx_name
+            generated_doc_options = [{'label': fn, 'value': fn} for fn in generated_documents.keys()]
+            generated_doc_value = new_selected_generated if new_selected_generated in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
+            generated_doc_list = get_generated_doc_list(generated_documents)
+            output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
 
         elif triggered_id == 'compliance-action-btn':
             result, _, _ = process_document('compliance', selected_filename, chat_input)
+            output_data_upload = html.Div(result, style={"wordWrap": "break-word"})
         elif triggered_id == 'recover-action-btn':
             result, _, _ = process_document('recover', selected_filename, chat_input)
+            output_data_upload = html.Div(result, style={"wordWrap": "break-word"})
         elif triggered_id == 'board-action-btn':
             result, _, _ = process_document('board', selected_filename, chat_input)
+            output_data_upload = html.Div(result, style={"wordWrap": "break-word"})
         elif triggered_id == 'loe-action-btn':
             result, _, _ = process_document('loe', selected_filename, chat_input)
+            output_data_upload = html.Div(result, style={"wordWrap": "break-word"})
         else:
             result = "Action not implemented yet."
-
-        if isinstance(result, str) and result.strip().startswith("Error"):
-            output_data_upload = html.Div(result, style={"wordWrap": "break-word"})
-        elif 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):
             output_data_upload = html.Div(result, style={"wordWrap": "break-word"})
-        else:
-            output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
-
-        generated_doc_options = [{'label': fn, 'value': fn} for fn in generated_documents.keys()]
-        generated_doc_value = new_selected_generated if new_selected_generated in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
-        generated_doc_list = get_generated_doc_list(generated_documents)
 
     elif triggered_id == 'select-generated-dropdown':
         sel_gen = selected_generated_dropdown
@@ -605,6 +644,7 @@ def master_callback(
         generated_doc_list,
         generated_doc_options,
         generated_doc_value,
+        shredded_doc_list_items,
         output_data_upload
     )