Update app.py via AI Editor

app.py

@@ -242,13 +242,14 @@ def update_shred_upload(contents, delete_clicks, filename, stored_data):
     if trig_id == "shred-upload":
         if contents and filename:
             logger.info(f"Document uploaded in Shred: {filename}")
-            text = process_document(contents, filename)
+            full_text = process_document(contents, filename)
             preview = html.Div([
                 html.B(f"Uploaded: {filename}"),
                 html.Br(),
-                html.Div(text[:2000] + ("..." if len(text) > 2000 else ""), style={"whiteSpace": "pre-wrap", "overflowWrap": "break-word", "fontSize": "small"})
+                html.Div(full_text[:2000] + ("..." if len(full_text) > 2000 else ""), style={"whiteSpace": "pre-wrap", "overflowWrap": "break-word", "fontSize": "small"})
             ])
-            return {"contents": contents, "filename": filename, "preview": text[:2000]}, preview, {"display": "block"}
+            # Store the full document text, filename, and preview
+            return {"contents": contents, "filename": filename, "full_text": full_text, "preview": full_text[:2000]}, preview, {"display": "block"}
         else:
             return None, "", {"display": "none"}
     elif trig_id == "shred-delete-btn":
@@ -294,10 +295,11 @@ def handle_all_tabs(*args):
                 logger.info(f"Generate Shred button pressed for {tab_id}")
                 shred_data = args[shred_upload_store_idx]
                 instr = args[instr_idx] or ""
-                if shred_data and "contents" in shred_data and "filename" in shred_data:
-                    doc = process_document(shred_data["contents"], shred_data["filename"])
+                if shred_data and "full_text" in shred_data and shred_data["full_text"]:
+                    doc = shred_data["full_text"]
                     logger.info(f"Shred document will be sent to Anthropic with instructions: {instr}")
                     content, df = generate_content(doc, "Shred", instr)
+                    # Store Anthropic result in outputs
                     if not df.empty:
                         outputs[out_idx] = dash_table.DataTable(
                             data=df.to_dict('records'),
@@ -315,13 +317,13 @@ def handle_all_tabs(*args):
                     outputs[out_idx] = "Please upload a document to begin."
             elif trig == shred_dl_btn:
                 prev_output = args[prev_output_idx]
-                shred_data = args[shred_upload_store_idx]
-                if prev_output and hasattr(prev_output, 'props') and 'data' in prev_output.props:
+                if (hasattr(prev_output, 'props') and 'data' in prev_output.props):
                     df = pd.DataFrame(prev_output.props['data'])
                     buffer = BytesIO()
                     df.to_csv(buffer, index=False)
                     outputs[dl_idx] = dcc.send_bytes(buffer.getvalue(), f"{tab_id}_report.csv")
                 elif prev_output:
+                    # If prev_output is markdown or text
                     buffer = BytesIO(prev_output.encode("utf-8") if isinstance(prev_output, str) else b"")
                     outputs[dl_idx] = dcc.send_bytes(buffer.getvalue(), f"{tab_id}_report.txt")
                 else: