Update app.py via AI Editor

app.py

@@ -668,6 +668,9 @@ def master_callback(
         )
 
     # File uploads
+    doc_value = selected_doc
+    proposal_value = selected_proposal
+
     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)
@@ -682,6 +685,7 @@ def master_callback(
             if fileid:
                 session_dict['uploaded_documents_fileid'][rfp_filename] = fileid
             logging.info(f"Document uploaded: {rfp_filename}")
+            doc_value = rfp_filename  # Set selection to the newly uploaded doc
         else:
             logging.error(f"Failed to decode uploaded document: {rfp_filename}")
 
@@ -697,10 +701,12 @@ def master_callback(
             if fileid:
                 session_dict['proposals_fileid'][proposal_filename] = fileid
             logging.info(f"Proposal uploaded: {proposal_filename}")
+            proposal_value = proposal_filename  # Set selection to the newly uploaded proposal
         else:
             logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
 
     # Deletion handling
+    deleted_doc = False
     if triggered_id and isinstance(doc_delete_clicks, list):
         for i, n_click in enumerate(doc_delete_clicks):
             if n_click:
@@ -720,10 +726,12 @@ def master_callback(
                 if del_filename in session_dict['shredded_documents']:
                     del session_dict['shredded_documents'][del_filename]
                     logging.info(f"Shredded doc deleted: {del_filename}")
-                if selected_doc == del_filename:
-                    selected_doc = None
+                if doc_value == del_filename:
+                    doc_value = None
+                    deleted_doc = True
                 break
 
+    deleted_proposal = False
     if triggered_id and isinstance(proposal_delete_clicks, list):
         for i, n_click in enumerate(proposal_delete_clicks):
             if n_click:
@@ -738,16 +746,20 @@ def master_callback(
                             logging.warning(f"Failed to delete Gemini proposal file {del_filename}: {e}")
                         del session_dict['proposals_fileid'][del_filename]
                     logging.info(f"Proposal deleted: {del_filename}")
-                if selected_proposal == del_filename:
-                    selected_proposal = None
+                if proposal_value == del_filename:
+                    proposal_value = None
+                    deleted_proposal = True
                 break
 
     doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
-    doc_value = selected_doc if selected_doc in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
+    # Always keep the selected doc if it's still present, otherwise select the first available or None
+    if not doc_value or doc_value not in session_dict['uploaded_documents']:
+        doc_value = next(iter(session_dict['uploaded_documents']), None)
+    proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]
+    if not proposal_value or proposal_value not in session_dict['proposals']:
+        proposal_value = next(iter(session_dict['proposals']), None)
     documents_list = get_documents_list(session_dict['uploaded_documents'], session_dict['shredded_documents'])
     proposals_list = get_proposals_list(session_dict['proposals'])
-    proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]
-    proposal_value = selected_proposal if selected_proposal in session_dict['proposals'] else (next(iter(session_dict['proposals']), None) if session_dict['proposals'] else None)
 
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
 
@@ -801,10 +813,13 @@ def master_callback(
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
         finally:
             lock.release()
+        # After processing, update lists/options/values, but do NOT clear any document/proposal unless deleted
         doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
-        doc_value = doc_value if doc_value in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
+        if not doc_value or doc_value not in session_dict['uploaded_documents']:
+            doc_value = next(iter(session_dict['uploaded_documents']), None)
         proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]
-        proposal_value = proposal_value if proposal_value in session_dict['proposals'] else (next(iter(session_dict['proposals']), None) if session_dict['proposals'] else None)
+        if not proposal_value or proposal_value not in session_dict['proposals']:
+            proposal_value = next(iter(session_dict['proposals']), None)
         documents_list = get_documents_list(session_dict['uploaded_documents'], session_dict['shredded_documents'])
         proposals_list = get_proposals_list(session_dict['proposals'])
         return (
@@ -814,8 +829,7 @@ def master_callback(
             "expanded"
         )
 
-    doc_value = doc_value if doc_value in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
-    proposal_value = proposal_value if proposal_value in session_dict['proposals'] else (next(iter(session_dict['proposals']), None) if session_dict['proposals'] else None)
+    # Always keep selections sticky, unless user has deleted the item
     return (
         output_data_upload,
         documents_list, doc_options, doc_value,