Update app.py via AI Editor

app.py

@@ -188,7 +188,6 @@ def process_document(action, selected_filename=None, chat_input=None):
         if result_holder["docx_bytes"] and result_holder["docx_name"]:
             docx_name = result_holder["docx_name"]
             try:
-                # We'll store the text as the plain shredded text (not binary docx) in uploaded_documents
                 uploaded_documents[docx_name] = result_holder["text"]
             except Exception as e:
                 logging.error(f"Error adding shredded docx to uploaded_documents: {e}")
@@ -549,7 +548,7 @@ def master_callback(
                         del uploaded_documents_fileid[del_filename]
                     logging.info(f"Document deleted: {del_filename}")
                     if selected_doc == del_filename:
-                        selected_doc = next(iter(uploaded_documents), None)
+                        selected_doc = None
                     upload_triggered = True
                     break
 
@@ -568,7 +567,7 @@ def master_callback(
                         del uploaded_proposals_fileid[del_filename]
                     logging.info(f"Proposal deleted: {del_filename}")
                     if selected_proposal == del_filename:
-                        selected_proposal = next(iter(uploaded_proposals), None)
+                        selected_proposal = None
                     upload_triggered = True
                     break
 
@@ -581,7 +580,7 @@ def master_callback(
                     del generated_documents[del_filename]
                     logging.info(f"Generated doc deleted: {del_filename}")
                     if selected_generated == del_filename:
-                        selected_generated = next(iter(generated_documents), None)
+                        selected_generated = None
                     upload_triggered = True
                     break
 
@@ -596,6 +595,7 @@ 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)
@@ -608,21 +608,20 @@ def master_callback(
 
     if triggered_id == 'shred-action-btn':
         output_data_upload = dcc.Loading(type="default", children=html.Div("Shredding document...", style={"wordWrap": "break-word"}))
-        shred_text, shredded_docx_bytes, shredded_docx_name, shredded_text = process_document('shred', selected_filename, chat_input)
+        shred_text, shredded_docx_bytes, shredded_docx_name, shredded_text = process_document('shred', doc_value, chat_input)
         shred_store = {'text': shred_text, 'docx_bytes': shredded_docx_bytes, 'docx_name': shredded_docx_name}
 
         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}")
-            # Add shredded docx to uploaded_documents as text so it appears as selectable
             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
+            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)
             shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
         output_data_upload = dcc.Markdown(shred_text, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
-
         return (
             shred_store, proposal_store, output_data_upload,
             uploaded_doc_list, doc_options, doc_value,
@@ -632,13 +631,13 @@ def master_callback(
 
     if triggered_id == 'proposal-action-btn':
         output_data_upload = dcc.Loading(type="default", children=html.Div("Generating proposal...", style={"wordWrap": "break-word"}))
-        proposal_text, proposal_docx_bytes, proposal_docx_name, proposal_text_preview = process_document('proposal', selected_filename, chat_input)
+        proposal_text, proposal_docx_bytes, proposal_docx_name, proposal_text_preview = process_document('proposal', doc_value, chat_input)
         proposal_store = {'text': proposal_text, 'docx_bytes': proposal_docx_bytes, 'docx_name': proposal_docx_name}
         new_generated_doc_value = generated_doc_value
         if proposal_docx_bytes and proposal_docx_name:
             generated_documents[proposal_docx_name] = proposal_docx_bytes
             generated_doc_options = [{'label': fn, 'value': fn} for fn in generated_documents.keys()]
-            new_generated_doc_value = proposal_docx_name
+            new_generated_doc_value = proposal_docx_name if proposal_docx_name in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
             generated_doc_list = get_generated_doc_list(generated_documents)
             logging.info(f"Generated proposal docx saved: {proposal_docx_name}")
         output_data_upload = dcc.Markdown(proposal_text, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
@@ -675,6 +674,8 @@ def master_callback(
         )
 
     if upload_triggered:
+        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)
         output_data_upload = html.Div("Upload/Delete completed.", style={"wordWrap": "break-word"})
         return (
             shred_store, proposal_store, output_data_upload,
@@ -683,6 +684,9 @@ 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 (
         shred_store, proposal_store, output_data_upload,
         uploaded_doc_list, doc_options, doc_value,