Update app.py via AI Editor

app.py

@@ -184,7 +184,6 @@ def process_document(action, selected_filename=None, chat_input=None):
         t = Thread(target=thread_shred)
         t.start()
         t.join()
-        # Add the shredded docx to uploaded_documents as a new doc
         if result_holder["docx_bytes"] and result_holder["docx_name"]:
             docx_name = result_holder["docx_name"]
             try:
@@ -335,6 +334,18 @@ def get_generated_doc_list(docdict):
         )
     return dbc.ListGroup(doc_list, flush=True)
 
+def get_ctx_input_list_item(ctx, idx, default):
+    try:
+        return ctx.inputs_list[idx]
+    except Exception:
+        return default
+
+def get_ctx_input_n_clicks(ctx, idx):
+    try:
+        return ctx.inputs_list[idx]
+    except Exception:
+        return []
+
 app.layout = dbc.Container([
     dcc.Store(id='shred-store', data={'text': None, 'docx_bytes': None, 'docx_name': None}),
     dcc.Store(id='proposal-store', data={'text': None, 'docx_bytes': None, 'docx_name': None}),
@@ -495,12 +506,22 @@ def master_callback(
     ctx = callback_context
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None
 
+    def safe_get_n_clicks(ctx, idx):
+        try:
+            return ctx.inputs_list[idx]
+        except Exception:
+            return []
+
     upload_triggered = False
 
     shred_store = {'text': None, 'docx_bytes': None, 'docx_name': None}
     proposal_store = {'text': None, 'docx_bytes': None, 'docx_name': None}
 
-    # Handle uploads and deletes
+    rfp_delete_clicks = safe_get_n_clicks(ctx, 4)
+    proposal_delete_clicks = safe_get_n_clicks(ctx, 8)
+    generated_delete_clicks = safe_get_n_clicks(ctx, 10)
+    shredded_delete_clicks = safe_get_n_clicks(ctx, 12)
+
     if triggered_id == 'upload-document' and rfp_content is not None and rfp_filename:
         content_type, content_string = rfp_content.split(',')
         decoded = base64.b64decode(content_string)
@@ -533,7 +554,7 @@ def master_callback(
             logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
         upload_triggered = True
 
-    if triggered_id and isinstance(ctx.inputs_list[4], list):
+    if triggered_id and isinstance(rfp_delete_clicks, list):
         for i, n_click in enumerate(rfp_delete_clicks):
             if n_click:
                 btn_id = ctx.inputs_list[4][i]['id']
@@ -552,7 +573,7 @@ def master_callback(
                     upload_triggered = True
                     break
 
-    if triggered_id and isinstance(ctx.inputs_list[8], list):
+    if triggered_id and isinstance(proposal_delete_clicks, list):
         for i, n_click in enumerate(proposal_delete_clicks):
             if n_click:
                 btn_id = ctx.inputs_list[8][i]['id']
@@ -571,7 +592,7 @@ def master_callback(
                     upload_triggered = True
                     break
 
-    if triggered_id and isinstance(ctx.inputs_list[10], list):
+    if triggered_id and isinstance(generated_delete_clicks, list):
         for i, n_click in enumerate(generated_delete_clicks):
             if n_click:
                 btn_id = ctx.inputs_list[10][i]['id']
@@ -584,7 +605,7 @@ def master_callback(
                     upload_triggered = True
                     break
 
-    if triggered_id and isinstance(ctx.inputs_list[12], list):
+    if triggered_id and isinstance(shredded_delete_clicks, list):
         for i, n_click in enumerate(shredded_delete_clicks):
             if n_click:
                 btn_id = ctx.inputs_list[12][i]['id']
@@ -595,7 +616,6 @@ def master_callback(
                 upload_triggered = True
                 break
 
-    # Always recalculate options/values after any operation
     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)
     shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
@@ -616,7 +636,6 @@ def master_callback(
             logging.info(f"Shredded docx saved: {shredded_docx_name}")
             if shredded_docx_name not in uploaded_documents:
                 uploaded_documents[shredded_docx_name] = shredded_text
-            # Recompute options/values after updating
             doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
             doc_value = shredded_docx_name if shredded_docx_name in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
             uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
@@ -684,7 +703,6 @@ def master_callback(
             generated_doc_list, generated_doc_options, generated_doc_value
         )
 
-    # Always return 10 outputs
     doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
     generated_doc_value = generated_doc_value if generated_doc_value in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
     return (