Update app.py via AI Editor

app.py

@@ -634,28 +634,6 @@ app.clientside_callback(
     Input('dummy-url', 'pathname')
 )
 
-@app.callback(
-    Output('documents-list', 'children'),
-    Output('select-document-dropdown', 'options'),
-    Output('select-document-dropdown', 'value'),
-    Output('proposals-list', 'children'),
-    Output('select-proposal-dropdown', 'options'),
-    Output('select-proposal-dropdown', 'value'),
-    Output('preview-window-state', 'data'),
-    Input('session-id-store', 'data'),
-    prevent_initial_call=False
-)
-def refresh_session_files(session_id):
-    sid = get_session_id(session_id)
-    sess_data = get_session_data(sid)
-    doc_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["uploaded_documents"].keys()]
-    doc_value = next(iter(sess_data["uploaded_documents"]), None) if sess_data["uploaded_documents"] else None
-    proposal_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["proposals"].keys()]
-    proposal_value = next(iter(sess_data["proposals"]), None) if sess_data["proposals"] else None
-    documents_list = get_documents_list(sess_data["uploaded_documents"], sess_data["shredded_documents"])
-    proposals_list = get_proposals_list(sess_data["proposals"])
-    return documents_list, doc_options, doc_value, proposals_list, proposal_options, proposal_value, "expanded"
-
 @app.callback(
     Output('output-data-upload', 'children'),
     Output('documents-list', 'children'),
@@ -683,21 +661,25 @@ def refresh_session_files(session_id):
         State('chat-input', 'value'),
         Input('cancel-action-btn', 'n_clicks'),
         State('preview-window-state', 'data'),
-        State('session-id-store', 'data')
+        State('session-id-store', 'data'),
+        Input('session-id-store', 'data')
     ],
-    prevent_initial_call=True
+    prevent_initial_call=False
 )
-def master_callback(
+def unified_master_callback(
     shred_clicks, proposal_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks,
     rfp_content, rfp_filename, doc_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
     chat_input, cancel_clicks, preview_window_state,
-    session_id
+    session_id_state, session_id_store_input
 ):
-    sid = get_session_id(session_id)
+    # Determine session_id
+    sid = get_session_id(session_id_state if session_id_state else session_id_store_input)
     sess_data = get_session_data(sid)
     ctx = callback_context
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None
+    output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
+    uploaded_rfp_decoded_bytes = None
 
     def safe_get_n_clicks(ctx, idx):
         try:
@@ -707,7 +689,21 @@ def master_callback(
 
     doc_delete_clicks = safe_get_n_clicks(ctx, 8)
     proposal_delete_clicks = safe_get_n_clicks(ctx, 12)
-    uploaded_rfp_decoded_bytes = None
+
+    # On first page load, just show the lists
+    if not ctx.triggered or triggered_id in ['session-id-store']:
+        doc_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["uploaded_documents"].keys()]
+        doc_value = next(iter(sess_data["uploaded_documents"]), None) if sess_data["uploaded_documents"] else None
+        proposal_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["proposals"].keys()]
+        proposal_value = next(iter(sess_data["proposals"]), None) if sess_data["proposals"] else None
+        documents_list = get_documents_list(sess_data["uploaded_documents"], sess_data["shredded_documents"])
+        proposals_list = get_proposals_list(sess_data["proposals"])
+        return (
+            output_data_upload,
+            documents_list, doc_options, doc_value,
+            proposals_list, proposal_options, proposal_value,
+            "expanded"
+        )
 
     if triggered_id == 'cancel-action-btn':
         lock = sess_data.get("gemini_lock")
@@ -811,8 +807,6 @@ def master_callback(
     proposal_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["proposals"].keys()]
     proposal_value = selected_proposal if selected_proposal in sess_data["proposals"] else (next(iter(sess_data["proposals"]), None) if sess_data["proposals"] else None)
 
-    output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
-
     action_btns = [
         'shred-action-btn', 'proposal-action-btn', 'compliance-action-btn',
         'recover-action-btn', 'board-action-btn', 'loe-action-btn'